fix: Trino get_columns (#29566)

This commit is contained in:
Beto Dealmeida
2024-07-12 16:37:49 -04:00
committed by GitHub
parent 0d352b4e06
commit fa095a98ed
7 changed files with 331 additions and 226 deletions

View File

@@ -311,3 +311,26 @@ def test_get_default_catalog(mocker: MockerFixture) -> None:
database = mocker.MagicMock()
assert BaseEngineSpec.get_default_catalog(database) is None
def test_quote_table() -> None:
"""
Test the `quote_table` function.
"""
from superset.db_engine_specs.base import BaseEngineSpec
dialect = sqlite.dialect()
assert BaseEngineSpec.quote_table(Table("table"), dialect) == '"table"'
assert (
BaseEngineSpec.quote_table(Table("table", "schema"), dialect)
== 'schema."table"'
)
assert (
BaseEngineSpec.quote_table(Table("table", "schema", "catalog"), dialect)
== 'catalog.schema."table"'
)
assert (
BaseEngineSpec.quote_table(Table("ta ble", "sche.ma", 'cata"log'), dialect)
== '"cata""log"."sche.ma"."ta ble"'
)