Revert "Temporarily delete tests and pgxtype to break recursive dependency with pgx"

This reverts commit 32e20a6031.
2021-07-24 10:40:30 -05:00
parent 32e20a6031
commit c16a4f7d6a
89 changed files with 14674 additions and 0 deletions
@@ -0,0 +1,84 @@
+package pgtype_test
+
+import (
+	"context"
+	"testing"
+
+	"github.com/jackc/pgtype"
+	"github.com/jackc/pgtype/testutil"
+	"github.com/stretchr/testify/require"
+)
+
+func TestArrayTypeValue(t *testing.T) {
+	arrayType := pgtype.NewArrayType("_text", pgtype.TextOID, func() pgtype.ValueTranscoder { return &pgtype.Text{} })
+
+	err := arrayType.Set(nil)
+	require.NoError(t, err)
+
+	gotValue := arrayType.Get()
+	require.Nil(t, gotValue)
+
+	slice := []string{"foo", "bar"}
+	err = arrayType.AssignTo(&slice)
+	require.NoError(t, err)
+	require.Nil(t, slice)
+
+	err = arrayType.Set([]string{})
+	require.NoError(t, err)
+
+	gotValue = arrayType.Get()
+	require.Len(t, gotValue, 0)
+
+	err = arrayType.AssignTo(&slice)
+	require.NoError(t, err)
+	require.EqualValues(t, []string{}, slice)
+
+	err = arrayType.Set([]string{"baz", "quz"})
+	require.NoError(t, err)
+
+	gotValue = arrayType.Get()
+	require.Len(t, gotValue, 2)
+
+	err = arrayType.AssignTo(&slice)
+	require.NoError(t, err)
+	require.EqualValues(t, []string{"baz", "quz"}, slice)
+}
+
+func TestArrayTypeTranscode(t *testing.T) {
+	conn := testutil.MustConnectPgx(t)
+	defer testutil.MustCloseContext(t, conn)
+
+	conn.ConnInfo().RegisterDataType(pgtype.DataType{
+		Value: pgtype.NewArrayType("_text", pgtype.TextOID, func() pgtype.ValueTranscoder { return &pgtype.Text{} }),
+		Name:  "_text",
+		OID:   pgtype.TextArrayOID,
+	})
+
+	var dstStrings []string
+	err := conn.QueryRow(context.Background(), "select $1::text[]", []string{"red", "green", "blue"}).Scan(&dstStrings)
+	require.NoError(t, err)
+
+	require.EqualValues(t, []string{"red", "green", "blue"}, dstStrings)
+}
+
+func TestArrayTypeEmptyArrayDoesNotBreakArrayType(t *testing.T) {
+	conn := testutil.MustConnectPgx(t)
+	defer testutil.MustCloseContext(t, conn)
+
+	conn.ConnInfo().RegisterDataType(pgtype.DataType{
+		Value: pgtype.NewArrayType("_text", pgtype.TextOID, func() pgtype.ValueTranscoder { return &pgtype.Text{} }),
+		Name:  "_text",
+		OID:   pgtype.TextArrayOID,
+	})
+
+	var dstStrings []string
+	err := conn.QueryRow(context.Background(), "select '{}'::text[]").Scan(&dstStrings)
+	require.NoError(t, err)
+
+	require.EqualValues(t, []string{}, dstStrings)
+
+	err = conn.QueryRow(context.Background(), "select $1::text[]", []string{"red", "green", "blue"}).Scan(&dstStrings)
+	require.NoError(t, err)
+
+	require.EqualValues(t, []string{"red", "green", "blue"}, dstStrings)
+}