wehub-resource-sync
121 lines
3.4 KiB
Go
121 lines
3.4 KiB
Go
// 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)
|
|
}
|