Function TestProtocolExtendedQuery

tests/integration/protocol_test.go:156–249

TestProtocolExtendedQuery tests the extended query protocol (prepared statements)

(t *testing.T)

// TestProtocolExtendedQuery tests the extended query protocol (prepared statements)
func TestProtocolExtendedQuery(t *testing.T) {
	t.Run("prepare_and_query", func(t *testing.T) {
		stmt, err := testHarness.DuckgresDB.Prepare("SELECT $1::int + $2::int")
		if err != nil {
			t.Fatalf("Prepare failed: %v", err)
		}
		defer func() { _ = stmt.Close() }()

		var result int
		err = stmt.QueryRow(10, 20).Scan(&result)
		if err != nil {
			t.Fatalf("QueryRow failed: %v", err)
		}
		if result != 30 {
			t.Errorf("Expected 30, got %d", result)
		}
	})

	t.Run("prepare_with_types", func(t *testing.T) {
		skipIfKnown(t)
		stmt, err := testHarness.DuckgresDB.Prepare("SELECT $1::text || ' ' || $2::text")
		if err != nil {
			t.Fatalf("Prepare failed: %v", err)
		}
		defer func() { _ = stmt.Close() }()

		var result string
		err = stmt.QueryRow("hello", "world").Scan(&result)
		if err != nil {
			t.Fatalf("QueryRow failed: %v", err)
		}
		if result != "hello world" {
			t.Errorf("Expected 'hello world', got %q", result)
		}
	})

	t.Run("prepare_select_where", func(t *testing.T) {
		stmt, err := testHarness.DuckgresDB.Prepare("SELECT name FROM users WHERE id = $1")
		if err != nil {
			t.Fatalf("Prepare failed: %v", err)
		}
		defer func() { _ = stmt.Close() }()

		var name string
		err = stmt.QueryRow(1).Scan(&name)
		if err != nil {
			t.Fatalf("QueryRow failed: %v", err)
		}
		if name != "Alice" {
			t.Errorf("Expected 'Alice', got %q", name)
		}
	})

	t.Run("prepare_reuse", func(t *testing.T) {
		stmt, err := testHarness.DuckgresDB.Prepare("SELECT $1::int * 2")
		if err != nil {
			t.Fatalf("Prepare failed: %v", err)
		}