chore: import upstream snapshot with attribution

testing/go/binding_test.go

@@ -0,0 +1,120 @@
+// Copyright 2025 Dolthub, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package _go
+
+import (
+	"strconv"
+	"testing"
+
+	"github.com/jackc/pgx/v5/pgtype"
+
+	"github.com/stretchr/testify/require"
+)
+
+// TestBindingWithOidZero tests the behavior of binding parameters when the client specifies a zero OID for any of
+// the parameters.
+func TestBindingWithOidZero(t *testing.T) {
+	// Start up a test server
+	ctx, connection, controller := CreateServer(t, "postgres")
+	defer func() {
+		connection.Close(ctx)
+		controller.Stop()
+		require.NoError(t, controller.WaitForStop())
+	}()
+	conn := connection.Default
+
+	// Create a table to insert into
+	_, err := connection.Exec(ctx, "CREATE TABLE my_table (id INT, name varchar(100));")
+	require.NoError(t, err)
+
+	args := [][]byte{
+		[]byte(strconv.Itoa(42)),
+		[]byte("Alice"),
+	}
+	paramOIDs := []uint32{0, 123}
+	paramFormats := []int16{0, 0}
+	sql := "INSERT INTO my_table (id, name) VALUES ($1, $2);"
+
+	// Execute a query with the zero OID and assert that we don't get an error
+	resultReader := conn.PgConn().ExecParams(ctx, sql, args, paramOIDs, paramFormats, nil)
+	result := resultReader.Read()
+	require.NoError(t, result.Err)
+}
+
+func TestIssue2386(t *testing.T) {
+	// https://github.com/dolthub/doltgresql/issues/2386
+	ctx, connection, controller := CreateServer(t, "postgres")
+	defer func() {
+		connection.Close(ctx)
+		controller.Stop()
+		require.NoError(t, controller.WaitForStop())
+	}()
+	conn := connection.Default
+	_, err := connection.Exec(ctx, "CREATE TABLE users (id INT PRIMARY KEY, name TEXT NOT NULL);")
+	require.NoError(t, err)
+	_, err = connection.Exec(ctx, "INSERT INTO users VALUES (1, 'alice'), (2, 'bob'), (3, 'carol'), (4, 'dave');")
+	require.NoError(t, err)
+	targetIDs := []int32{1, 3}
+	rows, err := conn.Query(ctx,
+		`SELECT id, name FROM users WHERE id = ANY($1)`,
+		targetIDs,
+	)
+	require.NoError(t, err)
+	defer rows.Close()
+	i := 0
+	for rows.Next() {
+		var id int32
+		var name string
+		err = rows.Scan(&id, &name)
+		require.NoError(t, err)
+		switch i {
+		case 0:
+			require.Equal(t, int32(1), id)
+			require.Equal(t, "alice", name)
+		case 1:
+			require.Equal(t, int32(3), id)
+			require.Equal(t, "carol", name)
+		default:
+			t.FailNow()
+		}
+		i++
+	}
+}
+
+func TestBindingWithTextArray(t *testing.T) {
+	ctx, connection, controller := CreateServer(t, "postgres")
+	defer func() {
+		connection.Close(ctx)
+		controller.Stop()
+		require.NoError(t, controller.WaitForStop())
+	}()
+	conn := connection.Default
+
+	m := pgtype.NewMap()
+	textArray := []string{"foo", "bar"}
+
+	plan := m.PlanEncode(pgtype.TextArrayOID, pgtype.BinaryFormatCode, textArray)
+	encodedArr, err := plan.Encode(textArray, nil)
+	require.NoError(t, err)
+
+	args := [][]byte{encodedArr}
+	paramOIDs := []uint32{1009}
+	paramFormats := []int16{1}
+	sql := "SELECT $1::text[]"
+
+	resultReader := conn.PgConn().ExecParams(ctx, sql, args, paramOIDs, paramFormats, nil)
+	result := resultReader.Read()
+	require.NoError(t, result.Err)
+}