QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-07-19 13:15:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: replace Engine.execute() with Connection.execute() (#41917)

Browse Source
This commit is contained in:
Hans Yu
2026-07-13 18:40:33 +02:00
committed by GitHub
parent 8f75f1a353
commit 68ebc22e1e
32 changed files with 342 additions and 266 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/integration_tests/datasource_tests.py
View File

@@ -63,27 +63,29 @@ def create_test_table_context(database: Database):
full_table_name = f"{schema}.test_table" if schema else "test_table"
with database.get_sqla_engine() as engine:
engine.execute(
text(f"""
CREATE TABLE IF NOT EXISTS {full_table_name} AS
SELECT 1 as first, 2 as second
""")
)
engine.execute(
text(f"""
INSERT INTO {full_table_name} (first, second) VALUES (1, 2)
""") # noqa: S608
)
engine.execute(
text(f"""
INSERT INTO {full_table_name} (first, second) VALUES (3, 4)
""") # noqa: S608
)
with engine.begin() as conn:
conn.execute(
text(f"""
CREATE TABLE IF NOT EXISTS {full_table_name} AS
SELECT 1 as first, 2 as second
""")
)
conn.execute(
text(f"""
INSERT INTO {full_table_name} (first, second) VALUES (1, 2)
""") # noqa: S608
)
conn.execute(
text(f"""
INSERT INTO {full_table_name} (first, second) VALUES (3, 4)
""") # noqa: S608
)
yield db.session
with database.get_sqla_engine() as engine:
engine.execute(text(f"DROP TABLE {full_table_name}"))
with engine.begin() as conn:
conn.execute(text(f"DROP TABLE {full_table_name}"))
@contextmanager