chore: import upstream snapshot with attribution

2026-07-13 13:34:48 +08:00
commit 77bb5bf71f
3762 changed files with 353249 additions and 0 deletions
@@ -0,0 +1,85 @@
+import { describe, it, expect, vi, beforeEach, MockInstance, Mock } from 'vitest'
+import { updateAllConversations, WorkflowProps } from './updateAllConversations'
+import * as summaryUpdater from './tagsUpdater'
+
+const as = <T>(x: Partial<T>): T => x as T
+
+type WorkflowProxy = WorkflowProps['workflow']
+type Client = WorkflowProps['client']
+type Logger = WorkflowProps['logger']
+type Conversation = Awaited<ReturnType<Client['listConversations']>>['conversations'][number]
+type Message = Awaited<ReturnType<Client['listMessages']>>['messages'][number]
+
+// NOTE: These tests are temporarily skipped because the mocks need to be
+//       adjusted to use plugin proxies instead of direct client calls (the
+//       plugin should not rely on the client at all).
+
+describe.skip('updateAllConversations', () => {
+  let updateTitleAndSummarySpy: MockInstance<typeof summaryUpdater.updateTitleAndSummary>
+  let acknowledgeStartOfProcessingMock: Mock<WorkflowProxy['acknowledgeStartOfProcessing']>
+  let setCompletedMock: Mock<WorkflowProxy['setCompleted']>
+  let listConversationsMock: Mock<Client['listConversations']>
+  let listMessagesMock: Mock<Client['listMessages']>
+  let loggerInfoMock: Mock<Logger['info']>
+
+  const initProps = (): WorkflowProps =>
+    as<WorkflowProps>({
+      workflow: as<WorkflowProxy>({
+        acknowledgeStartOfProcessing: acknowledgeStartOfProcessingMock,
+        setCompleted: setCompletedMock,
+      }),
+      client: as<Client>({
+        listConversations: listConversationsMock,
+        listMessages: listMessagesMock,
+      }),
+      logger: as<Logger>({
+        info: loggerInfoMock,
+      }),
+    })
+
+  beforeEach(() => {
+    updateTitleAndSummarySpy = vi.spyOn(summaryUpdater, 'updateTitleAndSummary').mockResolvedValue(undefined)
+    acknowledgeStartOfProcessingMock = vi.fn().mockResolvedValue(undefined)
+    setCompletedMock = vi.fn().mockResolvedValue(undefined)
+    listConversationsMock = vi.fn()
+    listMessagesMock = vi.fn()
+    loggerInfoMock = vi.fn()
+  })
+
+  it('should acknowledge start, update all dirty conversations, and complete if no nextToken', async () => {
+    listConversationsMock.mockResolvedValue({
+      conversations: [as<Conversation>({ id: 'c1' }), as<Conversation>({ id: 'c2' })],
+      meta: {},
+    })
+    listMessagesMock.mockResolvedValue({
+      messages: [
+        as<Message>({ id: 'm1', conversationId: 'c1', type: 'text', payload: { text: 'Hello1' } }),
+        as<Message>({ id: 'm2', conversationId: 'c1', type: 'text', payload: { text: 'Hello2' } }),
+      ],
+      meta: {},
+    })
+
+    const props = initProps()
+    await updateAllConversations(props)
+
+    expect(props.workflow.acknowledgeStartOfProcessing).toHaveBeenCalled()
+    expect(props.client.listConversations).toHaveBeenCalledWith({ tags: { isDirty: 'true' } })
+    expect(props.client.listMessages).toHaveBeenCalledTimes(2)
+    expect(updateTitleAndSummarySpy).toHaveBeenCalledTimes(2)
+    expect(props.workflow.setCompleted).toHaveBeenCalled()
+  })
+
+  it('should handle no dirty conversations gracefully', async () => {
+    listConversationsMock.mockResolvedValue({
+      conversations: [],
+      meta: { nextToken: undefined },
+    })
+
+    const props = initProps()
+    await updateAllConversations(props)
+
+    expect(props.client.listMessages).not.toHaveBeenCalled()
+    expect(updateTitleAndSummarySpy).not.toHaveBeenCalled()
+    expect(props.workflow.setCompleted).toHaveBeenCalled()
+  })
+})