chore: import upstream snapshot with attribution

apps/sim/hooks/queries/mothership-chats.test.ts

@@ -0,0 +1,181 @@
+/**
+ * @vitest-environment node
+ */
+
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
+import type { MothershipResource } from '@/lib/copilot/resources/types'
+
+const { queryClient } = vi.hoisted(() => ({
+  queryClient: {
+    cancelQueries: vi.fn().mockResolvedValue(undefined),
+    invalidateQueries: vi.fn().mockResolvedValue(undefined),
+    getQueryData: vi.fn(),
+    setQueryData: vi.fn(),
+  },
+}))
+
+vi.mock('@tanstack/react-query', () => ({
+  keepPreviousData: {},
+  useQuery: vi.fn(),
+  useQueryClient: vi.fn(() => queryClient),
+  useMutation: vi.fn((options) => options),
+}))
+
+import {
+  fetchMothershipChatHistory,
+  fetchMothershipChats,
+  useAddChatResource,
+} from '@/hooks/queries/mothership-chats'
+
+function jsonResponse(body: unknown, init?: ResponseInit): Response {
+  return new Response(JSON.stringify(body), {
+    status: 200,
+    headers: {
+      'Content-Type': 'application/json',
+    },
+    ...init,
+  })
+}
+
+describe('tasks query boundary parsing', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    vi.stubGlobal('fetch', vi.fn())
+  })
+
+  afterEach(() => {
+    vi.unstubAllGlobals()
+  })
+
+  it('parses valid task metadata responses', async () => {
+    vi.mocked(fetch).mockResolvedValueOnce(
+      jsonResponse({
+        success: true,
+        data: [
+          {
+            id: 'chat-1',
+            title: 'Launch plan',
+            updatedAt: '2026-04-11T10:00:00.000Z',
+            activeStreamId: 'stream-1',
+            lastSeenAt: null,
+            pinned: false,
+          },
+        ],
+      })
+    )
+
+    const tasks = await fetchMothershipChats('ws-1')
+
+    expect(tasks).toHaveLength(1)
+    expect(tasks[0]).toEqual(
+      expect.objectContaining({
+        id: 'chat-1',
+        name: 'Launch plan',
+        isActive: true,
+        isUnread: false,
+        isPinned: false,
+      })
+    )
+    expect(tasks[0]?.updatedAt.toISOString()).toBe('2026-04-11T10:00:00.000Z')
+  })
+
+  it('rejects invalid task metadata responses', async () => {
+    vi.mocked(fetch).mockResolvedValueOnce(
+      jsonResponse({
+        success: true,
+        data: [
+          {
+            id: 123,
+            title: 'Broken',
+            updatedAt: '2026-04-11T10:00:00.000Z',
+            activeStreamId: null,
+            lastSeenAt: null,
+            pinned: false,
+          },
+        ],
+      })
+    )
+
+    await expect(fetchMothershipChats('ws-1')).rejects.toThrow(
+      'Response failed contract validation'
+    )
+  })
+
+  it('parses valid mothership chat history responses', async () => {
+    vi.mocked(fetch).mockResolvedValueOnce(
+      jsonResponse({
+        success: true,
+        chat: {
+          id: 'chat-1',
+          title: 'Task history',
+          messages: [],
+          activeStreamId: 'stream-1',
+          resources: [{ type: 'file', id: 'file-1', title: 'Spec.md' }],
+          streamSnapshot: {
+            events: [],
+            previewSessions: [],
+            status: 'active',
+          },
+        },
+      })
+    )
+
+    const history = await fetchMothershipChatHistory('chat-1')
+
+    expect(history).toEqual({
+      id: 'chat-1',
+      title: 'Task history',
+      messages: [],
+      activeStreamId: 'stream-1',
+      resources: [{ type: 'file', id: 'file-1', title: 'Spec.md' }],
+      streamSnapshot: {
+        events: [],
+        previewSessions: [],
+        status: 'active',
+      },
+    })
+  })
+
+  it('rejects invalid fallback chat history responses', async () => {
+    vi.mocked(fetch)
+      .mockResolvedValueOnce(new Response('Not found', { status: 404 }))
+      .mockResolvedValueOnce(
+        jsonResponse({
+          success: true,
+          chat: {
+            id: 'chat-1',
+            title: null,
+            messages: [],
+            activeStreamId: null,
+            resources: [{ type: 'bogus', id: 'resource-1', title: 'Broken' }],
+          },
+        })
+      )
+
+    await expect(fetchMothershipChatHistory('chat-1')).rejects.toThrow(
+      'Invalid chat response: chat.resources[0].type is invalid'
+    )
+  })
+
+  it('rejects invalid chat resource mutation responses', async () => {
+    vi.mocked(fetch).mockResolvedValueOnce(
+      jsonResponse({
+        success: true,
+      })
+    )
+
+    const mutation = useAddChatResource('chat-1') as unknown as {
+      mutationFn: (input: {
+        chatId: string
+        resource: MothershipResource
+      }) => Promise<{ resources: MothershipResource[] }>
+    }
+
+    await expect(
+      mutation.mutationFn({
+        chatId: 'chat-1',
+        resource: { type: 'file', id: 'file-1', title: 'Spec.md' },
+      })
+    ).rejects.toThrow('Response failed contract validation')
+  })
+})