Function TestQuerySourceMixedBatchSetThenShow

server/query_source_test.go:131–177

TestQuerySourceMixedBatchSetThenShow is the regression for the e2e bug: a single simple query batching `SET duckgres.query_source='endpoints'; SHOW duckgres.query_source` must run BOTH statements in order — the SET (session-side) then the SHOW returning the just-set value. Previously the whole-batch

(t *testing.T)

		ctx:      context.Background(),
		cancel:   func() {},
		txStatus: txStatusIdle,
		executor: exec,
	}
	return c, &out
}

// selectOneExecutor answers any QueryContext with a single-row RowSet, so a
// SELECT in a batch alongside the query_source GUC actually executes.
type selectOneExecutor struct {
	noopProfiling
	queryCalls int
}

func (e *selectOneExecutor) QueryContext(context.Context, string, ...any) (RowSet, error) {
	e.queryCalls++
	return &staticCountRowSet{count: 1}, nil
}
func (e *selectOneExecutor) ExecContext(context.Context, string, ...any) (ExecResult, error) {
	return &fakeExecResult{}, nil
}
func (e *selectOneExecutor) Query(string, ...any) (RowSet, error) {
	return nil, errors.New("not implemented")
}
func (e *selectOneExecutor) Exec(string, ...any) (ExecResult, error) {
	return nil, errors.New("not implemented")
}
func (e *selectOneExecutor) ConnContext(context.Context) (RawConn, error) {
	return nil, errors.New("not implemented")
}
func (e *selectOneExecutor) PingContext(context.Context) error { return nil }
func (e *selectOneExecutor) Close() error                      { return nil }

// TestQuerySourceMixedBatchSetThenShow is the regression for the e2e bug: a
// single simple query batching `SET duckgres.query_source='endpoints'; SHOW
// duckgres.query_source` must run BOTH statements in order — the SET
// (session-side) then the SHOW returning the just-set value. Previously the
// whole-batch transpile surfaced QuerySourceSet on the first statement and
// returned early, swallowing the trailing SHOW so it never emitted `endpoints`.
func TestQuerySourceMixedBatchSetThenShow(t *testing.T) {
	c, out := newBufferedConn(&selectOneExecutor{})

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

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

	// Expect: CommandComplete(SET), RowDescription, DataRow(endpoints),
	// CommandComplete(SHOW), ReadyForQuery — in that order.
	var sawSetComplete, sawShowRow, sawShowComplete bool