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

feat: catalog support for Databricks native (#28394)

Browse Source
This commit is contained in:
Beto Dealmeida
2024-05-09 17:41:15 -04:00
committed by GitHub
parent e516bba8fc
commit f29e1e4c29
10 changed files with 443 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/unit_tests/db_engine_specs/test_databricks.py
View File

@@ -245,3 +245,22 @@ def test_convert_dttm(
from superset.db_engine_specs.databricks import DatabricksNativeEngineSpec as spec
assert_convert_dttm(spec, target_type, expected_result, dttm)
def test_get_prequeries() -> None:
"""
Test the ``get_prequeries`` method.
"""
from superset.db_engine_specs.databricks import DatabricksNativeEngineSpec
assert DatabricksNativeEngineSpec.get_prequeries() == []
assert DatabricksNativeEngineSpec.get_prequeries(schema="test") == [
"USE SCHEMA test",
]
assert DatabricksNativeEngineSpec.get_prequeries(catalog="test") == [
"USE CATALOG test",
]
assert DatabricksNativeEngineSpec.get_prequeries(catalog="foo", schema="bar") == [
"USE CATALOG foo",
"USE SCHEMA bar",
]