Function describeMsgs

server/query_source_test.go:218–233

(msgs []wireMsg)

// TestQuerySourceMixedBatchWithNormalStatement asserts a batch mixing the
// custom GUC with a normal statement (`SET duckgres.query_source='endpoints';
// SELECT 1`) still executes the SELECT — the GUC statement must not swallow the
// rest of the batch.
func TestQuerySourceMixedBatchWithNormalStatement(t *testing.T) {
	exec := &selectOneExecutor{}
	c, out := newBufferedConn(exec)

	if err := c.handleQuery([]byte("SET duckgres.query_source = 'endpoints'; SELECT 1\x00")); err != nil {
		t.Fatalf("handleQuery: %v", err)
	}

	if exec.queryCalls != 1 {
		t.Fatalf("SELECT did not run: QueryContext called %d times, want 1", exec.queryCalls)
	}
	if got := c.QuerySource(); got != "endpoints" {
		t.Fatalf("session QuerySource() = %q, want %q", got, "endpoints")
	}

	msgs := parseWireMsgs(t, out.Bytes())