chore: import upstream snapshot with attribution

apps/sim/lib/copilot/persistence/tool-confirm/tool-confirm.test.ts

@@ -0,0 +1,166 @@
+/**
+ * @vitest-environment node
+ */
+
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+const { getAsyncToolCalls } = vi.hoisted(() => ({
+  getAsyncToolCalls: vi.fn(),
+}))
+
+const channelHandlers = new Set<(event: any) => void>()
+
+vi.mock('@/lib/copilot/async-runs/repository', () => ({
+  getAsyncToolCalls,
+}))
+
+vi.mock('@/lib/events/pubsub', () => ({
+  createPubSubChannel: () => ({
+    publish(event: any) {
+      for (const handler of channelHandlers) handler(event)
+    },
+    subscribe(handler: (event: any) => void) {
+      channelHandlers.add(handler)
+      return () => {
+        channelHandlers.delete(handler)
+      }
+    },
+    dispose() {},
+  }),
+}))
+
+import {
+  getToolConfirmation,
+  publishToolConfirmation,
+  waitForToolConfirmation,
+} from '@/lib/copilot/persistence/tool-confirm'
+
+describe('copilot orchestrator persistence', () => {
+  let row: {
+    status: string
+    error?: string | null
+    result?: unknown
+    updatedAt: Date
+  } | null
+
+  beforeEach(() => {
+    vi.clearAllMocks()
+    channelHandlers.clear()
+    row = null
+    getAsyncToolCalls.mockImplementation(async () => (row ? [row] : []))
+  })
+
+  it('reads the durable DB row as the source of truth', async () => {
+    row = {
+      status: 'completed',
+      result: { ok: true },
+      error: null,
+      updatedAt: new Date('2026-01-01T00:00:00.000Z'),
+    }
+
+    await expect(getToolConfirmation('tool-1')).resolves.toEqual({
+      status: 'success',
+      message: undefined,
+      data: { ok: true },
+      timestamp: '2026-01-01T00:00:00.000Z',
+    })
+  })
+
+  it('preserves primitive durable results in confirmations', async () => {
+    row = {
+      status: 'completed',
+      result: 'done',
+      error: null,
+      updatedAt: new Date('2026-01-01T00:00:00.000Z'),
+    }
+
+    await expect(getToolConfirmation('tool-1')).resolves.toEqual({
+      status: 'success',
+      message: undefined,
+      data: 'done',
+      timestamp: '2026-01-01T00:00:00.000Z',
+    })
+  })
+
+  it('ignores delivered rows in request confirmation flow', async () => {
+    row = {
+      status: 'delivered',
+      result: { ok: true },
+      error: null,
+      updatedAt: new Date('2026-01-01T00:00:00.000Z'),
+    }
+
+    await expect(getToolConfirmation('tool-1')).resolves.toBeNull()
+  })
+
+  it('ignores background when waiting for a foreground terminal status', async () => {
+    row = {
+      status: 'pending',
+      error: null,
+      result: null,
+      updatedAt: new Date('2026-01-01T00:00:00.000Z'),
+    }
+
+    const waitPromise = waitForToolConfirmation('tool-1', 5_000, undefined, {
+      acceptStatus: (status) =>
+        status === 'success' || status === 'error' || status === 'cancelled',
+    })
+
+    publishToolConfirmation({
+      toolCallId: 'tool-1',
+      status: 'background',
+      message: 'Client disconnected, execution continuing server-side',
+      timestamp: '2026-01-01T00:00:01.000Z',
+    })
+
+    await Promise.resolve()
+
+    row = {
+      status: 'completed',
+      error: null,
+      result: { ok: true },
+      updatedAt: new Date('2026-01-01T00:00:02.000Z'),
+    }
+
+    publishToolConfirmation({
+      toolCallId: 'tool-1',
+      status: 'success',
+      timestamp: '2026-01-01T00:00:02.000Z',
+    })
+
+    await expect(waitPromise).resolves.toEqual({
+      status: 'success',
+      message: undefined,
+      data: { ok: true },
+      timestamp: '2026-01-01T00:00:02.000Z',
+    })
+  })
+
+  it('resolves background from the pubsub event when the durable row stays pending', async () => {
+    row = {
+      status: 'pending',
+      error: null,
+      result: null,
+      updatedAt: new Date('2026-01-01T00:00:00.000Z'),
+    }
+
+    const waitPromise = waitForToolConfirmation('tool-1', 5_000, undefined, {
+      acceptStatus: (status) => status === 'background',
+    })
+
+    await Promise.resolve()
+
+    publishToolConfirmation({
+      toolCallId: 'tool-1',
+      status: 'background',
+      message: 'Client disconnected, execution continuing server-side',
+      timestamp: '2026-01-01T00:00:01.000Z',
+    })
+
+    await expect(waitPromise).resolves.toEqual({
+      status: 'background',
+      message: 'Client disconnected, execution continuing server-side',
+      timestamp: '2026-01-01T00:00:01.000Z',
+    })
+  })
+})