add waiting status to rerun test

pepkit · Mar 13, 2024 · 7a52dfd · 7a52dfd
1 parent e0cd85d
commit 7a52dfd
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/smoketests/test_other.py b/tests/smoketests/test_other.py
@@ -62,11 +62,14 @@ def test_pipestat_configured(self, prep_temp_pep_pipestat, cmd):
 
 
 class TestLooperRerun:
+    @pytest.mark.parametrize(
+        "flags", [FLAGS[2], FLAGS[3]]
+    )  # Waiting and Failed flags should work
     @pytest.mark.parametrize("pipeline_name", ["example_pipestat_pipeline"])
-    def test_pipestat_rerun(self, prep_temp_pep_pipestat, pipeline_name):
-        """Verify that checking works when multiple flags are created"""
+    def test_pipestat_rerun(self, prep_temp_pep_pipestat, pipeline_name, flags):
+        """Verify that rerun works with either failed or waiting flags"""
         tp = prep_temp_pep_pipestat
-        _make_flags(tp, FLAGS[2], pipeline_name)
+        _make_flags(tp, flags, pipeline_name)
 
         x = ["rerun", "--looper-config", tp]
         try: