mirror of
https://github.com/apache/superset.git
synced 2026-04-17 07:05:04 +00:00
feat(db_engine): Implement user impersonation support for StarRocks (#28110)
This commit is contained in:
@@ -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"'
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user