From 5d71206b7219c60bc6027d29912a3a52223969ca Mon Sep 17 00:00:00 2001
From: wengshihao <45631373+wengshihao@users.noreply.github.com>
Date: Fri, 6 Nov 2026 12:09:49 +0800
Subject: [PATCH] test(app): wait for the active fork when tracing its source

---
 packages/app/e2e/browser/assistant-fork-menu.spec.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/app/e2e/browser/assistant-fork-menu.spec.ts b/packages/app/e2e/browser/assistant-fork-menu.spec.ts
index 533ac6cf33..8a55d3baf6 100644
--- a/packages/app/e2e/browser/assistant-fork-menu.spec.ts
+++ b/packages/app/e2e/browser/assistant-fork-menu.spec.ts
@@ -98,7 +98,10 @@ test.describe("Assistant fork menu", () => {
 
     // Fork the fork, then follow its immediate source rather than guessing by title.
     await openAgentRoute(page, { workspaceId: session.workspaceId, agentId: forkAgentId });
-    await awaitAssistantMessage(page);
+    await expect(page.getByTestId(`workspace-tab-agent_${forkAgentId}`)).toHaveAttribute(
+      "aria-selected",
+      "true",
+    );
     await forkMostRecentAssistantTurnToNewTab(page);
     await page.getByTestId("composer-chat-history-attachment-pill").click();
     await expect(page.getByTestId("chat-history-content")).toContainText("updates for the fork.");