From d9c7158e57f20e02c87cb9476d120b5089f95c29 Mon Sep 17 00:00:00 2001
From: Maska989 <damian.modzelewski@phoenix-rtos.com>
Date: Mon, 13 Jan 2025 10:30:54 +0100
Subject: [PATCH] trunner: fix output after flash

---
 trunner/test_runner.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/trunner/test_runner.py b/trunner/test_runner.py
index 4c40ca9b..5496c701 100644
--- a/trunner/test_runner.py
+++ b/trunner/test_runner.py
@@ -244,6 +244,10 @@ def run_tests(self, tests: Sequence[TestOptions]) -> Sequence[TestResult]:
         # Ensure first test will start with reboot
         last_test_failed = True
 
+        # Unload data before starting tests
+        if self.ctx.should_flash is not False and self.ctx.target.experimental is False:
+            self.target.dut.expect(r".+")
+
         for test in tests:
             # By default we don't want to reboot the entire device to speed up the test execution)
             # if not explicitly required by the test.