Add json/jsonb to pgtype

2017-03-11 18:46:51 -06:00
parent 743b98b298
commit 542eac08c6
7 changed files with 438 additions and 18 deletions
@@ -0,0 +1,135 @@
+package pgtype_test
+
+import (
+	"bytes"
+	"reflect"
+	"testing"
+
+	"github.com/jackc/pgx/pgtype"
+)
+
+func TestJsonTranscode(t *testing.T) {
+	testSuccessfulTranscode(t, "json", []interface{}{
+		pgtype.Json{Bytes: []byte("{}"), Status: pgtype.Present},
+		pgtype.Json{Bytes: []byte("null"), Status: pgtype.Present},
+		pgtype.Json{Bytes: []byte("42"), Status: pgtype.Present},
+		pgtype.Json{Bytes: []byte(`"hello"`), Status: pgtype.Present},
+		pgtype.Json{Status: pgtype.Null},
+	})
+}
+
+func TestJsonConvertFrom(t *testing.T) {
+	successfulTests := []struct {
+		source interface{}
+		result pgtype.Json
+	}{
+		{source: "{}", result: pgtype.Json{Bytes: []byte("{}"), Status: pgtype.Present}},
+		{source: []byte("{}"), result: pgtype.Json{Bytes: []byte("{}"), Status: pgtype.Present}},
+		{source: ([]byte)(nil), result: pgtype.Json{Status: pgtype.Null}},
+		{source: (*string)(nil), result: pgtype.Json{Status: pgtype.Null}},
+		{source: []int{1, 2, 3}, result: pgtype.Json{Bytes: []byte("[1,2,3]"), Status: pgtype.Present}},
+		{source: map[string]interface{}{"foo": "bar"}, result: pgtype.Json{Bytes: []byte(`{"foo":"bar"}`), Status: pgtype.Present}},
+	}
+
+	for i, tt := range successfulTests {
+		var d pgtype.Json
+		err := d.ConvertFrom(tt.source)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if !reflect.DeepEqual(d, tt.result) {
+			t.Errorf("%d: expected %v to convert to %v, but it was %v", i, tt.source, tt.result, d)
+		}
+	}
+}
+
+func TestJsonAssignTo(t *testing.T) {
+	var s string
+	var ps *string
+	var b []byte
+
+	rawStringTests := []struct {
+		src      pgtype.Json
+		dst      *string
+		expected string
+	}{
+		{src: pgtype.Json{Bytes: []byte("{}"), Status: pgtype.Present}, dst: &s, expected: "{}"},
+	}
+
+	for i, tt := range rawStringTests {
+		err := tt.src.AssignTo(tt.dst)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if *tt.dst != tt.expected {
+			t.Errorf("%d: expected %v to assign %v, but result was %v", i, tt.src, tt.expected, *tt.dst)
+		}
+	}
+
+	rawBytesTests := []struct {
+		src      pgtype.Json
+		dst      *[]byte
+		expected []byte
+	}{
+		{src: pgtype.Json{Bytes: []byte("{}"), Status: pgtype.Present}, dst: &b, expected: []byte("{}")},
+		{src: pgtype.Json{Status: pgtype.Null}, dst: &b, expected: (([]byte)(nil))},
+	}
+
+	for i, tt := range rawBytesTests {
+		err := tt.src.AssignTo(tt.dst)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if bytes.Compare(tt.expected, *tt.dst) != 0 {
+			t.Errorf("%d: expected %v to assign %v, but result was %v", i, tt.src, tt.expected, *tt.dst)
+		}
+	}
+
+	var mapDst map[string]interface{}
+	type structDst struct {
+		Name string `json:"name"`
+		Age  int    `json:"age"`
+	}
+	var strDst structDst
+
+	unmarshalTests := []struct {
+		src      pgtype.Json
+		dst      interface{}
+		expected interface{}
+	}{
+		{src: pgtype.Json{Bytes: []byte(`{"foo":"bar"}`), Status: pgtype.Present}, dst: &mapDst, expected: map[string]interface{}{"foo": "bar"}},
+		{src: pgtype.Json{Bytes: []byte(`{"name":"John","age":42}`), Status: pgtype.Present}, dst: &strDst, expected: structDst{Name: "John", Age: 42}},
+	}
+	for i, tt := range unmarshalTests {
+		err := tt.src.AssignTo(tt.dst)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if dst := reflect.ValueOf(tt.dst).Elem().Interface(); !reflect.DeepEqual(dst, tt.expected) {
+			t.Errorf("%d: expected %v to assign %v, but result was %v", i, tt.src, tt.expected, dst)
+		}
+	}
+
+	pointerAllocTests := []struct {
+		src      pgtype.Json
+		dst      **string
+		expected *string
+	}{
+		{src: pgtype.Json{Status: pgtype.Null}, dst: &ps, expected: ((*string)(nil))},
+	}
+
+	for i, tt := range pointerAllocTests {
+		err := tt.src.AssignTo(tt.dst)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if *tt.dst == tt.expected {
+			t.Errorf("%d: expected %v to assign %v, but result was %v", i, tt.src, tt.expected, *tt.dst)
+		}
+	}
+}