agentworkshop 0.7.37 → 0.7.39

package/scripts/.mimosa/hook-state/sess_aeaecb58-8833-4b83-b93c-050467d21e03.mtxwtxme-61028-1f4519fbc4.baseline/f10615a6fae18436e70d1201b2d44fdae3174a3aa7bb4db97dea1b3ed291d721.source

@@ -0,0 +1,433 @@
+/**
+ * TaskEngine 测试(node + tsx 直跑,无浏览器)
+ *
+ * 覆盖:
+ *  1. create/dispatch 父子关系(assignee/creator/state/parent)
+ *  2. dispatch 投递 assign 消息(parts + metadata 断言)
+ *  3. 状态机全合法迁移
+ *  4. 非法迁移(COMPLETED→WORKING、终态再迁等)抛 INVALID_TRANSITION
+ *  5. applyEvent artifact 分块 append + progress 折算 / totalChunks 折算
+ *  6. complete 置终态 + 进度 100
+ *  7. reassign retryCount+1 + 投递 assign 消息
+ *  8. cancel 置 CANCELED + 投递 cancel 消息
+ *  9. onChildCompleted 最后一个子任务完成 → 父恢复 WORKING + child-completed 消息
+ */
+import { openWorkshopDb } from '../server/services/workshop/db/database'
+import { createChannelRepo } from '../server/services/workshop/db/channel.repo'
+import { createAgentRepo } from '../server/services/workshop/db/agent.repo'
+import { createTaskRepo } from '../server/services/workshop/db/task.repo'
+import { createMessageRepo } from '../server/services/workshop/db/message.repo'
+import { TaskEngine } from '../server/services/workshop/runtime/task-engine'
+import type { Part, A2AArtifact } from '../server/services/workshop/types/a2a'
+import type { WorkspaceTask } from '../server/services/workshop/types/task'
+import { AppError } from '../server/utils/errors'
+
+let failures = 0
+function check(name: string, ok: boolean, detail = ''): void {
+  console.log(`${ok ? 'PASS' : 'FAIL'}  ${name}${detail ? ` — ${detail}` : ''}`)
+  if (!ok) failures += 1
+}
+
+/** 全新内存库 + engine + 一个 channel + lead/worker 两个 agent */
+function setup() {
+  const db = openWorkshopDb(':memory:')
+  const channels = createChannelRepo(db)
+  const agents = createAgentRepo(db)
+  const tasks = createTaskRepo(db)
+  const messages = createMessageRepo(db)
+  const engine = new TaskEngine({ tasks, messages })
+  const channel = channels.create({ name: 'test-channel' })
+  const lead = agents.create({ name: 'lead', harness: 'mock' })
+  const worker = agents.create({ name: 'worker', harness: 'mock' })
+  return { db, engine, messages, channel, lead, worker, agents }
+}
+
+/** 拉取某 agent 未消费消息并解析 parts/metadata */
+function pendingMessages(messages: ReturnType<typeof createMessageRepo>, channelId: string, agentId: string) {
+  return messages.listPendingByChannelAgent(channelId, agentId).map(row => ({
+    row,
+    parts: JSON.parse(row.partsJson) as Part[],
+    metadata: JSON.parse(row.metadataJson) as Record<string, unknown>,
+  }))
+}
+
+/** 断言抛 INVALID_TRANSITION(400) */
+function expectInvalidTransition(name: string, fn: () => unknown): void {
+  try {
+    fn()
+    check(name, false, '未抛出异常')
+  }
+  catch (e) {
+    const err = e as AppError
+    check(
+      name,
+      err instanceof AppError && err.code === 'INVALID_TRANSITION' && err.status === 400,
+      `code=${err?.code} status=${err?.status}`,
+    )
+  }
+}
+
+function testCreateDispatch(): void {
+  const { engine, lead, worker, channel } = setup()
+  const parent = engine.create({
+    channelId: channel.id,
+    creatorId: lead.id,
+    assigneeId: lead.id,
+    title: '主任务',
+    description: '统筹交付',
+  })
+  check('create 初始 SUBMITTED', parent.state === 'SUBMITTED', `state=${parent.state}`)
+  check('create 落库可查', engine.get(parent.id)?.id === parent.id)
+
+  const working = engine.transition(parent.id, 'WORKING', lead.id)
+  check('create 后 WORKING 迁移', working.state === 'WORKING')
+
+  const child = engine.dispatch(parent, {
+    assigneeId: worker.id,
+    title: '子任务1',
+    description: '实现功能 A',
+  })
+  check('dispatch 子任务 ASSIGNED', child.state === 'ASSIGNED', `state=${child.state}`)
+  check('dispatch 子任务 parentId', child.parentId === parent.id)
+  check('dispatch 子任务 assignee', child.assigneeId === worker.id)
+  check('dispatch 子任务 creator=父 assignee', child.creatorId === parent.assigneeId)
+  check('dispatch 后父转 WAITING', engine.get(parent.id)?.state === 'WAITING')
+}
+
+function testDispatchAssignMessage(): void {
+  const { engine, lead, worker, messages, channel } = setup()
+  const parent = engine.create({
+    channelId: channel.id,
+    creatorId: lead.id,
+    assigneeId: lead.id,
+    title: '主任务',
+  })
+  engine.transition(parent.id, 'WORKING', lead.id)
+  const child = engine.dispatch(parent, {
+    assigneeId: worker.id,
+    title: '子任务1',
+    description: '实现功能 A',
+  })
+
+  const msgs = pendingMessages(messages, channel.id, worker.id)
+  check('dispatch 投递 1 条 assign 消息', msgs.length === 1, `count=${msgs.length}`)
+  const m = msgs[0]
+  check('assign 消息 title part', m.parts[0]?.text === '子任务1', JSON.stringify(m.parts))
+  check('assign 消息 description part', m.parts[1]?.text === '实现功能 A')
+  check('assign 消息 kind', m.metadata['x-aw-task-kind'] === 'assign')
+  check('assign 消息 task-id = 子任务 id', m.metadata['x-aw-task-id'] === child.id)
+  check('assign 消息投递对象', m.row.toAgentId === worker.id)
+}
+
+function testAllLegalTransitions(): void {
+  const { engine, lead, worker, channel } = setup()
+  // 每个迁移用独立任务,避免状态污染
+  const cases: Array<{ from: WorkspaceTask['state'], to: WorkspaceTask['state'] }> = [
+    { from: 'SUBMITTED', to: 'WORKING' },
+    { from: 'SUBMITTED', to: 'ASSIGNED' },
+    { from: 'SUBMITTED', to: 'CANCELED' },
+    { from: 'ASSIGNED', to: 'WORKING' },
+    { from: 'ASSIGNED', to: 'CANCELED' },
+    { from: 'WORKING', to: 'WAITING' },
+    { from: 'WORKING', to: 'COMPLETED' },
+    { from: 'WORKING', to: 'FAILED' },
+    { from: 'WORKING', to: 'CANCELED' },
+    { from: 'WAITING', to: 'WORKING' },
+    { from: 'WAITING', to: 'CANCELED' },
+    { from: 'FAILED', to: 'ASSIGNED' },
+  ]
+  for (const { from, to } of cases) {
+    const t = engine.create({
+      channelId: channel.id,
+      creatorId: lead.id,
+      assigneeId: worker.id,
+      title: `t-${from}-${to}`,
+    })
+    if (from === 'ASSIGNED') {
+      engine.transition(t.id, 'ASSIGNED', lead.id)
+    }
+    else if (from === 'WORKING') {
+      engine.transition(t.id, 'WORKING', worker.id)
+    }
+    else if (from === 'WAITING') {
+      engine.transition(t.id, 'WORKING', worker.id)
+      engine.transition(t.id, 'WAITING', lead.id)
+    }
+    else if (from === 'FAILED') {
+      engine.transition(t.id, 'WORKING', worker.id)
+      engine.transition(t.id, 'FAILED', worker.id)
+    }
+    const r = engine.transition(t.id, to, lead.id)
+    check(`合法迁移 ${from}→${to}`, r.state === to, `got=${r.state}`)
+  }
+}
+
+function testIllegalTransitions(): void {
+  const { engine, lead, worker, channel } = setup()
+  // COMPLETED → WORKING
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: 'x' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    engine.transition(t.id, 'COMPLETED', worker.id)
+    expectInvalidTransition('COMPLETED→WORKING 拒绝', () => engine.transition(t.id, 'WORKING', lead.id))
+  }
+  // WORKING→COMPLETED 后再次 WORKING(终态不可迁)
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: 'y' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    engine.transition(t.id, 'COMPLETED', worker.id)
+    expectInvalidTransition('COMPLETED(终态)→WORKING 拒绝', () => engine.transition(t.id, 'WORKING', lead.id))
+  }
+  // 终态 COMPLETED → FAILED / CANCELED
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: 'z' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    engine.transition(t.id, 'COMPLETED', worker.id)
+    expectInvalidTransition('COMPLETED→FAILED 拒绝', () => engine.transition(t.id, 'FAILED', lead.id))
+    expectInvalidTransition('COMPLETED→CANCELED 拒绝', () => engine.transition(t.id, 'CANCELED', lead.id))
+  }
+  // 非法: SUBMITTED → WAITING / FAILED
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: 'w' })
+    expectInvalidTransition('SUBMITTED→WAITING 拒绝', () => engine.transition(t.id, 'WAITING', lead.id))
+    expectInvalidTransition('SUBMITTED→FAILED 拒绝', () => engine.transition(t.id, 'FAILED', lead.id))
+  }
+  // 非法: WORKING → ASSIGNED
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: 'v' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    expectInvalidTransition('WORKING→ASSIGNED 拒绝', () => engine.transition(t.id, 'ASSIGNED', lead.id))
+  }
+  // 非法: FAILED → WORKING(只能 FAILED→ASSIGNED)
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: 'u' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    engine.transition(t.id, 'FAILED', worker.id)
+    expectInvalidTransition('FAILED→WORKING 拒绝', () => engine.transition(t.id, 'WORKING', lead.id))
+  }
+}
+
+function testApplyEventArtifactChunking(): void {
+  const { engine, lead, worker, channel } = setup()
+  const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '分块任务' })
+  engine.transition(t.id, 'WORKING', worker.id)
+
+  const mkArtifact = (text: string): A2AArtifact => ({
+    artifactId: 'art-1',
+    name: 'report',
+    parts: [{ text }],
+  })
+
+  engine.applyEvent(t.id, { kind: 'artifact', artifact: mkArtifact('chunk-1'), append: true, totalChunks: 4 })
+  let task = engine.get(t.id)!
+  check('第 1 块 append 后 parts=1', task.artifacts[0].parts.length === 1)
+  check('第 1 块 progress=25', task.progress === 25, `progress=${task.progress}`)
+
+  engine.applyEvent(t.id, { kind: 'artifact', artifact: mkArtifact('chunk-2'), append: true, totalChunks: 4 })
+  task = engine.get(t.id)!
+  check('第 2 块 append 后 parts=2(同名合并)', task.artifacts[0].parts.length === 2)
+  check('第 2 块 progress=50', task.progress === 50, `progress=${task.progress}`)
+  check('同名 append 不新增 artifact', task.artifacts.length === 1)
+
+  engine.applyEvent(t.id, { kind: 'artifact', artifact: mkArtifact('chunk-3'), append: true, totalChunks: 4 })
+  engine.applyEvent(t.id, { kind: 'artifact', artifact: mkArtifact('chunk-4'), append: true, totalChunks: 4 })
+  task = engine.get(t.id)!
+  check('4 块收齐 parts=4', task.artifacts[0].parts.length === 4)
+  check('4 块 progress=100', task.progress === 100, `progress=${task.progress}`)
+
+  // 非 append 直接 push 新 artifact
+  const other: A2AArtifact = { artifactId: 'art-2', name: 'notes', parts: [{ text: '总结' }] }
+  engine.applyEvent(t.id, { kind: 'artifact', artifact: other })
+  task = engine.get(t.id)!
+  check('非 append 新增 artifact', task.artifacts.length === 2)
+  check('非 append 不影响 progress', task.progress === 100, `progress=${task.progress}`)
+}
+
+function testComplete(): void {
+  const { engine, lead, worker, channel } = setup()
+  const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '完成任务' })
+  engine.transition(t.id, 'WORKING', worker.id)
+  const done = engine.complete(t.id)
+  check('complete 置 COMPLETED', done.state === 'COMPLETED', `state=${done.state}`)
+  check('complete 进度 100', done.progress === 100, `progress=${done.progress}`)
+}
+
+function testReassign(): void {
+  const { engine, lead, worker, messages, agents, channel } = setup()
+  const other = agents.create({ name: 'worker2', harness: 'mock' })
+  const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '失败任务' })
+  engine.transition(t.id, 'WORKING', worker.id)
+  engine.transition(t.id, 'FAILED', worker.id)
+
+  const r = engine.reassign(t.id, other.id)
+  check('reassign 置 ASSIGNED', r.state === 'ASSIGNED', `state=${r.state}`)
+  check('reassign 换 assignee', r.assigneeId === other.id)
+  check('reassign retryCount+1', r.retryCount === 1, `retry=${r.retryCount}`)
+
+  const msgs = pendingMessages(messages, channel.id, other.id)
+  check('reassign 投递 assign 消息', msgs.length === 1, `count=${msgs.length}`)
+  check('reassign 消息 kind=assign', msgs[0]?.metadata['x-aw-task-kind'] === 'assign')
+  check('reassign 消息 task-id', msgs[0]?.metadata['x-aw-task-id'] === t.id)
+}
+
+function testCancel(): void {
+  const { engine, lead, worker, messages, channel } = setup()
+  const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '取消任务' })
+  engine.transition(t.id, 'WORKING', worker.id)
+
+  const c = engine.cancel(t.id, lead.id)
+  check('cancel 置 CANCELED', c.state === 'CANCELED', `state=${c.state}`)
+
+  const msgs = pendingMessages(messages, channel.id, worker.id)
+  check('cancel 投递 cancel 消息', msgs.length === 1, `count=${msgs.length}`)
+  check('cancel 消息 kind=cancel', msgs[0]?.metadata['x-aw-task-kind'] === 'cancel')
+  check('cancel 消息 task-id', msgs[0]?.metadata['x-aw-task-id'] === t.id)
+}
+
+function testOnChildCompleted(): void {
+  const { engine, lead, worker, messages, channel } = setup()
+  const parent = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: lead.id, title: '主任务' })
+  engine.transition(parent.id, 'WORKING', lead.id)
+
+  const child1 = engine.dispatch(parent, { assigneeId: worker.id, title: '子任务1' })
+  const child2 = engine.dispatch(parent, { assigneeId: worker.id, title: '子任务2' })
+  check('两个子任务后父保持 WAITING', engine.get(parent.id)?.state === 'WAITING')
+
+  // 完成子任务 1(ASSIGNED → WORKING → COMPLETED)
+  engine.transition(child1.id, 'WORKING', worker.id)
+  const done1 = engine.complete(child1.id)
+  engine.onChildCompleted(done1)
+  check('子任务 1 完成后父仍 WAITING', engine.get(parent.id)?.state === 'WAITING')
+
+  // 完成子任务 2(最后一个)
+  engine.transition(child2.id, 'WORKING', worker.id)
+  const done2 = engine.complete(child2.id)
+  engine.onChildCompleted(done2)
+  check('最后一个子任务完成后父恢复 WORKING', engine.get(parent.id)?.state === 'WORKING')
+
+  // child-completed 消息:metadata 断言(x-aw-task-id=父 id,x-aw-child-task-id=子 id)
+  const leadMsgs = pendingMessages(messages, channel.id, lead.id)
+  check('投递 2 条 child-completed 消息', leadMsgs.length === 2, `count=${leadMsgs.length}`)
+  const last = leadMsgs[leadMsgs.length - 1]
+  check('child-completed kind', last?.metadata['x-aw-task-kind'] === 'child-completed')
+  check('child-completed x-aw-task-id=父 id', last?.metadata['x-aw-task-id'] === parent.id)
+  check('child-completed x-aw-child-task-id=子2 id', last?.metadata['x-aw-child-task-id'] === child2.id)
+}
+
+function testApplyEventErrorOnTerminal(): void {
+  const { engine, lead, worker, channel } = setup()
+  // WORKING 收到错误事件 → FAILED(正常路径)
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '正常失败' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    engine.applyEvent(t.id, { kind: 'error', error: { code: 'OMP_LLM_ERROR', message: 'boom' } })
+    check('WORKING 错误事件 → FAILED', engine.get(t.id)?.state === 'FAILED', `state=${engine.get(t.id)?.state}`)
+  }
+  // 已 CANCELED 的任务(取消后 abort 回合报 "Interrupted by user")→ 幂等 no-op,不抛非法迁移
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '取消后回合报错' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    engine.cancel(t.id, lead.id)
+    let threw = false
+    try {
+      engine.applyEvent(t.id, { kind: 'error', error: { code: 'OMP_LLM_ERROR', message: 'Interrupted by user' } })
+    }
+    catch {
+      threw = true
+    }
+    check('CANCELED 错误事件不抛非法迁移', !threw)
+    check('CANCELED 错误事件不改变状态', engine.get(t.id)?.state === 'CANCELED', `state=${engine.get(t.id)?.state}`)
+  }
+  // 已 COMPLETED / FAILED → 同样幂等 no-op
+  {
+    const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '完成后回合报错' })
+    engine.transition(t.id, 'WORKING', worker.id)
+    engine.complete(t.id)
+    engine.applyEvent(t.id, { kind: 'error', error: { code: 'OMP_LLM_ERROR', message: 'late' } })
+    check('COMPLETED 错误事件保持 COMPLETED', engine.get(t.id)?.state === 'COMPLETED', `state=${engine.get(t.id)?.state}`)
+  }
+}
+
+function testSyncBroadcast(): void {
+  // 带 hooks 的 engine:验证 onTaskChange 正常收到 state 与 progress(前端同步链的事实源)
+  const db = openWorkshopDb(':memory:')
+  const channels = createChannelRepo(db)
+  const agents = createAgentRepo(db)
+  const tasks = createTaskRepo(db)
+  const messages = createMessageRepo(db)
+  const events: Array<{ state?: string, progress?: number }> = []
+  const engine = new TaskEngine({ tasks, messages }, {
+    onTaskChange: (e) => {
+      events.push({ state: e.state, progress: e.progress })
+    },
+  })
+  const channel = channels.create({ name: 'sync-channel' })
+  const lead = agents.create({ name: 'lead', harness: 'mock' })
+  const worker = agents.create({ name: 'worker', harness: 'mock' })
+
+  // ① 分块折算进度 → hook 广播 progress(不重复广播相同进度)
+  const t = engine.create({ channelId: channel.id, creatorId: lead.id, assigneeId: worker.id, title: '同步任务' })
+  engine.transition(t.id, 'WORKING', worker.id)
+  const before = events.length
+  engine.applyEvent(t.id, {
+    kind: 'artifact',
+    artifact: { artifactId: 'a1', name: 'report', parts: [{ text: 'chunk-1' }] },
+    append: true,
+    totalChunks: 4,
+  })
+  engine.applyEvent(t.id, {
+    kind: 'artifact',
+    artifact: { artifactId: 'a1', name: 'report', parts: [{ text: 'chunk-2' }] },
+    append: true,
+    totalChunks: 4,
+  })
+  const progressEvents = events.slice(before).filter(e => e.progress !== undefined)
+  check('分块折算进度触发 progress 广播', progressEvents.length === 2, `count=${progressEvents.length}`)
+  check('progress 数值递增正确', progressEvents[0]?.progress === 25 && progressEvents[1]?.progress === 50,
+    `p0=${progressEvents[0]?.progress} p1=${progressEvents[1]?.progress}`)
+  // 相同进度不重复广播(append 无折算变化)
+  engine.applyEvent(t.id, { kind: 'artifact', artifact: { artifactId: 'a2', name: 'notes', parts: [{ text: 'x' }] } })
+  const afterExtra = events.filter(e => e.progress !== undefined).length
+  check('无进度变化的 artifact 不触发 progress 广播', afterExtra === progressEvents.length, `count=${afterExtra}`)
+
+  // ② complete:先置 100 再迁移 → 状态广播时任务进度已是 100(状态帧携带完成态进度)
+  const stateEvents: Array<{ state?: string, progressAtBroadcast?: number }> = []
+  const engine2 = new TaskEngine({ tasks, messages }, {
+    onTaskChange: (e) => {
+      if (e.state === 'COMPLETED') {
+        // 状态广播时刻的 DB 进度(transition 内部读行构造 task.status → 直接可见)
+        stateEvents.push({ state: e.state, progressAtBroadcast: tasks.findById(t.id)?.progress })
+      }
+    },
+  })
+  engine2.complete(t.id)
+  const completed = stateEvents[0]
+  check('COMPLETED 状态广播时进度已为 100', completed?.state === 'COMPLETED' && completed?.progressAtBroadcast === 100,
+    `progress=${completed?.progressAtBroadcast}`)
+}
+
+function main(): void {
+  testCreateDispatch()
+  testDispatchAssignMessage()
+  testAllLegalTransitions()
+  testIllegalTransitions()
+  testApplyEventArtifactChunking()
+  testApplyEventErrorOnTerminal()
+  testSyncBroadcast()
+  testComplete()
+  testReassign()
+  testCancel()
+  testOnChildCompleted()
+
+  console.log('')
+  if (failures === 0) {
+    console.log('ALL PASS')
+    process.exit(0)
+  }
+  else {
+    console.log(`${failures} FAILED`)
+    process.exit(1)
+  }
+}
+
+main()