From 8141df0d3f93a77c3af432c244e9f4b8c51a2e54 Mon Sep 17 00:00:00 2001
From: Xubin Ren <52506698+Re-bin@users.noreply.github.com>
Date: Thu, 21 May 2026 01:32:27 +0800
Subject: [PATCH] fix(tools): stabilize session output test

---
 tests/tools/test_exec_session_tools.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/tools/test_exec_session_tools.py b/tests/tools/test_exec_session_tools.py
index ad2506739..76b3c9781 100644
--- a/tests/tools/test_exec_session_tools.py
+++ b/tests/tools/test_exec_session_tools.py
@@ -269,7 +269,7 @@ def test_write_stdin_can_wait_for_expected_output(tmp_path):
     initial, waited, cleanup = asyncio.run(run())
 
     assert "Process running" in initial
-    assert "booting" in waited
+    assert "booting" in initial + waited
     assert "ready" in waited
     assert "Wait target not observed" not in waited
     assert "Session terminated." in cleanup
@@ -298,7 +298,7 @@ def test_write_stdin_wait_for_reports_timeout_without_killing_session(tmp_path):
     initial, waited, cleanup = asyncio.run(run())
 
     assert "Process running" in initial
-    assert "booting" in waited
+    assert "booting" in initial + waited
     assert "Process running" in waited
     assert "Wait target not observed: 'never-ready'" in waited
     assert "Session terminated." in cleanup