package pgtype_test import ( "reflect" "testing" "github.com/jackc/pgtype" "github.com/jackc/pgtype/testutil" ) func TestInt2ArrayTranscode(t *testing.T) { testutil.TestSuccessfulTranscode(t, "int2[]", []interface{}{ &pgtype.Int2Array{ Elements: nil, Dimensions: nil, Valid: true, }, &pgtype.Int2Array{ Elements: []pgtype.Int2{ {Int: 1, Valid: true}, {}, }, Dimensions: []pgtype.ArrayDimension{{Length: 2, LowerBound: 1}}, Valid: true, }, &pgtype.Int2Array{}, &pgtype.Int2Array{ Elements: []pgtype.Int2{ {Int: 1, Valid: true}, {Int: 2, Valid: true}, {Int: 3, Valid: true}, {Int: 4, Valid: true}, {}, {Int: 6, Valid: true}, }, Dimensions: []pgtype.ArrayDimension{{Length: 3, LowerBound: 1}, {Length: 2, LowerBound: 1}}, Valid: true, }, &pgtype.Int2Array{ Elements: []pgtype.Int2{ {Int: 1, Valid: true}, {Int: 2, Valid: true}, {Int: 3, Valid: true}, {Int: 4, Valid: true}, }, Dimensions: []pgtype.ArrayDimension{ {Length: 2, LowerBound: 4}, {Length: 2, LowerBound: 2}, }, Valid: true, }, }) } func TestInt2ArraySet(t *testing.T) { successfulTests := []struct { source interface{} result pgtype.Int2Array }{ { source: []int64{1}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true}, }, { source: []int32{1}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true}, }, { source: []int16{1}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true}, }, { source: []int{1}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true}, }, { source: []uint64{1}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true}, }, { source: []uint32{1}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true}, }, { source: []uint16{1}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true}, }, { source: (([]int16)(nil)), result: pgtype.Int2Array{}, }, { source: [][]int16{{1}, {2}}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}, {Int: 2, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}}, Valid: true}, }, { source: [][][][]int16{{{{1, 2, 3}}}, {{{4, 5, 6}}}}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{ {Int: 1, Valid: true}, {Int: 2, Valid: true}, {Int: 3, Valid: true}, {Int: 4, Valid: true}, {Int: 5, Valid: true}, {Int: 6, Valid: true}}, Dimensions: []pgtype.ArrayDimension{ {LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 3}}, Valid: true}, }, { source: [2][1]int16{{1}, {2}}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}, {Int: 2, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}}, Valid: true}, }, { source: [2][1][1][3]int16{{{{1, 2, 3}}}, {{{4, 5, 6}}}}, result: pgtype.Int2Array{ Elements: []pgtype.Int2{ {Int: 1, Valid: true}, {Int: 2, Valid: true}, {Int: 3, Valid: true}, {Int: 4, Valid: true}, {Int: 5, Valid: true}, {Int: 6, Valid: true}}, Dimensions: []pgtype.ArrayDimension{ {LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 3}}, Valid: true}, }, } for i, tt := range successfulTests { var r pgtype.Int2Array err := r.Set(tt.source) if err != nil { t.Errorf("%d: %v", i, err) } if !reflect.DeepEqual(r, tt.result) { t.Errorf("%d: expected %v to convert to %v, but it was %v", i, tt.source, tt.result, r) } } } func TestInt2ArrayAssignTo(t *testing.T) { var int16Slice []int16 var uint16Slice []uint16 var namedInt16Slice _int16Slice var int16SliceDim2 [][]int16 var int16SliceDim4 [][][][]int16 var int16ArrayDim2 [2][1]int16 var int16ArrayDim4 [2][1][1][3]int16 simpleTests := []struct { src pgtype.Int2Array dst interface{} expected interface{} }{ { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true, }, dst: &int16Slice, expected: []int16{1}, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true, }, dst: &uint16Slice, expected: []uint16{1}, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true, }, dst: &namedInt16Slice, expected: _int16Slice{1}, }, { src: pgtype.Int2Array{}, dst: &int16Slice, expected: (([]int16)(nil)), }, { src: pgtype.Int2Array{Valid: true}, dst: &int16Slice, expected: []int16{}, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}, {Int: 2, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}}, Valid: true}, expected: [][]int16{{1}, {2}}, dst: &int16SliceDim2, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{ {Int: 1, Valid: true}, {Int: 2, Valid: true}, {Int: 3, Valid: true}, {Int: 4, Valid: true}, {Int: 5, Valid: true}, {Int: 6, Valid: true}}, Dimensions: []pgtype.ArrayDimension{ {LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 3}}, Valid: true}, expected: [][][][]int16{{{{1, 2, 3}}}, {{{4, 5, 6}}}}, dst: &int16SliceDim4, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}, {Int: 2, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}}, Valid: true}, expected: [2][1]int16{{1}, {2}}, dst: &int16ArrayDim2, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{ {Int: 1, Valid: true}, {Int: 2, Valid: true}, {Int: 3, Valid: true}, {Int: 4, Valid: true}, {Int: 5, Valid: true}, {Int: 6, Valid: true}}, Dimensions: []pgtype.ArrayDimension{ {LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 3}}, Valid: true}, expected: [2][1][1][3]int16{{{{1, 2, 3}}}, {{{4, 5, 6}}}}, dst: &int16ArrayDim4, }, } 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(); !reflect.DeepEqual(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.Int2Array dst interface{} }{ { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true, }, dst: &int16Slice, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: -1, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}}, Valid: true, }, dst: &uint16Slice, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}, {Int: 2, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 2}}, Valid: true}, dst: &int16ArrayDim2, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}, {Int: 2, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 1}, {LowerBound: 1, Length: 2}}, Valid: true}, dst: &int16Slice, }, { src: pgtype.Int2Array{ Elements: []pgtype.Int2{{Int: 1, Valid: true}, {Int: 2, Valid: true}}, Dimensions: []pgtype.ArrayDimension{{LowerBound: 1, Length: 2}, {LowerBound: 1, Length: 1}}, Valid: true}, dst: &int16ArrayDim4, }, } 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) } } }