Function TestSQLFormatter_WithClauses

cmd/gosqlx/cmd/sql_formatter_test.go:177–233

TestSQLFormatter_WithClauses tests CTE (WITH clause) formatting

(t *testing.T)

// TestSQLFormatter_WithClauses tests CTE (WITH clause) formatting
func TestSQLFormatter_WithClauses(t *testing.T) {
	tests := []struct {
		name        string
		sql         string
		expectWords []string
	}{
		{
			name:        "simple WITH clause",
			sql:         "WITH temp AS (SELECT id FROM users) SELECT * FROM temp",
			expectWords: []string{"with", "as", "select"},
		},
		{
			name:        "recursive CTE",
			sql:         "WITH RECURSIVE cte AS (SELECT 1 as n UNION ALL SELECT n + 1 FROM cte WHERE n < 10) SELECT * FROM cte",
			expectWords: []string{"with", "recursive", "as", "union"},
		},
	}

	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)
				}
			}
		})
	}
}