Add float4, float8 and arrays

2017-03-04 13:29:04 -06:00
parent 39b60605ae
commit 0f115477de
14 changed files with 1559 additions and 5 deletions
@@ -0,0 +1,148 @@
+package pgtype_test
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/jackc/pgx/pgtype"
+)
+
+func TestFloat8Transcode(t *testing.T) {
+	testSuccessfulTranscode(t, "float8", []interface{}{
+		pgtype.Float8{Float: -1, Status: pgtype.Present},
+		pgtype.Float8{Float: 0, Status: pgtype.Present},
+		pgtype.Float8{Float: 0.00001, Status: pgtype.Present},
+		pgtype.Float8{Float: 1, Status: pgtype.Present},
+		pgtype.Float8{Float: 9999.99, Status: pgtype.Present},
+		pgtype.Float8{Float: 0, Status: pgtype.Null},
+	})
+}
+
+func TestFloat8ConvertFrom(t *testing.T) {
+	successfulTests := []struct {
+		source interface{}
+		result pgtype.Float8
+	}{
+		{source: float32(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: float64(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: int8(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: int16(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: int32(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: int64(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: int8(-1), result: pgtype.Float8{Float: -1, Status: pgtype.Present}},
+		{source: int16(-1), result: pgtype.Float8{Float: -1, Status: pgtype.Present}},
+		{source: int32(-1), result: pgtype.Float8{Float: -1, Status: pgtype.Present}},
+		{source: int64(-1), result: pgtype.Float8{Float: -1, Status: pgtype.Present}},
+		{source: uint8(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: uint16(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: uint32(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: uint64(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: "1", result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+		{source: _int8(1), result: pgtype.Float8{Float: 1, Status: pgtype.Present}},
+	}
+
+	for i, tt := range successfulTests {
+		var r pgtype.Float8
+		err := r.ConvertFrom(tt.source)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if r != tt.result {
+			t.Errorf("%d: expected %v to convert to %v, but it was %v", i, tt.source, tt.result, r)
+		}
+	}
+}
+
+func TestFloat8AssignTo(t *testing.T) {
+	var i8 int8
+	var i16 int16
+	var i32 int32
+	var i64 int64
+	var i int
+	var ui8 uint8
+	var ui16 uint16
+	var ui32 uint32
+	var ui64 uint64
+	var ui uint
+	var pi8 *int8
+	var _i8 _int8
+	var _pi8 *_int8
+	var f32 float32
+	var f64 float64
+	var pf32 *float32
+	var pf64 *float64
+
+	simpleTests := []struct {
+		src      pgtype.Float8
+		dst      interface{}
+		expected interface{}
+	}{
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &f32, expected: float32(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &f64, expected: float64(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &i16, expected: int16(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &i32, expected: int32(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &i64, expected: int64(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &i, expected: int(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &ui8, expected: uint8(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &ui16, expected: uint16(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &ui32, expected: uint32(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &ui64, expected: uint64(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &ui, expected: uint(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &_i8, expected: _int8(42)},
+		{src: pgtype.Float8{Float: 0, Status: pgtype.Null}, dst: &pi8, expected: ((*int8)(nil))},
+		{src: pgtype.Float8{Float: 0, Status: pgtype.Null}, dst: &_pi8, expected: ((*_int8)(nil))},
+	}
+
+	for i, tt := range simpleTests {
+		err := tt.src.AssignTo(tt.dst)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if dst := reflect.ValueOf(tt.dst).Elem().Interface(); 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.Float8
+		dst      interface{}
+		expected interface{}
+	}{
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &pf32, expected: float32(42)},
+		{src: pgtype.Float8{Float: 42, Status: pgtype.Present}, dst: &pf64, expected: float64(42)},
+	}
+
+	for i, tt := range pointerAllocTests {
+		err := tt.src.AssignTo(tt.dst)
+		if err != nil {
+			t.Errorf("%d: %v", i, err)
+		}
+
+		if dst := reflect.ValueOf(tt.dst).Elem().Elem().Interface(); dst != tt.expected {
+			t.Errorf("%d: expected %v to assign %v, but result was %v", i, tt.src, tt.expected, dst)
+		}
+	}
+
+	errorTests := []struct {
+		src pgtype.Float8
+		dst interface{}
+	}{
+		{src: pgtype.Float8{Float: 150, Status: pgtype.Present}, dst: &i8},
+		{src: pgtype.Float8{Float: 40000, Status: pgtype.Present}, dst: &i16},
+		{src: pgtype.Float8{Float: -1, Status: pgtype.Present}, dst: &ui8},
+		{src: pgtype.Float8{Float: -1, Status: pgtype.Present}, dst: &ui16},
+		{src: pgtype.Float8{Float: -1, Status: pgtype.Present}, dst: &ui32},
+		{src: pgtype.Float8{Float: -1, Status: pgtype.Present}, dst: &ui64},
+		{src: pgtype.Float8{Float: -1, Status: pgtype.Present}, dst: &ui},
+		{src: pgtype.Float8{Float: 0, Status: pgtype.Null}, dst: &i32},
+	}
+
+	for i, tt := range errorTests {
+		err := tt.src.AssignTo(tt.dst)
+		if err == nil {
+			t.Errorf("%d: expected error but none was returned (%v -> %v)", i, tt.src, tt.dst)
+		}
+	}
+}