Function TestSQLFormatter_WindowFunctions

cmd/gosqlx/cmd/sql_formatter_test.go:236–297

TestSQLFormatter_WindowFunctions tests window function formatting

(t *testing.T)

// TestSQLFormatter_WindowFunctions tests window function formatting
func TestSQLFormatter_WindowFunctions(t *testing.T) {
	tests := []struct {
		name        string
		sql         string
		expectWords []string
	}{
		{
			name:        "ROW_NUMBER with OVER",
			sql:         "SELECT name, ROW_NUMBER() OVER (ORDER BY salary DESC) as rank FROM employees",
			expectWords: []string{"select", "over", "order by"},
		},
		{
			name:        "RANK with PARTITION BY",
			sql:         "SELECT dept, name, RANK() OVER (PARTITION BY dept ORDER BY salary) FROM employees",
			expectWords: []string{"select", "over", "partition by", "order by"},
		},
		{
			name:        "Window frame",
			sql:         "SELECT date, SUM(amount) OVER (ORDER BY date ROWS BETWEEN 2 PRECEDING AND CURRENT ROW) FROM transactions",
			expectWords: []string{"select", "over", "order by", "rows", "between"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tkz := tokenizer.GetTokenizer()
			defer tokenizer.PutTokenizer(tkz)

			tokens, err := tkz.Tokenize([]byte(tt.sql))
			if err != nil {
				t.Fatalf("Tokenization failed: %v", err)
			}

			p := parser.NewParser()
			astObj := ast.NewAST()
			defer ast.ReleaseAST(astObj)

			result, err := p.ParseFromModelTokens(tokens)
			if err != nil {
				t.Skipf("Parsing failed (may not be supported yet): %v", err)
				return
			}
			astObj.Statements = result.Statements

			formatter := NewSQLFormatter(FormatterOptions{
				Indent:      "  ",
				UppercaseKw: false,
			})

			output, err := formatter.Format(astObj)
			if err != nil {
				t.Fatalf("Formatting failed: %v", err)
			}

			for _, word := range tt.expectWords {
				if !strings.Contains(strings.ToLower(output), word) {
					t.Errorf("Expected '%s' in output:\n%s", word, output)
				}