fix tests

ModelSurge · Aug 18, 2023 · c304418 · c304418
1 parent ca5d2f8
commit c304418
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/lib_comfyui_tests/external_code_tests/run_workflow_test.py b/tests/lib_comfyui_tests/external_code_tests/run_workflow_test.py
@@ -28,6 +28,7 @@ def test_valid_workflow_with_dict_types(self, mock_start_workflow):
         mock_start_workflow.assert_called_once_with(
             batch_input_args=("value", 123),
             workflow_type_id="test_tab",
+            workflow_input_types=('IMAGE', 'LATENT'),
             queue_front=True,
         )
         self.assertEqual(result, mock_start_workflow.return_value)
@@ -48,6 +49,7 @@ def test_valid_workflow_with_tuple_types(self, mock_start_workflow):
         mock_start_workflow.assert_called_once_with(
             batch_input_args=("value", 123),
             workflow_type_id="test_tab",
+            workflow_input_types=('IMAGE', 'LATENT'),
             queue_front=True,
         )
         self.assertEqual(result, [tuple(batch.values()) for batch in mock_start_workflow.return_value])
@@ -68,6 +70,7 @@ def test_valid_workflow_with_str_types(self, mock_start_workflow):
         mock_start_workflow.assert_called_once_with(
             batch_input_args=("value",),
             workflow_type_id="test_tab",
+            workflow_input_types=('IMAGE',),
             queue_front=True,
         )
         self.assertEqual(result, [next(iter(batch.values())) for batch in mock_start_workflow.return_value])
@@ -179,6 +182,7 @@ def test_identity_on_error(self, mock_start_workflow):
         mock_start_workflow.assert_called_with(
             batch_input_args=("value",),
             workflow_type_id="test_tab",
+            workflow_input_types=('IMAGE',),
             queue_front=True,
         )