| 12 | ) |
| 13 | |
| 14 | func TestFormatSQLWithDialect_PostgreSQL(t *testing.T) { |
| 15 | tree, err := ParseSQL("SELECT * FROM users u") |
| 16 | if err != nil { |
| 17 | t.Fatalf("parse: %v", err) |
| 18 | } |
| 19 | stmt := tree.Statements[0] |
| 20 | if err := Apply(stmt, SetLimit(100)); err != nil { |
| 21 | t.Fatalf("apply: %v", err) |
| 22 | } |
| 23 | |
| 24 | got := FormatSQLWithDialect(stmt, keywords.DialectPostgreSQL) |
| 25 | if !strings.Contains(got, "LIMIT 100") { |
| 26 | t.Errorf("postgresql: expected LIMIT 100, got: %s", got) |
| 27 | } |
| 28 | } |
| 29 | |
| 30 | func TestFormatSQLWithDialect_SQLServer(t *testing.T) { |
| 31 | tree, err := ParseSQL("SELECT * FROM users u") |