chore: import upstream snapshot with attribution

2026-07-13 12:32:25 +08:00
commit e014feafe1
2285 changed files with 1131979 additions and 0 deletions
@@ -0,0 +1,5 @@
+module libpq-test
+
+go 1.26.3
+
+require github.com/lib/pq v1.12.3 // indirect
@@ -0,0 +1,2 @@
+github.com/lib/pq v1.12.3 h1:tTWxr2YLKwIvK90ZXEw8GP7UFHtcbTtty8zsI+YjrfQ=
+github.com/lib/pq v1.12.3/go.mod h1:/p+8NSbOcwzAEI7wiMXFlgydTwcgTr3OSKMsD2BitpA=
@@ -0,0 +1,191 @@
+// Copyright 2026 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 main
+
+import (
+	"database/sql"
+	"fmt"
+	"os"
+
+	_ "github.com/lib/pq"
+)
+
+type ResFunc func(rows *sql.Rows) error
+
+type StmtTest struct {
+	Query string
+	Args  []interface{}
+	Res   []ResFunc
+}
+
+func main() {
+	user := os.Args[1]
+	port := os.Args[2]
+
+	connStr := fmt.Sprintf("host=localhost port=%s user=%s password=password dbname=postgres sslmode=disable", port, user)
+	db, err := sql.Open("postgres", connStr)
+	if err != nil {
+		panic(err)
+	}
+	defer db.Close()
+
+	if err = db.Ping(); err != nil {
+		panic(err)
+	}
+
+	var pk int
+	err = db.QueryRow("SELECT pk FROM test_table LIMIT 1").Scan(&pk)
+	if err != nil {
+		panic(err)
+	}
+	if pk != 1 {
+		panic(fmt.Sprintf("expected pk=1, got %d", pk))
+	}
+
+	_, err = db.Exec("INSERT INTO test_table VALUES (2)")
+	if err != nil {
+		panic(err)
+	}
+
+	var count int
+	err = db.QueryRow("SELECT COUNT(*) FROM test_table").Scan(&count)
+	if err != nil {
+		panic(err)
+	}
+	if count != 2 {
+		panic(fmt.Sprintf("expected count=2, got %d", count))
+	}
+
+	stmt, err := db.Prepare("SELECT pk FROM test_table WHERE pk = $1")
+	if err != nil {
+		panic(err)
+	}
+	defer stmt.Close()
+
+	err = stmt.QueryRow(1).Scan(&pk)
+	if err != nil {
+		panic(err)
+	}
+	if pk != 1 {
+		panic(fmt.Sprintf("expected pk=1 from prepared stmt, got %d", pk))
+	}
+
+	// dolt workflow: create table, insert, commit, branch, insert, commit, merge
+	for _, q := range []string{
+		"DROP TABLE IF EXISTS test",
+		"CREATE TABLE test (pk int, value int, PRIMARY KEY(pk))",
+		"INSERT INTO test (pk, value) VALUES (0, 0)",
+		"SELECT dolt_add('-A')",
+		"SELECT dolt_commit('-m', 'added table test')",
+		"SELECT dolt_checkout('-b', 'mybranch')",
+		"INSERT INTO test VALUES (1, 1)",
+		"SELECT dolt_commit('-a', '-m', 'updated test')",
+		"SELECT dolt_checkout('main')",
+		"SELECT dolt_merge('mybranch')",
+	} {
+		if _, err = db.Exec(q); err != nil {
+			panic(fmt.Sprintf("failed to execute %q: %v", q, err))
+		}
+	}
+
+	stmtTests := []StmtTest{
+		{
+			Query: "SELECT pk, value FROM test WHERE pk = $1",
+			Args:  []interface{}{int64(0)},
+			Res: []ResFunc{
+				func(rows *sql.Rows) error {
+					var pk, value int64
+					if err := rows.Scan(&pk, &value); err != nil {
+						return err
+					}
+					if pk != 0 || value != 0 {
+						return fmt.Errorf("expected pk=0 value=0, got pk=%d value=%d", pk, value)
+					}
+					return nil
+				},
+			},
+		},
+		{
+			Query: "SELECT COUNT(*) FROM dolt_log",
+			Res: []ResFunc{
+				func(rows *sql.Rows) error {
+					var size int64
+					if err := rows.Scan(&size); err != nil {
+						return err
+					}
+					if size != 4 {
+						return fmt.Errorf("expected 4 dolt_log entries, got %d", size)
+					}
+					return nil
+				},
+			},
+		},
+		{
+			Query: "SELECT COUNT(*) FROM test",
+			Res: []ResFunc{
+				func(rows *sql.Rows) error {
+					var size int64
+					if err := rows.Scan(&size); err != nil {
+						return err
+					}
+					if size != 2 {
+						return fmt.Errorf("expected 2 rows in test, got %d", size)
+					}
+					return nil
+				},
+			},
+		},
+	}
+
+	for _, test := range stmtTests {
+		func() {
+			stmt, err := db.Prepare(test.Query)
+			if err != nil {
+				panic(fmt.Sprintf("prepare %q: %v", test.Query, err))
+			}
+			defer func() {
+				if err := stmt.Close(); err != nil {
+					panic(fmt.Sprintf("stmt.Close() for %q: %v", test.Query, err))
+				}
+			}()
+
+			rows, err := stmt.Query(test.Args...)
+			if err != nil {
+				panic(fmt.Sprintf("query %q: %v", test.Query, err))
+			}
+			defer func() {
+				if err := rows.Close(); err != nil {
+					panic(fmt.Sprintf("rows.Close() for %q: %v", test.Query, err))
+				}
+			}()
+
+			i := 0
+			for rows.Next() {
+				if i >= len(test.Res) {
+					panic(fmt.Sprintf("too many rows for %q", test.Query))
+				}
+				if err := test.Res[i](rows); err != nil {
+					panic(fmt.Sprintf("result %d of %q: %v", i, test.Query, err))
+				}
+				i++
+			}
+			if err := rows.Err(); err != nil {
+				panic(fmt.Sprintf("rows.Err() for %q: %v", test.Query, err))
+			}
+		}()
+	}
+
+	fmt.Println("lib/pq test passed")
+}
@@ -0,0 +1,10 @@
+module pgx-test
+
+go 1.26.3
+
+require (
+	github.com/jackc/pgpassfile v1.0.0 // indirect
+	github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
+	github.com/jackc/pgx/v5 v5.10.0 // indirect
+	golang.org/x/text v0.29.0 // indirect
+)
@@ -0,0 +1,15 @@
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
+github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
+github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
+github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=
+github.com/jackc/pgx/v5 v5.10.0 h1:VhSvgU2jSli8o3AqIEOTJr7rZwAEUVo4E4XhR94Zfr0=
+github.com/jackc/pgx/v5 v5.10.0/go.mod h1:mal1tBGAFfLHvZzaYh77YS/eC6IX9OWbRV1QIIM0Jn4=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+golang.org/x/text v0.29.0 h1:1neNs90w9YzJ9BocxfsQNHKuAT4pkghyXc4nhZ6sJvk=
+golang.org/x/text v0.29.0/go.mod h1:7MhJOA9CD2qZyOKYazxdYMF85OwPdEr9jTtBpO7ydH4=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
@@ -0,0 +1,179 @@
+// Copyright 2026 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 main
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/jackc/pgx/v5"
+)
+
+type ResFunc func(rows pgx.Rows) error
+
+type StmtTest struct {
+	Name  string
+	Query string
+	Args  []any
+	Res   []ResFunc
+}
+
+func main() {
+	user := os.Args[1]
+	port := os.Args[2]
+
+	ctx := context.Background()
+	connStr := fmt.Sprintf("postgres://%s:password@localhost:%s/postgres", user, port)
+	conn, err := pgx.Connect(ctx, connStr)
+	if err != nil {
+		panic(err)
+	}
+	defer conn.Close(ctx)
+
+	var pk int
+	err = conn.QueryRow(ctx, "SELECT pk FROM test_table LIMIT 1").Scan(&pk)
+	if err != nil {
+		panic(err)
+	}
+	if pk != 1 {
+		panic(fmt.Sprintf("expected pk=1, got %d", pk))
+	}
+
+	_, err = conn.Exec(ctx, "INSERT INTO test_table VALUES (2)")
+	if err != nil {
+		panic(err)
+	}
+
+	var count int
+	err = conn.QueryRow(ctx, "SELECT COUNT(*) FROM test_table").Scan(&count)
+	if err != nil {
+		panic(err)
+	}
+	if count != 2 {
+		panic(fmt.Sprintf("expected count=2, got %d", count))
+	}
+
+	_, err = conn.Prepare(ctx, "select_pk", "SELECT pk FROM test_table WHERE pk = $1")
+	if err != nil {
+		panic(err)
+	}
+	err = conn.QueryRow(ctx, "select_pk", 1).Scan(&pk)
+	if err != nil {
+		panic(err)
+	}
+	if pk != 1 {
+		panic(fmt.Sprintf("expected pk=1 from prepared stmt, got %d", pk))
+	}
+
+	// dolt workflow: create table, insert, commit, branch, insert, commit, merge
+	for _, q := range []string{
+		"DROP TABLE IF EXISTS test",
+		"CREATE TABLE test (pk int, value int, PRIMARY KEY(pk))",
+		"INSERT INTO test (pk, value) VALUES (0, 0)",
+		"SELECT dolt_add('-A')",
+		"SELECT dolt_commit('-m', 'added table test')",
+		"SELECT dolt_checkout('-b', 'mybranch')",
+		"INSERT INTO test VALUES (1, 1)",
+		"SELECT dolt_commit('-a', '-m', 'updated test')",
+		"SELECT dolt_checkout('main')",
+		"SELECT dolt_merge('mybranch')",
+	} {
+		if _, err = conn.Exec(ctx, q); err != nil {
+			panic(fmt.Sprintf("failed to execute %q: %v", q, err))
+		}
+	}
+
+	stmtTests := []StmtTest{
+		{
+			Name:  "select_test_by_pk",
+			Query: "SELECT pk, value FROM test WHERE pk = $1",
+			Args:  []any{int64(0)},
+			Res: []ResFunc{
+				func(rows pgx.Rows) error {
+					var pk, value int64
+					if err := rows.Scan(&pk, &value); err != nil {
+						return err
+					}
+					if pk != 0 || value != 0 {
+						return fmt.Errorf("expected pk=0 value=0, got pk=%d value=%d", pk, value)
+					}
+					return nil
+				},
+			},
+		},
+		{
+			Name:  "count_dolt_log",
+			Query: "SELECT COUNT(*) FROM dolt_log",
+			Res: []ResFunc{
+				func(rows pgx.Rows) error {
+					var size int64
+					if err := rows.Scan(&size); err != nil {
+						return err
+					}
+					if size != 4 {
+						return fmt.Errorf("expected 4 dolt_log entries, got %d", size)
+					}
+					return nil
+				},
+			},
+		},
+		{
+			Name:  "count_test",
+			Query: "SELECT COUNT(*) FROM test",
+			Res: []ResFunc{
+				func(rows pgx.Rows) error {
+					var size int64
+					if err := rows.Scan(&size); err != nil {
+						return err
+					}
+					if size != 2 {
+						return fmt.Errorf("expected 2 rows in test, got %d", size)
+					}
+					return nil
+				},
+			},
+		},
+	}
+
+	for _, test := range stmtTests {
+		func() {
+			if _, err := conn.Prepare(ctx, test.Name, test.Query); err != nil {
+				panic(fmt.Sprintf("prepare %q: %v", test.Query, err))
+			}
+			rows, err := conn.Query(ctx, test.Name, test.Args...)
+			if err != nil {
+				panic(fmt.Sprintf("query %q: %v", test.Query, err))
+			}
+			defer rows.Close()
+
+			i := 0
+			for rows.Next() {
+				if i >= len(test.Res) {
+					panic(fmt.Sprintf("too many rows for %q", test.Query))
+				}
+				if err := test.Res[i](rows); err != nil {
+					panic(fmt.Sprintf("result %d of %q: %v", i, test.Query, err))
+				}
+				i++
+			}
+			if err := rows.Err(); err != nil {
+				panic(fmt.Sprintf("rows.Err() for %q: %v", test.Query, err))
+			}
+		}()
+	}
+
+	fmt.Println("pgx test passed")
+}