Test SaveObjectToFile.

algorand · Nov 13, 2023 · 6a7af22 · 6a7af22
1 parent 231494e
commit 6a7af22
Showing 1 changed file with 40 additions and 0 deletions.
diff --git a/util/codecs/json_test.go b/util/codecs/json_test.go
@@ -18,6 +18,8 @@ package codecs
 
 import (
 	"bytes"
+	"os"
+	"path"
 	"testing"
 	"time"
 
@@ -58,6 +60,44 @@ func TestIsDefaultValue(t *testing.T) {
 	a.True(isDefaultValue("Missing", objectValues, defaultValues))
 }
 
+func TestSaveObjectToFile(t *testing.T) {
+	partitiontest.PartitionTest(t)
+	t.Parallel()
+
+	type TestType struct {
+		A uint64
+		B string
+	}
+
+	obj := TestType{1024, "test"}
+
+	// prettyFormat = false
+	{
+		filename := path.Join(t.TempDir(), "test.json")
+		SaveObjectToFile(filename, obj, false)
+		data, err := os.ReadFile(filename)
+		require.NoError(t, err)
+		expected := `{"A":1024,"B":"test"}
+`
+		require.Equal(t, expected, string(data))
+	}
+
+	// prettyFormat = true
+	{
+		filename := path.Join(t.TempDir(), "test.json")
+		SaveObjectToFile(filename, obj, true)
+		data, err := os.ReadFile(filename)
+		require.NoError(t, err)
+		expected := `{
+	"A": 1024,
+	"B": "test"
+}
+`
+		require.Equal(t, expected, string(data))
+	}
+
+}
+
 func TestWriteNonDefaultValue(t *testing.T) {
 	partitiontest.PartitionTest(t)
 	t.Parallel()