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

chore(Databricks): New Databricks driver (#28393)

Browse Source
This commit is contained in:
Vitor Avila
2024-05-09 15:58:03 -03:00
committed by GitHub
parent e6a85c5901
commit 307ebeaa19
6 changed files with 333 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -35,13 +35,13 @@ def test_get_parameters_from_uri() -> None:
"""
from superset.db_engine_specs.databricks import (
DatabricksNativeEngineSpec,
DatabricksParametersType,
DatabricksNativeParametersType,
)
parameters = DatabricksNativeEngineSpec.get_parameters_from_uri(
"databricks+connector://token:abc12345@my_hostname:1234/test"
)
assert parameters == DatabricksParametersType(
assert parameters == DatabricksNativeParametersType(
{
"access_token": "abc12345",
"host": "my_hostname",
@@ -60,10 +60,10 @@ def test_build_sqlalchemy_uri() -> None:
"""
from superset.db_engine_specs.databricks import (
DatabricksNativeEngineSpec,
DatabricksParametersType,
DatabricksNativeParametersType,
)
parameters = DatabricksParametersType(
parameters = DatabricksNativeParametersType(
{
"access_token": "abc12345",
"host": "my_hostname",