add @ok_to_fail decorator

redpanda-data · Sep 8, 2023 · 8a6567d · 8a6567d
1 parent 62e0285
commit 8a6567d
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/ducktape/tests/runner_client.py b/ducktape/tests/runner_client.py
@@ -131,7 +131,27 @@ def run(self):
                 if num_runs != self.deflake_num:
                     msg += "\n" + "~" * max(len(line) for line in summary.split('\n'))
 
-                self.log(logging.INFO, msg)
+            data = self.run_test()
+
+            if self.test_context.ok_to_fail:
+                test_status = OPASS
+                self.log(logging.INFO, "OPASS")
+            else:
+                test_status = PASS
+                self.log(logging.INFO, "PASS")
+
+        except BaseException as e:
+            if self.test_context.ok_to_fail:
+                test_status = OFAIL
+                err_trace = self._exc_msg(e)
+                summary += err_trace
+                self.log(logging.INFO, "OFAIL: " + err_trace)
+            else:
+                # mark the test as failed before doing anything else
+                test_status = FAIL
+                err_trace = self._exc_msg(e)
+                summary += err_trace
+                self.log(logging.INFO, "FAIL: " + err_trace)
 
         finally:
             stop_time = time.time()