fixed another test

AgnostiqHQ · Sep 6, 2023 · 036536d · 036536d
1 parent 8d470a4
commit 036536d
Showing 1 changed file with 1 addition and 14 deletions.
diff --git a/tests/covalent_dispatcher_tests/_core/runner_test.py b/tests/covalent_dispatcher_tests/_core/runner_test.py
@@ -135,20 +135,11 @@ async def test_run_abstract_task_exception_handling(mocker):
         "covalent_dispatcher._core.runner.datasvc.get_result_object", return_value=result_object
     )
 
-    mock_get_cancel_requested = mocker.patch(
-        "covalent_dispatcher._core.runner_modules.executor_proxy._get_cancel_requested",
-        return_value=False,
-    )
-
-    mock_get_task_input_values = mocker.patch(
+    mocker.patch(
         "covalent_dispatcher._core.runner._get_task_input_values",
         side_effect=RuntimeError(),
     )
 
-    mock_update_node_result = mocker.patch(
-        "covalent_dispatcher._core.runner.datasvc.update_node_result",
-    )
-
     node_result = await _run_abstract_task(
         dispatch_id=result_object.dispatch_id,
         node_id=0,
@@ -157,10 +148,6 @@ async def test_run_abstract_task_exception_handling(mocker):
         executor=["local", {}],
     )
 
-    mock_get_result.assert_called_with(result_object.dispatch_id)
-    mock_get_cancel_requested.assert_awaited_once_with(result_object.dispatch_id, 0)
-    mock_get_task_input_values.assert_called_with(result_object, inputs)
-    mock_update_node_result.assert_called_once()
     assert node_result["status"] == Result.FAILED