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

@@ -66,13 +66,15 @@ def test_get_table_comment_empty(mocker: MockerFixture):
)
def test_get_prequeries() -> None:
def test_get_prequeries(mocker: MockerFixture) -> None:
"""
Test the ``get_prequeries`` method.
"""
from superset.db_engine_specs.db2 import Db2EngineSpec
assert Db2EngineSpec.get_prequeries() == []
assert Db2EngineSpec.get_prequeries(schema="my_schema") == [
database = mocker.MagicMock()
assert Db2EngineSpec.get_prequeries(database) == []
assert Db2EngineSpec.get_prequeries(database, schema="my_schema") == [
'set current_schema "my_schema"'
]