Function TestSQLFormatter_JOINStatements

cmd/gosqlx/cmd/sql_formatter_test.go:113–174

TestSQLFormatter_JOINStatements tests JOIN formatting

(t *testing.T)

// TestSQLFormatter_JOINStatements tests JOIN formatting
func TestSQLFormatter_JOINStatements(t *testing.T) {
	tests := []struct {
		name        string
		sql         string
		expectWords []string
	}{
		{
			name:        "INNER JOIN",
			sql:         "SELECT u.name, o.total FROM users u INNER JOIN orders o ON u.id = o.user_id",
			expectWords: []string{"select", "from", "inner join", "on"},
		},
		{
			name:        "LEFT JOIN",
			sql:         "SELECT u.name, o.total FROM users u LEFT JOIN orders o ON u.id = o.user_id",
			expectWords: []string{"select", "from", "left join", "on"},
		},
		{
			name:        "Multiple JOINs",
			sql:         "SELECT u.name FROM users u INNER JOIN orders o ON u.id = o.user_id LEFT JOIN products p ON o.product_id = p.id",
			expectWords: []string{"select", "from", "inner join", "left join"},
		},
	}

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