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

fix: search_path in RDS (#24739)

Browse Source
This commit is contained in:
Beto Dealmeida
2023-07-20 12:57:48 -07:00
committed by GitHub
parent b2831b419e
commit 7675e0db10
5 changed files with 185 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/unit_tests/models/core_test.py
View File

@@ -212,3 +212,21 @@ def test_dttm_sql_literal(
def test_table_column_database() -> None:
database = Database(database_name="db")
assert TableColumn(database=database).database is database
def test_get_prequeries(mocker: MockFixture) -> None:
"""
Tests for ``get_prequeries``.
"""
mocker.patch.object(
Database,
"get_sqla_engine_with_context",
)
db_engine_spec = mocker.patch.object(Database, "db_engine_spec")
db_engine_spec.get_prequeries.return_value = ["set a=1", "set b=2"]
database = Database(database_name="db")
with database.get_raw_connection() as conn:
conn.cursor().execute.assert_has_calls(
[mocker.call("set a=1"), mocker.call("set b=2")]
)