workspacePresentation.test.ts 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. import { afterEach, describe, expect, it, vi } from 'vitest'
  2. import { createUuid } from '@/utils/id'
  3. import { workspaceMessagesToUi } from '@/components/workspace/viewModels'
  4. afterEach(() => vi.unstubAllGlobals())
  5. describe('workspace chat uses the established presentation', () => {
  6. it('creates valid request ids on LAN HTTP without crypto.randomUUID', () => {
  7. vi.stubGlobal('crypto', { getRandomValues: (bytes: Uint8Array) => { for (let i=0;i<bytes.length;i++) bytes[i]=i; return bytes } })
  8. expect(createUuid()).toMatch(/^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/)
  9. })
  10. it('preserves role identity, reasoning, files and tools for MessageTimeline', () => {
  11. const [message] = workspaceMessagesToUi([{id:'m',conversationId:'c',seq:1,role:'assistant',agentId:'a',agentName:'编辑',content:'[报告](/tmp/report.txt)',reasoning:'检查资料',status:'complete',createdAt:10,attachments:[{id:'f',name:'report.txt',mimeType:'text/plain',size:4,createdAt:10,sender:'agent',sourcePath:'/tmp/report.txt'}],tools:[{id:'t',name:'read_file',status:'tool.complete',result:'ok'}]}])
  12. expect(message).toMatchObject({profile:'a',author:'编辑',reasoning:'检查资料',status:'settled',content:'[报告](/api/app/files/f/download)'})
  13. expect(message?.tools?.[0]).toMatchObject({name:'read_file',status:'success',output:'ok'})
  14. expect(message?.attachments?.[0]?.url).toBe('/api/app/files/f/download')
  15. })
  16. })
  17. import { mount } from '@vue/test-utils'
  18. import { nextTick } from 'vue'
  19. import ConversationList from '@/components/workspace/ConversationList.vue'
  20. import ResourceSidebar from '@/components/app/ResourceSidebar.vue'
  21. import TeamAvatar from '@/components/common/TeamAvatar.vue'
  22. import type { WorkspaceAgent, WorkspaceConversation } from '../../src/shared/workspace'
  23. it('composes real member avatars and refreshes them without changing group membership', async () => {
  24. const agents: WorkspaceAgent[] = ['first','second'].map((id,index) => ({id,name:id,avatar:`yaoyao-mascot:v1:${index ? 'square' : 'circle'}:377fe6:friendly`,instructions:'',nodeId:'local',profile:'default',archived:false,revision:1,createdAt:1,updatedAt:1}))
  25. const c: WorkspaceConversation = {id:'g',kind:'group',name:'群聊',avatar:'data:image/png;base64,AA==',memberIds:['second','first'],instructions:'',administratorId:'first',mode:'host',autoReplyIds:[],maxReplyRounds:1,archived:false,pinned:false,readSeq:0,lastSeq:0,preview:'',createdAt:Date.now(),updatedAt:Date.now(),lastMessageAt:Date.now()}
  26. const wrapper = mount(ConversationList, {props:{conversations:[c],agents}})
  27. expect(wrapper.getComponent(TeamAvatar).props('members')!.map((m: {name:string}) => m.name)).toEqual(['second','first'])
  28. expect(wrapper.findAll('.team-avatar__member')).toHaveLength(2)
  29. expect(wrapper.find('.team-avatar__image').exists()).toBe(false)
  30. expect(wrapper.find('.sidebar-item__row small').text()).toMatch(/\d{2}:\d{2}/)
  31. await wrapper.setProps({agents:agents.map(a=>a.id==='first'?{...a,avatar:'yaoyao-mascot:v1:triangle:d94b52:curious'}:a)})
  32. expect(wrapper.getComponent(TeamAvatar).props('members')![1]!.avatar).toBe('yaoyao-mascot:v1:triangle:d94b52:curious')
  33. await wrapper.setProps({conversations:[{...c,activeRunId:'run',activeAgentId:'second',activeRunStatus:'running'}]})
  34. expect(wrapper.findAll('.team-avatar__member.agent-avatar--working')).toHaveLength(1)
  35. expect(wrapper.find('.team-avatar__member--1').classes()).toContain('agent-avatar--working')
  36. await wrapper.setProps({conversations:[{...c,activeRunId:'run',activeAgentId:'second',activeRunStatus:'waiting'}]})
  37. expect(wrapper.findAll('.team-avatar__member.agent-avatar--waiting')).toHaveLength(1)
  38. await wrapper.setProps({conversations:[c]})
  39. expect(wrapper.findAll('.team-avatar__member.agent-avatar--working')).toHaveLength(0)
  40. expect(wrapper.findAll('.team-avatar__member.agent-avatar--waiting')).toHaveLength(0)
  41. await wrapper.setProps({conversations:[{...c,kind:'direct',memberIds:['first'],activeRunId:'direct-run',activeAgentId:'first',activeRunStatus:'running'}]})
  42. expect(wrapper.find('.sidebar-item__icon .agent-avatar').exists() || wrapper.find('.sidebar-item .agent-avatar').exists()).toBe(true)
  43. expect(wrapper.find('.sidebar-item .agent-avatar--working').exists()).toBe(true)
  44. await wrapper.setProps({conversations:[{...c,kind:'direct',memberIds:['first'],activeRunId:'direct-run',activeAgentId:'first',activeRunStatus:'waiting'}]})
  45. expect(wrapper.find('.sidebar-item .agent-avatar--waiting').exists()).toBe(true)
  46. wrapper.unmount()
  47. })
  48. it.each([
  49. ['direct', '查看机器人'],
  50. ['group', '查看群聊'],
  51. ] as const)('opens a %s conversation from its actions menu with the expected label', async (kind, label) => {
  52. const conversation: WorkspaceConversation = {
  53. id: kind,
  54. kind,
  55. name: kind === 'direct' ? '机器人' : '群聊',
  56. avatar: '',
  57. memberIds: [],
  58. instructions: '',
  59. administratorId: '',
  60. mode: 'host',
  61. autoReplyIds: [],
  62. maxReplyRounds: 1,
  63. archived: false,
  64. pinned: false,
  65. readSeq: 0,
  66. lastSeq: 0,
  67. preview: '',
  68. createdAt: 1,
  69. updatedAt: 1,
  70. }
  71. const wrapper = mount(ConversationList, { attachTo: document.body, props: { conversations: [conversation] } })
  72. wrapper.getComponent(ResourceSidebar).vm.$emit('more', conversation.id, new MouseEvent('click', { clientX: 100, clientY: 100 }))
  73. await nextTick()
  74. const menu = document.body.querySelector<HTMLElement>('[aria-label="聊天操作"]')!
  75. const view = [...menu.querySelectorAll<HTMLButtonElement>('button')].find(button => button.textContent === label)!
  76. view.click()
  77. await nextTick()
  78. expect(wrapper.emitted('select')).toEqual([[conversation.id]])
  79. expect(document.body.querySelector('[aria-label="聊天操作"]')).toBeNull()
  80. wrapper.unmount()
  81. })
  82. it('keeps inline thinking in the process trace and its media out of the message body', () => {
  83. const messages = workspaceMessagesToUi([{
  84. id: 'inline-thinking', conversationId: 'chat', seq: 1, role: 'assistant',
  85. content: '<think>![过程图](/tmp/process.png)</think>[报告](/tmp/report.pdf)',
  86. reasoning: '', status: 'complete', attachments: [], tools: [], createdAt: 1,
  87. }])
  88. expect(messages[0].content).toBe('[报告](/tmp/report.pdf)')
  89. expect(messages[0].reasoning).toBe('![过程图](/tmp/process.png)')
  90. })