From e5d8f1be3ccf10e5467c8106091a49312b98a33a Mon Sep 17 00:00:00 2001
From: Dotta <bippadotta@protonmail.com>
Date: Mon, 20 Apr 2026 15:29:21 -0500
Subject: [PATCH] fix(test): clean heartbeat dependency test documents

---
 .../src/__tests__/heartbeat-dependency-scheduling.test.ts   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/server/src/__tests__/heartbeat-dependency-scheduling.test.ts b/server/src/__tests__/heartbeat-dependency-scheduling.test.ts
index 1ba59ebc55d..8e9d2245c82 100644
--- a/server/src/__tests__/heartbeat-dependency-scheduling.test.ts
+++ b/server/src/__tests__/heartbeat-dependency-scheduling.test.ts
@@ -7,9 +7,12 @@ import {
   agentWakeupRequests,
   companies,
   createDb,
+  documentRevisions,
+  documents,
   heartbeatRunEvents,
   heartbeatRuns,
   issueComments,
+  issueDocuments,
   issueRelations,
   issues,
 } from "@paperclipai/db";
@@ -93,6 +96,9 @@ describeEmbeddedPostgres("heartbeat dependency-aware queued run selection", () =
     vi.clearAllMocks();
     runningProcesses.clear();
     await db.delete(issueComments);
+    await db.delete(issueDocuments);
+    await db.delete(documentRevisions);
+    await db.delete(documents);
     await db.delete(issueRelations);
     await db.delete(issues);
     await db.delete(heartbeatRunEvents);