Function TestSQLFormatter_DDLStatements

cmd/gosqlx/cmd/sql_formatter_test.go:354–415

TestSQLFormatter_DDLStatements tests DDL statement formatting

(t *testing.T)

// TestSQLFormatter_DDLStatements tests DDL statement formatting
func TestSQLFormatter_DDLStatements(t *testing.T) {
	tests := []struct {
		name        string
		sql         string
		expectWords []string
	}{
		{
			name:        "CREATE TABLE",
			sql:         "CREATE TABLE users (id INT PRIMARY KEY, name VARCHAR(100))",
			expectWords: []string{"create", "table"},
		},
		{
			name:        "ALTER TABLE",
			sql:         "ALTER TABLE users ADD COLUMN email VARCHAR(255)",
			expectWords: []string{"alter", "table"},
		},
		{
			name:        "DROP TABLE",
			sql:         "DROP TABLE temp_data",
			expectWords: []string{"drop", "table"},
		},
	}

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