From bfc4dc89f2a1579a7de0d47bb04e1e3e7dc64fbc Mon Sep 17 00:00:00 2001 From: Michael Hoang Date: Tue, 29 Apr 2025 12:11:58 +1000 Subject: [PATCH] container-test-driver: add `Machine.fail` --- lib/test/container-test-driver/test_driver/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/lib/test/container-test-driver/test_driver/__init__.py b/lib/test/container-test-driver/test_driver/__init__.py index ea0737c5d..42c26979b 100644 --- a/lib/test/container-test-driver/test_driver/__init__.py +++ b/lib/test/container-test-driver/test_driver/__init__.py @@ -333,6 +333,15 @@ class Machine: raise RuntimeError(msg) return res.stdout + def fail(self, command: str, timeout: int | None = None) -> str: + res = self.execute(command, timeout=timeout) + if res.returncode == 0: + msg = f"command `{command}` unexpectedly succeeded\n" + msg += f"Exit code: {res.returncode}\n" + msg += f"Stdout: {res.stdout}" + raise RuntimeError(msg) + return res.stdout + def shutdown(self) -> None: """ Shut down the machine, waiting for the VM to exit.