From 3205de0030ac5f45ef0360d9abdcfd3ea3a82cf9 Mon Sep 17 00:00:00 2001
From: Dotta <bippadotta@protonmail.com>
Date: Wed, 17 Jun 2026 16:24:05 +0000
Subject: [PATCH] fix routine document cleanup in tests

Co-Authored-By: Paperclip <noreply@paperclip.ing>
---
 server/src/__tests__/plugin-managed-routines.test.ts | 6 ++++++
 server/src/__tests__/qa-routine-secrets-e2e.test.ts  | 6 ++++++
 server/src/__tests__/routine-run-telemetry.test.ts   | 6 ++++++
 3 files changed, 18 insertions(+)

diff --git a/server/src/__tests__/plugin-managed-routines.test.ts b/server/src/__tests__/plugin-managed-routines.test.ts
index 6f89cac17ae..97f93ac6f0b 100644
--- a/server/src/__tests__/plugin-managed-routines.test.ts
+++ b/server/src/__tests__/plugin-managed-routines.test.ts
@@ -7,10 +7,13 @@ import {
   agents,
   companies,
   createDb,
+  documentRevisions,
+  documents,
   issues,
   pluginManagedResources,
   plugins,
   projects,
+  routineDocuments,
   routineRuns,
   routineTriggers,
   routines,
@@ -108,7 +111,10 @@ describeEmbeddedPostgres("plugin-managed routines", () => {
   afterEach(async () => {
     await db.delete(routineRuns);
     await db.delete(routineTriggers);
+    await db.delete(routineDocuments);
     await db.delete(routines);
+    await db.delete(documentRevisions);
+    await db.delete(documents);
     await db.delete(issues);
     await db.delete(agentConfigRevisions);
     await db.delete(activityLog);
diff --git a/server/src/__tests__/qa-routine-secrets-e2e.test.ts b/server/src/__tests__/qa-routine-secrets-e2e.test.ts
index 43d7910ac14..a1ad93e902d 100644
--- a/server/src/__tests__/qa-routine-secrets-e2e.test.ts
+++ b/server/src/__tests__/qa-routine-secrets-e2e.test.ts
@@ -25,7 +25,10 @@ import {
   companySecrets,
   companySecretVersions,
   createDb,
+  documentRevisions,
+  documents,
   projects,
+  routineDocuments,
   routineRuns,
   routines,
   secretAccessEvents,
@@ -61,7 +64,10 @@ describeEmbedded("PAP-9522 QA: routine secrets end-to-end", () => {
     await db.delete(secretAccessEvents);
     await db.delete(companySecretBindings);
     await db.delete(routineRuns);
+    await db.delete(routineDocuments);
     await db.delete(routines);
+    await db.delete(documentRevisions);
+    await db.delete(documents);
     await db.delete(companySecretVersions);
     await db.delete(companySecrets);
     await db.delete(projects);
diff --git a/server/src/__tests__/routine-run-telemetry.test.ts b/server/src/__tests__/routine-run-telemetry.test.ts
index ded455971a7..c3ac93173ad 100644
--- a/server/src/__tests__/routine-run-telemetry.test.ts
+++ b/server/src/__tests__/routine-run-telemetry.test.ts
@@ -5,11 +5,14 @@ import {
   agents,
   companies,
   createDb,
+  documentRevisions,
+  documents,
   executionWorkspaces,
   heartbeatRuns,
   issues,
   projectWorkspaces,
   projects,
+  routineDocuments,
   routineRuns,
   routines,
   routineTriggers,
@@ -54,7 +57,10 @@ describeEmbeddedPostgres("routine run telemetry", () => {
     vi.clearAllMocks();
     await db.delete(routineRuns);
     await db.delete(routineTriggers);
+    await db.delete(routineDocuments);
     await db.delete(routines);
+    await db.delete(documentRevisions);
+    await db.delete(documents);
     await db.delete(heartbeatRuns);
     await db.delete(issues);
     await db.delete(executionWorkspaces);