QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-19 08:04:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Trino get_columns (#29566)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/integration_tests/db_engine_specs/presto_tests.py
View File

@@ -78,7 +78,10 @@ class TestPrestoDbEngineSpec(TestDbEngineSpec):
def verify_presto_column(self, column, expected_results):
inspector = mock.Mock()
inspector.engine.dialect.identifier_preparer.quote_identifier = mock.Mock()
preparer = inspector.engine.dialect.identifier_preparer
preparer.quote_identifier = preparer.quote = preparer.quote_schema = (
lambda x: f'"{x}"'
)
row = mock.Mock()
row.Column, row.Type, row.Null = column
inspector.bind.execute.return_value.fetchall = mock.Mock(return_value=[row])
@@ -798,7 +801,8 @@ class TestPrestoDbEngineSpec(TestDbEngineSpec):
def test_show_columns(self):
inspector = mock.MagicMock()
inspector.engine.dialect.identifier_preparer.quote_identifier = (
preparer = inspector.engine.dialect.identifier_preparer
preparer.quote_identifier = preparer.quote = preparer.quote_schema = (
lambda x: f'"{x}"'
)
inspector.bind.execute.return_value.fetchall = mock.MagicMock(
@@ -813,7 +817,8 @@ class TestPrestoDbEngineSpec(TestDbEngineSpec):
def test_show_columns_with_schema(self):
inspector = mock.MagicMock()
inspector.engine.dialect.identifier_preparer.quote_identifier = (
preparer = inspector.engine.dialect.identifier_preparer
preparer.quote_identifier = preparer.quote = preparer.quote_schema = (
lambda x: f'"{x}"'
)
inspector.bind.execute.return_value.fetchall = mock.MagicMock(