wehub-resource-sync
86 lines
3.4 KiB
TypeScript
86 lines
3.4 KiB
TypeScript
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()
|
|
})
|
|
})
|