fix: Use singlestoredb dialect for sqlglot (#36096)

This commit is contained in:
Vitor Avila
2025-11-13 16:10:55 -03:00
committed by GitHub
parent 4515d18ddd
commit 6701d0ae0c
2 changed files with 15 additions and 1 deletions

View File

@@ -2803,6 +2803,19 @@ def test_kqlstatement_is_select(kql: str, expected: bool) -> None:
assert KustoKQLStatement(kql, "kustokql").is_select() == expected
def test_singlestore_engine_mapping():
"""
Test the `singlestoredb` dialect is properly used.
"""
sql = "SELECT COUNT(*) AS `COUNT(*)`"
statement = SQLStatement(sql, engine="singlestoredb")
assert statement.is_select()
# Should parse without errors
formatted = statement.format()
assert "COUNT(*)" in formatted
def test_remove_quotes() -> None:
"""
Test the `remove_quotes` helper function.