From 2f5eb7e74bb18b52f0032f483446faca39894ff0 Mon Sep 17 00:00:00 2001
From: Dotta <bippadotta@protonmail.com>
Date: Tue, 14 Apr 2026 12:19:20 -0500
Subject: [PATCH] test(server): refresh issue wake and workspace config mocks

---
 server/src/__tests__/execution-workspaces-service.test.ts | 2 --
 .../__tests__/issue-update-comment-wakeup-routes.test.ts  | 8 ++++++++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/server/src/__tests__/execution-workspaces-service.test.ts b/server/src/__tests__/execution-workspaces-service.test.ts
index dbeef4720ad..ca6b38d56b5 100644
--- a/server/src/__tests__/execution-workspaces-service.test.ts
+++ b/server/src/__tests__/execution-workspaces-service.test.ts
@@ -43,7 +43,6 @@ describe("execution workspace config helpers", () => {
       teardownCommand: "bash ./scripts/teardown-worktree.sh",
       cleanupCommand: "pkill -f vite || true",
       desiredState: null,
-      serviceStates: null,
       workspaceRuntime: {
         services: [{ name: "web", command: "pnpm dev", port: 3100 }],
       },
@@ -74,7 +73,6 @@ describe("execution workspace config helpers", () => {
         teardownCommand: "bash ./scripts/teardown-worktree.sh",
         cleanupCommand: "pkill -f vite || true",
         desiredState: null,
-        serviceStates: null,
         workspaceRuntime: {
           services: [{ name: "web", command: "pnpm dev" }],
         },
diff --git a/server/src/__tests__/issue-update-comment-wakeup-routes.test.ts b/server/src/__tests__/issue-update-comment-wakeup-routes.test.ts
index 63a6fed53cb..1b32f8fca5b 100644
--- a/server/src/__tests__/issue-update-comment-wakeup-routes.test.ts
+++ b/server/src/__tests__/issue-update-comment-wakeup-routes.test.ts
@@ -29,6 +29,10 @@ vi.mock("../services/index.js", () => ({
   }),
   agentService: () => ({
     getById: vi.fn(async () => null),
+    resolveByReference: vi.fn(async (_companyId: string, raw: string) => ({
+      ambiguous: false,
+      agent: { id: raw },
+    })),
   }),
   documentService: () => ({}),
   executionWorkspaceService: () => ({}),
@@ -66,6 +70,10 @@ function registerModuleMocks() {
     }),
     agentService: () => ({
       getById: vi.fn(async () => null),
+      resolveByReference: vi.fn(async (_companyId: string, raw: string) => ({
+        ambiguous: false,
+        agent: { id: raw },
+      })),
     }),
     documentService: () => ({}),
     executionWorkspaceService: () => ({}),