Function TestKeywords_AggregateFunctions

pkg/sql/keywords/keywords_test.go:634–665

TestKeywords_AggregateFunctions tests aggregate function keywords

(t *testing.T)

// TestKeywords_AggregateFunctions tests aggregate function keywords
func TestKeywords_AggregateFunctions(t *testing.T) {
	k := New(DialectGeneric, true)

	aggregateFunctions := []struct {
		word         string
		expectedType models.TokenType
	}{
		{"COUNT", models.TokenTypeCount},
		{"SUM", models.TokenTypeSum},
		{"AVG", models.TokenTypeAvg},
		{"MIN", models.TokenTypeMin},
		{"MAX", models.TokenTypeMax},
	}

	for _, af := range aggregateFunctions {
		// Test case variations
		testVariations := []string{af.word, strings.ToLower(af.word)}
		for _, word := range testVariations {
			if !k.IsKeyword(word) {
				t.Errorf("Aggregate function %q should be a keyword", word)
			}
			if !k.IsReserved(word) {
				t.Errorf("Aggregate function %q should be reserved", word)
			}

			tokenType := k.GetTokenType(word)
			if tokenType != af.expectedType {
				t.Errorf("GetTokenType(%q) = %v, want %v", word, tokenType, af.expectedType)
			}
		}
	}
}

// TestKeywords_CaseSensitiveMode tests case-sensitive keyword matching
func TestKeywords_CaseSensitiveMode(t *testing.T) {