feat(db_engine): Implement user impersonation support for StarRocks (#28110)

This commit is contained in:
Patrick Schmidt
2024-09-06 18:13:38 +02:00
committed by GitHub
parent d3f5c795ff
commit 6294e339e2
11 changed files with 120 additions and 13 deletions

View File

@@ -137,14 +137,16 @@ def test_get_schema_from_engine_params() -> None:
)
def test_get_prequeries() -> None:
def test_get_prequeries(mocker: MockerFixture) -> None:
"""
Test the ``get_prequeries`` method.
"""
from superset.db_engine_specs.postgres import PostgresEngineSpec
assert PostgresEngineSpec.get_prequeries() == []
assert PostgresEngineSpec.get_prequeries(schema="test") == [
database = mocker.MagicMock()
assert PostgresEngineSpec.get_prequeries(database) == []
assert PostgresEngineSpec.get_prequeries(database, schema="test") == [
'set search_path = "test"'
]