One more time

spotify · Apr 3, 2021 · 13313da · 13313da
1 parent 62251b6
commit 13313da
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/exec/tests/unit/commands/test_run.py b/exec/tests/unit/commands/test_run.py
@@ -656,20 +656,17 @@ def test_get_pipeline_options(
 
 
 @pytest.mark.parametrize(
-    "direct_runner,run_error",
+    "direct_runner,streaming,run_error,exp_call_count",
     (
-        (True, None),
-        (False, None),
-        (True, ValueError("No running job found with name")),
+        (True, True, None, 2),
+        (False, True, None, 2),
+        (True, True, ValueError("No running job found with name"), 2),
+        (True, False, None, 1),
+        (False, False, None, 1),
+        (True, False, ValueError("No running job found with name"), 2),
     ),
 )
 @pytest.mark.parametrize("blocking", (True, False))
-@pytest.mark.parametrize("streaming,exp_call_count", 
-    (
-        (True, 2),
-        (False, 2),
-    ),
-)
 def test_run_pipeline(
     streaming,
     blocking,
@@ -734,7 +731,7 @@ def test_run_pipeline(
         mock_output.to_io_kwargs.return_value = {
             "topic": "projects/foo/topics/bar",
         }
-        config.job_config.events.inputs = [mock_input, mock_input]
+        config.job_config.events.inputs = [mock_input]
         config.job_config.events.outputs = [mock_output]
 
     if run_error: