Change tests assertions to reflect changes in error handling

Author: PawelPeczek-Roboflow

tests/workflows/integration_tests/execution/test_workflow_with_detections_consensus_block.py

@@ -5,7 +5,7 @@
 from inference.core.env import WORKFLOWS_MAX_CONCURRENT_STEPS
 from inference.core.managers.base import ModelManager
 from inference.core.workflows.core_steps.common.entities import StepExecutionMode
-from inference.core.workflows.errors import RuntimeInputError, StepExecutionError
+from inference.core.workflows.errors import RuntimeInputError, StepExecutionError, ClientCausedStepExecutionError
 from inference.core.workflows.execution_engine.core import ExecutionEngine
 from tests.workflows.integration_tests.execution.workflows_gallery_collector.decorators import (
     add_to_workflows_gallery,
@@ -273,7 +273,7 @@ def test_consensus_workflow_when_model_id_cannot_be_resolved_to_valid_model(
     )
 
     # when
-    with pytest.raises(StepExecutionError):
+    with pytest.raises(ClientCausedStepExecutionError):
         _ = execution_engine.run(
             runtime_parameters={
                 "image": crowd_image,

tests/workflows/integration_tests/execution/test_workflow_with_filtering.py

@@ -8,7 +8,7 @@
 from inference.core.workflows.core_steps.common.query_language.errors import (
     EvaluationEngineError,
 )
-from inference.core.workflows.errors import RuntimeInputError, StepExecutionError
+from inference.core.workflows.errors import RuntimeInputError, StepExecutionError, ClientCausedStepExecutionError
 from inference.core.workflows.execution_engine.core import ExecutionEngine
 from tests.workflows.integration_tests.execution.workflows_gallery_collector.decorators import (
     add_to_workflows_gallery,
@@ -310,7 +310,7 @@ def test_filtering_workflow_when_model_id_cannot_be_resolved_to_valid_model(
     )
 
     # when
-    with pytest.raises(StepExecutionError):
+    with pytest.raises(ClientCausedStepExecutionError):
         _ = execution_engine.run(
             runtime_parameters={
                 "image": crowd_image,

tests/workflows/integration_tests/execution/test_workflow_with_single_model.py

@@ -5,7 +5,7 @@
 from inference.core.env import WORKFLOWS_MAX_CONCURRENT_STEPS
 from inference.core.managers.base import ModelManager
 from inference.core.workflows.core_steps.common.entities import StepExecutionMode
-from inference.core.workflows.errors import RuntimeInputError, StepExecutionError
+from inference.core.workflows.errors import RuntimeInputError, StepExecutionError, ClientCausedStepExecutionError
 from inference.core.workflows.execution_engine.core import ExecutionEngine
 from tests.workflows.integration_tests.execution.workflows_gallery_collector.decorators import (
     add_to_workflows_gallery,
@@ -332,7 +332,7 @@ def test_object_detection_workflow_when_model_id_cannot_be_resolved_to_valid_mod
     )
 
     # when
-    with pytest.raises(StepExecutionError):
+    with pytest.raises(ClientCausedStepExecutionError):
         _ = execution_engine.run(
             runtime_parameters={
                 "image": crowd_image,