From f1eeef32ac41185c365753adff8f352bba13426d Mon Sep 17 00:00:00 2001
From: Mohamed Boudra <boudra.moha@gmail.com>
Date: Tue, 18 Aug 2026 22:51:11 +0200
Subject: [PATCH] test(e2e): unbreak tool call shimmer gate and increase
 eviction test timeout

---
 .../app/e2e/browser/agent-message-submission.spec.ts   |  2 +-
 packages/app/e2e/browser/tool-call-shimmer.spec.ts     | 10 +++-------
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/packages/app/e2e/browser/agent-message-submission.spec.ts b/packages/app/e2e/browser/agent-message-submission.spec.ts
index 4c17a0383..6bf5b13ce 100644
--- a/packages/app/e2e/browser/agent-message-submission.spec.ts
+++ b/packages/app/e2e/browser/agent-message-submission.spec.ts
@@ -1272,7 +1272,7 @@ test.describe("Agent message submission", () => {
   test("keeps a streaming hidden submission before its output after workspace eviction", async ({
     page,
   }, testInfo) => {
-    test.setTimeout(120_000);
+    test.setTimeout(240_000);
     await expectHiddenStreamingSubmissionOrderAfterWorkspaceEviction(page, testInfo);
   });
 
diff --git a/packages/app/e2e/browser/tool-call-shimmer.spec.ts b/packages/app/e2e/browser/tool-call-shimmer.spec.ts
index a988626b9..58660c3df 100644
--- a/packages/app/e2e/browser/tool-call-shimmer.spec.ts
+++ b/packages/app/e2e/browser/tool-call-shimmer.spec.ts
@@ -101,16 +101,11 @@ async function gateSecondToolCall(page: Page, agentId: string) {
         firstCallId = toolCall.callId;
       }
       if (toolCall.callId === firstCallId) {
-        if (toolCall.status === "running" || toolCall.status === "executing") {
-          return;
-        }
+        ws.send(message);
         if (toolCall.status === "completed") {
-          // The mock tool completes inside the daemon's coalescing window, so the
-          // browser naturally receives its authoritative completed state first.
-          ws.send(message);
           resolveFirstCompleted();
-          return;
         }
+        return;
       }
 
       secondCallId ??= toolCall.callId;
@@ -148,6 +143,7 @@ async function gateSecondToolCall(page: Page, agentId: string) {
     waitForSecondRunning: () => secondRunning,
     releaseSecondRunning() {
       releaseSecondRequested = true;
+      pauseServerMessages = false;
       if (secondRunningMessage) {
         forwardSecondRunning?.();
       }