From 13313dab133612c45f5ef3dcd43c3e9f3cd4a897 Mon Sep 17 00:00:00 2001
From: Paolo Genissel <paologenissel@gmail.com>
Date: Sat, 3 Apr 2021 13:32:16 +0200
Subject: [PATCH] One more time

---
 exec/tests/unit/commands/test_run.py | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/exec/tests/unit/commands/test_run.py b/exec/tests/unit/commands/test_run.py
index bb511a7d..afdbf8d6 100644
--- 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: