From d8092314113db8b5f511d885e1de5dd408d425e3 Mon Sep 17 00:00:00 2001
From: Dotta <bippadotta@protonmail.com>
Date: Sat, 1 Aug 2026 18:26:53 +0000
Subject: [PATCH] test: classify heartbeat run fixtures

Co-Authored-By: Paperclip <noreply@paperclip.ing>
---
 ui/src/components/ActivityCharts.test.tsx      | 1 +
 ui/src/lib/inbox.test.ts                       | 1 +
 ui/src/pages/Inbox.test.tsx                    | 1 +
 ui/storybook/stories/data-viz-misc.stories.tsx | 1 +
 4 files changed, 4 insertions(+)

diff --git a/ui/src/components/ActivityCharts.test.tsx b/ui/src/components/ActivityCharts.test.tsx
index c5a50897f13..9fe6b2f59c0 100644
--- a/ui/src/components/ActivityCharts.test.tsx
+++ b/ui/src/components/ActivityCharts.test.tsx
@@ -83,6 +83,7 @@ function createRun(overrides: Partial<HeartbeatRun> = {}): HeartbeatRun {
     createdAt: new Date("2026-04-20T11:58:00.000Z"),
     updatedAt: new Date("2026-04-20T11:59:00.000Z"),
     ...overrides,
+    scopeKind: overrides.scopeKind ?? "company",
   };
 }
 
diff --git a/ui/src/lib/inbox.test.ts b/ui/src/lib/inbox.test.ts
index b23b32f6914..b5f9e3b0e0a 100644
--- a/ui/src/lib/inbox.test.ts
+++ b/ui/src/lib/inbox.test.ts
@@ -133,6 +133,7 @@ function makeRun(id: string, status: HeartbeatRun["status"], createdAt: string,
     id,
     companyId: "company-1",
     agentId,
+    scopeKind: "company",
     issueId: null,
     responsibleUserId: null,
     invocationSource: "assignment",
diff --git a/ui/src/pages/Inbox.test.tsx b/ui/src/pages/Inbox.test.tsx
index 7f26ae587a7..c3f9e4faade 100644
--- a/ui/src/pages/Inbox.test.tsx
+++ b/ui/src/pages/Inbox.test.tsx
@@ -849,6 +849,7 @@ describe("FailedRunInboxRow", () => {
       continuationAttempt: 0,
       lastUsefulActionAt: null,
       nextAction: null,
+      scopeKind: "company",
       stdoutExcerpt: null,
       stderrExcerpt: null,
       contextSnapshot: null,
diff --git a/ui/storybook/stories/data-viz-misc.stories.tsx b/ui/storybook/stories/data-viz-misc.stories.tsx
index d25777b64cc..d002df8ad65 100644
--- a/ui/storybook/stories/data-viz-misc.stories.tsx
+++ b/ui/storybook/stories/data-viz-misc.stories.tsx
@@ -140,6 +140,7 @@ function makeHeartbeatRun(overrides: Partial<HeartbeatRun>): HeartbeatRun {
     nextAction: null,
     contextSnapshot: null,
     ...overrides,
+    scopeKind: overrides.scopeKind ?? "company",
     createdAt,
     updatedAt: overrides.updatedAt ?? createdAt,
   };