fix(tests): bind the real cancellation handler in the router test; mock the narrowed contract path

Author: waleedlatif1

apps/sim/lib/copilot/server/agent-url.test.ts

@@ -16,7 +16,7 @@ const { envMock } = vi.hoisted(() => ({
   },
 }))
 
-vi.mock('@/lib/api/contracts', () => ({
+vi.mock('@/lib/api/contracts/user', () => ({
   mothershipEnvironmentSchema: {
     safeParse: (value: unknown) =>
       ['default', 'dev', 'staging', 'prod'].includes(String(value))

apps/sim/lib/copilot/tool-executor/router.test.ts

@@ -7,9 +7,11 @@ import { describe, expect, it, vi } from 'vitest'
 /**
  * The handler map is a wiring table from tool id to implementation. Only its
  * shape is asserted here, so every implementation module it imports is stubbed
- * — loading them for real reaches the block registry, the executor, and most
- * of `lib/`. Every export resolves to a mock function, which is all the table
- * needs to bind.
+ * except `workflow/mutations`, which holds the cancellation handler under test
+ * and loads for real so a renamed or removed export fails at link time.
+ * Loading the rest reaches the block registry, the executor, and most of
+ * `lib/`; every stubbed export resolves to a mock function, which is all the
+ * table needs to bind.
  */
 const { stubHandlerModule } = vi.hoisted(() => ({
   stubHandlerModule: () =>
@@ -40,19 +42,20 @@ vi.mock('@/lib/copilot/tools/handlers/restore-resource', stubHandlerModule)
 vi.mock('@/lib/copilot/tools/handlers/run-code', stubHandlerModule)
 vi.mock('@/lib/copilot/tools/handlers/vfs', stubHandlerModule)
 vi.mock('@/lib/copilot/tools/handlers/vfs-mutate', stubHandlerModule)
-vi.mock('@/lib/copilot/tools/handlers/workflow/mutations', stubHandlerModule)
 vi.mock('@/lib/copilot/tools/handlers/workflow/queries', stubHandlerModule)
 
 /** Server-router tools are appended to the map from their own registry, which this test does not cover. */
 vi.mock('@/lib/copilot/tools/server/router', () => ({ getRegisteredServerToolNames: () => [] }))
 
 import { hasHandler } from '@/lib/copilot/tool-executor/executor'
+import { buildHandlerMap } from '@/lib/copilot/tool-executor/handler-map'
 import { ensureHandlersRegistered } from '@/lib/copilot/tool-executor/register-handlers'
 import {
   getToolEntry,
   isSimExecuted,
   toolRequiresApproval,
 } from '@/lib/copilot/tool-executor/router'
+import { executeCancelWorkflowRun } from '@/lib/copilot/tools/handlers/workflow/mutations'
 
 describe('workflow-run cancellation tool routing', () => {
   it('routes cancellation through Sim with write permission and explicit approval', () => {
@@ -68,5 +71,7 @@ describe('workflow-run cancellation tool routing', () => {
     await ensureHandlersRegistered()
 
     expect(hasHandler('cancel_workflow_run')).toBe(true)
+    expect(executeCancelWorkflowRun).toBeTypeOf('function')
+    expect(buildHandlerMap().cancel_workflow_run).toBe(executeCancelWorkflowRun)
   })
 })