From 553cc80027389dcba4770cf4afeaacaaf345dbbc Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Mon, 20 Apr 2026 20:50:57 +0100
Subject: [PATCH] perf(test): flush mcp notifications directly

---
 src/mcp/channel-server.test.ts | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/src/mcp/channel-server.test.ts b/src/mcp/channel-server.test.ts
index 5d5339edba7..783ad899133 100644
--- a/src/mcp/channel-server.test.ts
+++ b/src/mcp/channel-server.test.ts
@@ -67,6 +67,11 @@ function attachReadyGateway(
   ).resolveReady();
 }
 
+async function flushMcpNotifications() {
+  await Promise.resolve();
+  await Promise.resolve();
+}
+
 describe("openclaw channel mcp server", () => {
   describe("gateway-backed flows", () => {
     describe("gateway integration", () => {
@@ -190,9 +195,8 @@ describe("openclaw channel mcp server", () => {
             },
           });
 
-          await vi.waitFor(() => {
-            expect(channelNotifications).toHaveLength(1);
-          });
+          await flushMcpNotifications();
+          expect(channelNotifications).toHaveLength(1);
           expect(channelNotifications[0]).toMatchObject({
             content: "hello Claude",
             meta: expect.objectContaining({
@@ -229,9 +233,8 @@ describe("openclaw channel mcp server", () => {
             },
           });
 
-          await vi.waitFor(() => {
-            expect(permissionNotifications).toHaveLength(1);
-          });
+          await flushMcpNotifications();
+          expect(permissionNotifications).toHaveLength(1);
           expect(permissionNotifications[0]).toEqual({
             request_id: "abcde",
             behavior: "allow",
@@ -253,9 +256,8 @@ describe("openclaw channel mcp server", () => {
             },
           });
 
-          await vi.waitFor(() => {
-            expect(channelNotifications).toHaveLength(2);
-          });
+          await flushMcpNotifications();
+          expect(channelNotifications).toHaveLength(2);
           expect(channelNotifications[1]).toMatchObject({
             content: "plain string user turn",
             meta: expect.objectContaining({