mirror of
https://github.com/jackc/pgx.git
synced 2025-05-31 11:42:24 +00:00
ee93440ac1
Added ValueRoundTripTest to pgxtest Removed pgtype/testutil pgtype tests now run with all (applicable) query modes. This gives better coverage than before and revealed several bugs which are also fixed in this commit.
128 lines
3.4 KiB
Go
128 lines
3.4 KiB
Go
package pgtype_test
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
"github.com/jackc/pgx/v5/pgxtest"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUUIDCodec(t *testing.T) {
|
|
pgxtest.RunValueRoundTripTests(context.Background(), t, defaultConnTestRunner, nil, "uuid", []pgxtest.ValueRoundTripTest{
|
|
{
|
|
pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Valid: true},
|
|
new(pgtype.UUID),
|
|
isExpectedEq(pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Valid: true}),
|
|
},
|
|
{
|
|
"00010203-0405-0607-0809-0a0b0c0d0e0f",
|
|
new(pgtype.UUID),
|
|
isExpectedEq(pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Valid: true}),
|
|
},
|
|
{
|
|
"000102030405060708090a0b0c0d0e0f",
|
|
new(pgtype.UUID),
|
|
isExpectedEq(pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Valid: true}),
|
|
},
|
|
{pgtype.UUID{}, new([]byte), isExpectedEqBytes([]byte(nil))},
|
|
{pgtype.UUID{}, new(pgtype.UUID), isExpectedEq(pgtype.UUID{})},
|
|
{nil, new(pgtype.UUID), isExpectedEq(pgtype.UUID{})},
|
|
})
|
|
|
|
pgxtest.RunValueRoundTripTests(context.Background(), t, defaultConnTestRunner, pgxtest.KnownOIDQueryExecModes, "uuid", []pgxtest.ValueRoundTripTest{
|
|
{
|
|
[16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
|
|
new(pgtype.UUID),
|
|
isExpectedEq(pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Valid: true}),
|
|
},
|
|
{
|
|
[]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
|
|
new(pgtype.UUID),
|
|
isExpectedEq(pgtype.UUID{Bytes: [16]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}, Valid: true}),
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestUUID_MarshalJSON(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
src pgtype.UUID
|
|
want []byte
|
|
}{
|
|
{
|
|
name: "first",
|
|
src: pgtype.UUID{
|
|
Bytes: [16]byte{29, 72, 90, 122, 109, 24, 69, 153, 140, 108, 52, 66, 86, 22, 136, 122},
|
|
Valid: true,
|
|
},
|
|
want: []byte(`"1d485a7a-6d18-4599-8c6c-34425616887a"`),
|
|
},
|
|
{
|
|
name: "third",
|
|
src: pgtype.UUID{
|
|
Bytes: [16]byte{},
|
|
},
|
|
want: []byte("null"),
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := tt.src.MarshalJSON()
|
|
require.NoError(t, err)
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("MarshalJSON() got = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUUID_UnmarshalJSON(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
want *pgtype.UUID
|
|
src []byte
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "first",
|
|
want: &pgtype.UUID{
|
|
Bytes: [16]byte{29, 72, 90, 122, 109, 24, 69, 153, 140, 108, 52, 66, 86, 22, 136, 122},
|
|
Valid: true,
|
|
},
|
|
src: []byte(`"1d485a7a-6d18-4599-8c6c-34425616887a"`),
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "second",
|
|
want: &pgtype.UUID{
|
|
Bytes: [16]byte{},
|
|
},
|
|
src: []byte("null"),
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "third",
|
|
want: &pgtype.UUID{
|
|
Bytes: [16]byte{},
|
|
Valid: false,
|
|
},
|
|
src: []byte("1d485a7a-6d18-4599-8c6c-34425616887a"),
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := &pgtype.UUID{}
|
|
if err := got.UnmarshalJSON(tt.src); (err != nil) != tt.wantErr {
|
|
t.Errorf("UnmarshalJSON() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("UnmarshalJSON() got = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|