trunner: add missing expect during flash procedure

It is essential to include an assertion after performing a reboot to verify that PSH has loaded correctly and to avoid excessive output during the subsequent test case.
phoenix-rtos · Jan 13, 2025 · b1a6bc8 · b1a6bc8
1 parent 1c676a4
commit b1a6bc8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/trunner/harness/plo.py b/trunner/harness/plo.py
@@ -285,6 +285,11 @@ def __call__(self):
 
         self.rebooter(flash=False, hard=True)
 
+        try:
+            self.dut.expect_exact(f"(psh)%", timeout=10)
+        except pexpect.TIMEOUT as e:
+            raise PloError("Error: missing psh prompt after flash!", output=self.dut.before) from e
+
 
 class PloPhoenixdAppLoader(TerminalHarness, PloInterface):
     """Harness to load the binaries to syspage using plo bootloader.