From 082d9c0e4af41e8429f3f40ecb102633c6d21b68 Mon Sep 17 00:00:00 2001
From: Mohamed Boudra <boudra.moha@gmail.com>
Date: Sun, 2 Aug 2026 23:07:22 +0000
Subject: [PATCH] test(heartbeats): match active workspace label

---
 packages/app/e2e/support/helpers/heartbeats.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/app/e2e/support/helpers/heartbeats.ts b/packages/app/e2e/support/helpers/heartbeats.ts
index 3f074473c2..71158b4777 100644
--- a/packages/app/e2e/support/helpers/heartbeats.ts
+++ b/packages/app/e2e/support/helpers/heartbeats.ts
@@ -111,13 +111,15 @@ export async function seedAgentWithHeartbeat(): Promise<AgentHeartbeatScenario>
 export async function expectWorkspaceRunning(page: Page): Promise<void> {
   await expect(
     page
-      .getByRole("button", { name: "Heartbeat workspace", exact: true })
+      .getByRole("button", { name: /^Heartbeat workspace(?:, Working)?$/ })
       .getByTestId("workspace-status-indicator-running"),
   ).toBeVisible({ timeout: 60_000 });
 }
 
 export async function expectWorkspaceDone(page: Page): Promise<void> {
-  const workspaceRow = page.getByRole("button", { name: "Heartbeat workspace", exact: true });
+  const workspaceRow = page.getByRole("button", {
+    name: /^Heartbeat workspace(?:, Working)?$/,
+  });
   await expect(workspaceRow.getByTestId("workspace-status-indicator-done")).toBeVisible({
     timeout: 60_000,
   });