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

chore: improve schema security (#23385)

Browse Source
This commit is contained in:
Beto Dealmeida
2023-03-17 08:05:50 -07:00
committed by GitHub
parent d4657f6198
commit 42e8d1b498
17 changed files with 346 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/unit_tests/db_engine_specs/test_drill.py
View File

@@ -20,6 +20,7 @@ from datetime import datetime
from typing import Optional
import pytest
from sqlalchemy.engine.url import make_url
from tests.unit_tests.db_engine_specs.utils import assert_convert_dttm
from tests.unit_tests.fixtures.common import dttm
@@ -106,3 +107,18 @@ def test_convert_dttm(
from superset.db_engine_specs.drill import DrillEngineSpec as spec
assert_convert_dttm(spec, target_type, expected_result, dttm)
def test_get_schema_from_engine_params() -> None:
"""
Test ``get_schema_from_engine_params``.
"""
from superset.db_engine_specs.drill import DrillEngineSpec
assert (
DrillEngineSpec.get_schema_from_engine_params(
make_url("drill+sadrill://localhost:8047/dfs/test?use_ssl=False"),
{},
)
== "dfs.test"
)