Function TestCompletion

backend/plugin/parser/trino/completion_test.go:41–140

TestCompletion exercises the omni-backed Trino completer through the public Completion entry point. DIVERGENCE FROM THE LEGACY ANTLR COMPLETER (intentional, documented): The previous Completion was a ~1730-line CodeCompletionCore (c3) port. The omni Trino completer is explicitly NOT a c3 port (see

(t *testing.T)

// These cases therefore assert the candidate *set* (Text, Type) the omni
// completer actually returns, rather than the legacy YAML fixture.
func TestCompletion(t *testing.T) {
	type want struct {
		text string
		typ  base.CandidateType
	}
	tests := []struct {
		description string
		input       string // caret marked by "|"
		// wantPresent candidates that MUST appear.
		wantPresent []want
		// wantAbsentText candidate texts that must NOT appear (any type).
		wantAbsentText []string
	}{
		{
			description: "FROM offers catalogs, schemas, and current-schema tables",
			input:       "SELECT * FROM |",
			wantPresent: []want{
				{"company", base.CandidateTypeDatabase},
				{"school", base.CandidateTypeDatabase},
				{"dbo", base.CandidateTypeSchema},
				{"myschema", base.CandidateTypeSchema},
				{"address", base.CandidateTypeTable},
				{"employees", base.CandidateTypeTable},
			},
		},
		{
			description: "Schema-qualified FROM offers that schema's tables",
			input:       "SELECT * FROM dbo.|",
			wantPresent: []want{
				{"address", base.CandidateTypeTable},
				{"employees", base.CandidateTypeTable},
			},
			wantAbsentText: []string{"salarylevel"},
		},
		{
			description: "Schema-qualified FROM for a non-current schema",
			input:       "SELECT * FROM myschema.|",
			wantPresent: []want{
				{"salarylevel", base.CandidateTypeTable},
			},
		},
		{
			description: "SELECT column context offers in-scope columns",
			input:       "SELECT | FROM Employees",
			wantPresent: []want{
				{"id", base.CandidateTypeColumn},
				{"name", base.CandidateTypeColumn},
			},
		},
		{
			description: "WHERE column context offers in-scope columns",
			input:       "SELECT * FROM Employees WHERE |",
			wantPresent: []want{
				{"id", base.CandidateTypeColumn},
				{"name", base.CandidateTypeColumn},
			},
		},
		{