feat: initial release v0.3.0
This commit is contained in:
133
tests/unit/run-runtime/task-bridge.test.ts
Normal file
133
tests/unit/run-runtime/task-bridge.test.ts
Normal file
@@ -0,0 +1,133 @@
|
||||
import { describe, expect, it } from 'vitest'
|
||||
import { mapTaskSSEEventToRunEvents } from '@/lib/run-runtime/task-bridge'
|
||||
import { RUN_EVENT_TYPE } from '@/lib/run-runtime/types'
|
||||
import { TASK_EVENT_TYPE, TASK_SSE_EVENT_TYPE, type SSEEvent } from '@/lib/task/types'
|
||||
|
||||
function buildEvent(input: Partial<SSEEvent>): SSEEvent {
|
||||
return {
|
||||
id: input.id || '1',
|
||||
type: input.type || TASK_SSE_EVENT_TYPE.LIFECYCLE,
|
||||
taskId: input.taskId || 'task_1',
|
||||
projectId: input.projectId || 'project_1',
|
||||
userId: input.userId || 'user_1',
|
||||
ts: input.ts || new Date().toISOString(),
|
||||
payload: input.payload || {},
|
||||
taskType: input.taskType || null,
|
||||
targetType: input.targetType || null,
|
||||
targetId: input.targetId || null,
|
||||
episodeId: input.episodeId || null,
|
||||
}
|
||||
}
|
||||
|
||||
describe('task->run event bridge', () => {
|
||||
it('maps task.stream to step.chunk and normalizes lane by kind', () => {
|
||||
const event = buildEvent({
|
||||
type: TASK_SSE_EVENT_TYPE.STREAM,
|
||||
payload: {
|
||||
runId: 'run_1',
|
||||
stepId: 'step_a',
|
||||
stream: {
|
||||
kind: 'reasoning',
|
||||
delta: 'abc',
|
||||
seq: 1,
|
||||
},
|
||||
},
|
||||
})
|
||||
|
||||
const mapped = mapTaskSSEEventToRunEvents(event)
|
||||
expect(mapped).toHaveLength(1)
|
||||
expect(mapped[0]).toMatchObject({
|
||||
runId: 'run_1',
|
||||
eventType: RUN_EVENT_TYPE.STEP_CHUNK,
|
||||
stepKey: 'step_a',
|
||||
lane: 'reasoning',
|
||||
})
|
||||
})
|
||||
|
||||
it('uses taskType-based fallback stepKey for stream when stepId missing', () => {
|
||||
const event = buildEvent({
|
||||
type: TASK_SSE_EVENT_TYPE.STREAM,
|
||||
taskType: 'story_to_script_run',
|
||||
payload: {
|
||||
runId: 'run_1',
|
||||
stream: {
|
||||
kind: 'text',
|
||||
delta: 'hello',
|
||||
seq: 1,
|
||||
},
|
||||
},
|
||||
})
|
||||
|
||||
const mapped = mapTaskSSEEventToRunEvents(event)
|
||||
expect(mapped).toHaveLength(1)
|
||||
expect(mapped[0]).toMatchObject({
|
||||
eventType: RUN_EVENT_TYPE.STEP_CHUNK,
|
||||
stepKey: 'step:story_to_script_run',
|
||||
lane: 'text',
|
||||
})
|
||||
})
|
||||
|
||||
it('maps task.processing + done=true to step.start then step.complete', () => {
|
||||
const event = buildEvent({
|
||||
payload: {
|
||||
runId: 'run_2',
|
||||
stepId: 'step_b',
|
||||
lifecycleType: TASK_EVENT_TYPE.PROCESSING,
|
||||
done: true,
|
||||
},
|
||||
})
|
||||
|
||||
const mapped = mapTaskSSEEventToRunEvents(event)
|
||||
expect(mapped).toHaveLength(2)
|
||||
expect(mapped[0]?.eventType).toBe(RUN_EVENT_TYPE.STEP_START)
|
||||
expect(mapped[1]?.eventType).toBe(RUN_EVENT_TYPE.STEP_COMPLETE)
|
||||
})
|
||||
|
||||
it('maps processing error stage to step.error', () => {
|
||||
const event = buildEvent({
|
||||
payload: {
|
||||
meta: { runId: 'run_3' },
|
||||
stepId: 'step_c',
|
||||
lifecycleType: TASK_EVENT_TYPE.PROCESSING,
|
||||
stage: 'worker_llm_error',
|
||||
error: {
|
||||
message: 'boom',
|
||||
},
|
||||
},
|
||||
})
|
||||
|
||||
const mapped = mapTaskSSEEventToRunEvents(event)
|
||||
expect(mapped).toHaveLength(2)
|
||||
expect(mapped[0]?.eventType).toBe(RUN_EVENT_TYPE.STEP_START)
|
||||
expect(mapped[1]).toMatchObject({
|
||||
eventType: RUN_EVENT_TYPE.STEP_ERROR,
|
||||
runId: 'run_3',
|
||||
stepKey: 'step_c',
|
||||
})
|
||||
})
|
||||
|
||||
it('maps task.completed to step.complete and run.complete', () => {
|
||||
const event = buildEvent({
|
||||
payload: {
|
||||
runId: 'run_4',
|
||||
stepId: 'step_d',
|
||||
lifecycleType: TASK_EVENT_TYPE.COMPLETED,
|
||||
},
|
||||
})
|
||||
|
||||
const mapped = mapTaskSSEEventToRunEvents(event)
|
||||
expect(mapped).toHaveLength(2)
|
||||
expect(mapped[0]?.eventType).toBe(RUN_EVENT_TYPE.STEP_COMPLETE)
|
||||
expect(mapped[1]?.eventType).toBe(RUN_EVENT_TYPE.RUN_COMPLETE)
|
||||
})
|
||||
|
||||
it('returns empty when runId is missing', () => {
|
||||
const event = buildEvent({
|
||||
payload: {
|
||||
stepId: 'step_x',
|
||||
lifecycleType: TASK_EVENT_TYPE.PROCESSING,
|
||||
},
|
||||
})
|
||||
expect(mapTaskSSEEventToRunEvents(event)).toEqual([])
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user