Function TestQuerySourceGetter

server/query_source_test.go:15–42

TestQuerySourceGetter covers the session accessor a future compute meter reads: unset defaults to "standard", a SET value round-trips, and reset to empty falls back to the default.

(t *testing.T)

// TestQuerySourceGetter covers the session accessor the compute meter reads:
// unset defaults to "standard", a SET value round-trips, and reset to empty
// falls back to the default. Callers of setQuerySource pass already-validated
// canonical values (the transpiler / startup-option validation reject anything
// outside {standard, endpoints} with 22023 before the setter is reached).
func TestQuerySourceGetter(t *testing.T) {
	c := &clientConn{}

	// Unset → default "standard" (never an error for a missing value).
	if got := c.QuerySource(); got != "standard" {
		t.Fatalf("unset QuerySource() = %q, want %q", got, "standard")
	}
	if got := c.QuerySource(); got != defaultQuerySource {
		t.Fatalf("unset QuerySource() = %q, want defaultQuerySource %q", got, defaultQuerySource)
	}

	// SET → value round-trips.
	c.setQuerySource("endpoints")
	if got := c.QuerySource(); got != "endpoints" {
		t.Fatalf("after set, QuerySource() = %q, want %q", got, "endpoints")
	}

	// RESET / empty → back to default.
	c.setQuerySource("")
	if got := c.QuerySource(); got != "standard" {
		t.Fatalf("after reset, QuerySource() = %q, want %q", got, "standard")
	}
}

// TestQuerySourceStartupOption asserts a `-c duckgres.query_source=...` startup
// option is parsed into the map ParseStartupOptions returns and applied to the
// session by applyStartupQuerySource, with the same validation as SET: