QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-05-10 02:15:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

pylint and current_app related change in spec

Browse Source
This commit is contained in:
justinpark
2026-01-06 08:56:35 -08:00
parent 3a6bd39dd7
commit 3f45502bb3
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/unit_tests/db_engine_specs/test_trino.py
View File

@@ -386,7 +386,7 @@ def test_prepare_cancel_query(
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_early_cancel(
mock_app: Mock,
mock_db: Mock,
@@ -924,7 +924,7 @@ def test_timegrain_expressions(time_grain: str, expected_result: str) -> None:
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_progress_updates(
mock_app: Mock,
mock_db: Mock,
@@ -974,7 +974,7 @@ def test_handle_cursor_progress_updates(
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_cancels_on_stopped_status(
mock_app: Mock,
mock_db: Mock,
@@ -1009,7 +1009,7 @@ def test_handle_cursor_cancels_on_stopped_status(
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_cancels_on_timed_out_status(
mock_app: Mock,
mock_db: Mock,
@@ -1044,7 +1044,7 @@ def test_handle_cursor_cancels_on_timed_out_status(
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_breaks_on_execute_error(
mock_app: Mock,
mock_db: Mock,
@@ -1079,7 +1079,7 @@ def test_handle_cursor_breaks_on_execute_error(
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_breaks_on_execute_event_set(
mock_app: Mock,
mock_db: Mock,
@@ -1120,7 +1120,7 @@ def test_handle_cursor_breaks_on_execute_event_set(
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_handles_zero_total_splits(
mock_app: Mock,
mock_db: Mock,
@@ -1165,7 +1165,7 @@ def test_handle_cursor_handles_zero_total_splits(
@patch("superset.db_engine_specs.presto.PrestoBaseEngineSpec.handle_cursor")
@patch("superset.db_engine_specs.trino.TrinoEngineSpec.cancel_query")
@patch("superset.db_engine_specs.trino.db")
@patch("superset.db_engine_specs.trino.app")
@patch("superset.db_engine_specs.trino.current_app")
def test_handle_cursor_only_commits_on_progress_change(
mock_app: Mock,
mock_db: Mock,