update failing test

apache · Dec 27, 2024 · 6d73105 · 6d73105
1 parent 9981466
commit 6d73105
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/api_fastapi/execution_api/routes/test_task_instances.py b/tests/api_fastapi/execution_api/routes/test_task_instances.py
@@ -234,7 +234,7 @@ def test_ti_update_state_database_error(self, client, session, create_task_insta
         with mock.patch(
             "airflow.api_fastapi.common.db.common.Session.execute",
             side_effect=[
-                mock.Mock(one=lambda: ("running",)),  # First call returns "queued"
+                mock.Mock(one=lambda: ("running", 1, 0)),  # First call returns "queued"
                 SQLAlchemyError("Database error"),  # Second call raises an error
             ],
         ):