workspace-chat.spec.ts 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477
  1. import { test, expect, type Page } from '@playwright/test'
  2. async function login(page: Page) {
  3. await page.goto('/conversations')
  4. await page.getByRole('textbox', { name: '账号', exact: true }).fill('fixture')
  5. await page.getByRole('textbox', { name: '密码', exact: true }).fill('fixture-pass')
  6. await page.getByRole('button', { name: '登录', exact: true }).click()
  7. await expect(page.getByRole('heading', { name: '还没有聊天' })).toBeVisible()
  8. }
  9. async function openCreate(page: Page, name: '新建 Bot' | '新建群聊') {
  10. await page.locator('.desktop-sidebar .sidebar-create-trigger').click()
  11. await page.getByRole('menuitem', { name, exact: true }).click()
  12. }
  13. async function createAgent(page: Page, name: string) {
  14. await openCreate(page, '新建 Bot')
  15. const dialog = page.getByRole('dialog')
  16. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill(name)
  17. await dialog
  18. .getByRole('textbox', { name: '角色提示词与规则', exact: true })
  19. .fill(`你是${name},每次回复给出验收证据。`)
  20. await dialog.getByRole('button', { name: `使用颜色 ${name === '产品经理' ? '#ff6b00' : '#00b9ac'}`, exact: true }).click()
  21. await dialog.locator('.mascot-grid--shape button').filter({ hasText: name === '产品经理' ? '圆形' : '小三角' }).first().click()
  22. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  23. await expect(page.getByRole('heading', { name, exact: true })).toBeVisible()
  24. }
  25. test('an open Web chat list receives cross-client sessions and real title events without reload', async ({
  26. page,
  27. context,
  28. }) => {
  29. await login(page)
  30. const pageAChannelOpened = page.waitForResponse(response =>
  31. response.request().method() === 'POST'
  32. && response.url().endsWith('/api/realtime/channels'))
  33. await page.goto('/chat')
  34. await pageAChannelOpened
  35. await expect(page.getByRole('button', { name: '新建聊天', exact: true })).toBeVisible()
  36. const marker = `${Date.now()}-${Math.random().toString(16).slice(2)}`
  37. const initialTitle = `跨端新会话 ${marker}`
  38. const generatedTitle = `自动标题 ${marker}`
  39. const sentinel = `page-a-${marker}`
  40. await page.evaluate(value => {
  41. ;(window as Window & { __crossClientProjectionSentinel?: string })
  42. .__crossClientProjectionSentinel = value
  43. }, sentinel)
  44. const pageB = await context.newPage()
  45. const pageBChannelOpened = pageB.waitForResponse(response =>
  46. response.request().method() === 'POST'
  47. && response.url().endsWith('/api/realtime/channels'))
  48. await pageB.goto('/chat')
  49. await pageBChannelOpened
  50. await pageB.getByRole('button', { name: '新建聊天', exact: true }).click()
  51. const pageBComposer = pageB.getByRole('textbox', {
  52. name: '输入消息,Enter 发送,Shift + Enter 换行',
  53. })
  54. await pageBComposer.fill(`[cross-client-title:${initialTitle}] 创建跨端会话`)
  55. const pageBSend = pageB.getByRole('button', { name: '发送消息', exact: true })
  56. await expect(pageBSend).toBeEnabled()
  57. await pageBSend.click()
  58. await expect(pageB).toHaveURL(/\/chat\/(?!draft-)[^/?]+(?:\?.*)?$/)
  59. const sessionId = decodeURIComponent(new URL(pageB.url()).pathname.split('/').at(-1) || '')
  60. expect(sessionId).not.toMatch(/^draft-/)
  61. await expect(pageB.locator('.message--assistant')).toContainText('会话独立保存')
  62. const sessionRow = page.locator(`.desktop-sidebar .sidebar-item[data-sidebar-id="${sessionId}"]`)
  63. await expect(sessionRow).toContainText(initialTitle)
  64. expect(await page.evaluate(() =>
  65. (window as Window & { __crossClientProjectionSentinel?: string })
  66. .__crossClientProjectionSentinel)).toBe(sentinel)
  67. await pageB.close()
  68. const titled = await page.request.post(
  69. `http://127.0.0.1:19120/__test/sessions/title?id=${encodeURIComponent(sessionId)}`
  70. + `&title=${encodeURIComponent(generatedTitle)}`,
  71. )
  72. expect(titled.ok(), await titled.text()).toBe(true)
  73. await expect(sessionRow).toContainText(generatedTitle)
  74. await expect(sessionRow).not.toContainText(initialTitle)
  75. expect(await page.evaluate(() =>
  76. (window as Window & { __crossClientProjectionSentinel?: string })
  77. .__crossClientProjectionSentinel)).toBe(sentinel)
  78. })
  79. test('created roles and editable teams share a durable chat list without plugin calls', async ({
  80. page,
  81. context,
  82. }, testInfo) => {
  83. const pluginCalls: string[] = []
  84. page.on('request', (r) => {
  85. if (r.url().includes('/plugins/yaoyao')) pluginCalls.push(r.url())
  86. })
  87. await page.setViewportSize({ width: 1638, height: 874 })
  88. await login(page)
  89. const rail = page.locator('.desktop-sidebar')
  90. await expect(rail.locator('.sidebar-feature-nav')).toHaveCount(0)
  91. await expect(rail.locator('.sidebar-primary-action')).toHaveCount(0)
  92. await expect(rail).toHaveCSS('width', '360px')
  93. await expect(rail.locator('.sidebar-context__heading')).toHaveCount(0)
  94. await expect(rail.locator('.bot-logo-trigger .brand-mark')).toBeVisible()
  95. await expect(rail.locator('.sidebar-footer')).toBeVisible()
  96. await expect(rail.locator('.sidebar-account-switcher__main')).toContainText('fixture')
  97. await expect(rail.locator('.sidebar-account-switcher__main .account-initial-avatar')).toHaveText('F')
  98. await rail.getByRole('button', { name: '设置与模式', exact: true }).click()
  99. await expect(page.getByRole('menu', { name: '设置与模式' }).getByRole('menuitem')).toHaveText(['进入设置', '进入聊天模式'])
  100. await page.getByRole('menuitem', { name: '进入设置', exact: true }).click()
  101. const settings = page.getByRole('dialog', { name: '设置中心', exact: true })
  102. await expect(settings.locator('.settings-account-summary')).toContainText('fixture')
  103. await expect(settings.locator('.settings-account-summary .account-initial-avatar')).toHaveText('F')
  104. await settings.locator('.account-avatar-card input[type=file]').setInputFiles({name:'account.png',mimeType:'image/png',buffer:Buffer.from('iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII=','base64')})
  105. await expect(settings.locator('.settings-account-summary .account-initial-avatar img')).toBeVisible()
  106. expect((await (await page.request.get('/api/app/bootstrap')).json()).user.avatar).toMatch(/^data:image\/png;base64,/)
  107. await settings.getByRole('button', {name:'使用首字母',exact:true}).click()
  108. await expect(settings.locator('.settings-account-summary .account-initial-avatar')).toHaveText('F')
  109. await settings.getByRole('button', { name: '身份与头像', exact: true }).click()
  110. await settings.getByRole('button', { name: '造型:星星', exact: true }).click()
  111. await settings.getByRole('button', { name: '使用颜色 #1488ff', exact: true }).click()
  112. await settings.locator('.mascot-grid--expression button').filter({ hasText: '开心' }).click()
  113. await settings.getByRole('button', { name: '保存更改', exact: true }).click()
  114. await expect(settings.getByRole('button', { name: '保存更改', exact: true })).toBeDisabled()
  115. await settings.getByRole('button', { name: '关闭设置中心', exact: true }).click()
  116. await rail.locator('.sidebar-account-switcher__main').click()
  117. await settings.getByRole('button', { name: '身份与头像', exact: true }).click()
  118. await expect(settings.getByRole('button', { name: '造型:星星', exact: true })).toHaveAttribute('aria-pressed', 'true')
  119. await expect(settings.getByRole('button', { name: '使用颜色 #1488ff', exact: true })).toHaveAttribute('aria-pressed', 'true')
  120. await expect(settings.locator('.mascot-grid--expression button').filter({ hasText: '开心' })).toHaveAttribute('aria-pressed', 'true')
  121. await settings.getByRole('button', { name: '关闭设置中心', exact: true }).click()
  122. await expect(rail.getByRole('checkbox', { name: '显示已归档' })).toHaveCount(0)
  123. const plus = rail.getByRole('button', { name: '新建', exact: true })
  124. await expect(rail.locator('.sidebar-collapse')).toHaveCount(0)
  125. await plus.click()
  126. await expect(page.getByRole('menu', { name: '新建聊天' }).getByRole('menuitem')).toHaveText(['新建 Bot', '添加远程 Agent', '新建群聊'])
  127. await page.keyboard.press('Escape')
  128. await rail.getByRole('button', { name: '搜索', exact: true }).click()
  129. await expect(page.getByRole('dialog', { name: '搜索聊天' })).toBeVisible()
  130. await page.keyboard.press('Escape')
  131. await page.goto('/chat')
  132. await expect(rail).toHaveCSS('width', '264px')
  133. await expect(rail.getByRole('button', { name: '聊天', exact: true })).toHaveCount(0)
  134. await rail.getByRole('button', { name: '设置与模式', exact: true }).click()
  135. const settingsMenu = page.getByRole('menu', { name: '设置与模式', exact: true })
  136. await expect(settingsMenu.getByRole('menuitem')).toHaveText(['进入设置', '进入 Bot 模式'])
  137. await page.screenshot({ path: testInfo.outputPath('chat-settings-mode-menu.png') })
  138. await settingsMenu.getByRole('menuitem', { name: '进入设置', exact: true }).click()
  139. await expect(page.getByRole('dialog', { name: '设置中心', exact: true })).toBeVisible()
  140. await page.getByRole('button', { name: '关闭设置中心', exact: true }).click()
  141. await expect(rail.getByRole('button', { name: '设置与模式', exact: true })).toBeFocused()
  142. await rail.getByRole('button', { name: '设置与模式', exact: true }).click()
  143. await settingsMenu.getByRole('menuitem', { name: '进入 Bot 模式', exact: true }).click()
  144. await expect(page).toHaveURL(/\/conversations$/)
  145. await expect(rail).toHaveCSS('width', '360px')
  146. await page.addInitScript(() => { Object.defineProperty(crypto, 'randomUUID', { value: undefined, configurable: true }) })
  147. await page.reload()
  148. await createAgent(page, '产品经理')
  149. await page.getByRole('button', {name:'聊天设置',exact:true}).click()
  150. const avatarEditor = page.getByRole('dialog')
  151. await expect(avatarEditor.locator('.color-grid button')).toHaveCount(11)
  152. await expect(avatarEditor.locator('.mascot-grid--expression button')).toHaveCount(10)
  153. await avatarEditor.getByRole('button', {name:'造型:星星',exact:true}).click()
  154. await avatarEditor.locator('.mascot-grid--expression button').filter({hasText:'自豪'}).click()
  155. await avatarEditor.getByRole('button', {name:'使用颜色 #000000',exact:true}).click()
  156. await avatarEditor.getByRole('button', {name:'保存',exact:true}).click()
  157. const identities = await (await page.request.get('/api/app/agents')).json()
  158. const edited = identities.agents.find((a: {name:string}) => a.name === '产品经理')
  159. expect(JSON.parse(edited.avatar.slice('yaoyao-avatar:v2:'.length))).toMatchObject({version:2,bodyId:'star',expression:'proud',color:'#000000'})
  160. await page.reload()
  161. await page.getByRole('button', {name:'聊天设置',exact:true}).click()
  162. await expect(avatarEditor.getByRole('button', {name:'造型:星星',exact:true})).toHaveAttribute('aria-pressed','true')
  163. await avatarEditor.locator('input[type=file]').setInputFiles({name:'avatar.png',mimeType:'image/png',buffer:Buffer.from('iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII=','base64')})
  164. await expect(avatarEditor.getByText('图片裁剪',{exact:true})).toBeVisible()
  165. let photoBytes=''
  166. for(const [label,crop] of [['圆形','circle'],['方形','square'],['圆角方形','rounded']]){
  167. await avatarEditor.getByRole('button',{name:label,exact:true}).click()
  168. await avatarEditor.getByRole('button',{name:'保存',exact:true}).click()
  169. const payload=await (await page.request.get('/api/app/agents')).json()
  170. const avatar=JSON.parse(payload.agents.find((a:{name:string})=>a.name==='产品经理').avatar.slice('yaoyao-avatar:v2:'.length))
  171. expect(avatar).toMatchObject({avatarMode:'image',imageCrop:crop})
  172. if(photoBytes)expect(avatar.imageDataURL).toBe(photoBytes)
  173. else photoBytes=avatar.imageDataURL
  174. await page.getByRole('button',{name:'聊天设置',exact:true}).click()
  175. }
  176. await avatarEditor.getByRole('button',{name:'动态吉祥物',exact:true}).click()
  177. await avatarEditor.locator('.mascot-grid--shape button').filter({hasText:'圆形'}).first().click()
  178. await avatarEditor.getByRole('button', {name:'使用颜色 #ff6b00',exact:true}).click()
  179. await avatarEditor.locator('.mascot-grid--expression button').filter({hasText:'默认'}).click()
  180. await avatarEditor.getByRole('button', {name:'保存',exact:true}).click()
  181. await expect(avatarEditor).toBeHidden()
  182. const composerInput = page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行')
  183. const sendButton = page.getByRole('button', { name: '发送消息', exact: true })
  184. await expect(composerInput).toBeEnabled()
  185. await expect(rail.getByText('聊天', { exact: true })).toHaveCount(0)
  186. await expect(rail.getByText('聊天列表', { exact: true })).toHaveCount(0)
  187. await expect(rail.locator('.sidebar-item__row strong').first()).toHaveCSS('font-size', '15px')
  188. await expect(rail.locator('.sidebar-item__row--secondary').first()).toHaveCSS('font-size', '13px')
  189. await expect(rail.locator('.sidebar-item__row small').first()).toHaveCSS('font-size', '10px')
  190. await composerInput.fill('请说明角色职责')
  191. await expect(sendButton).toBeEnabled()
  192. await sendButton.click()
  193. await expect(page.locator('.message--assistant')).toContainText('我是产品经理')
  194. for (const colorScheme of ['light', 'dark'] as const) {
  195. await page.emulateMedia({ colorScheme })
  196. for (const selector of ['.message__avatar.agent-avatar', '.desktop-sidebar .sidebar-item__icon--avatar', '.desktop-sidebar .agent-avatar']) {
  197. await expect(page.locator(selector).first()).toHaveCSS('background-color', 'rgba(0, 0, 0, 0)')
  198. await expect(page.locator(selector).first()).toHaveCSS('background-image', 'none')
  199. }
  200. }
  201. await page.emulateMedia({ colorScheme: 'light' })
  202. await expect(page.locator('.message--assistant a').first()).toHaveAttribute(
  203. 'href',
  204. /\/api\/app\/files\/.+\/download/,
  205. )
  206. await page.locator('.message--assistant a').first().click()
  207. const textPreview = page.getByRole('dialog', { name: '预览 workspace-report.txt', exact: true })
  208. await expect(textPreview).toBeVisible()
  209. await expect(textPreview).toContainText('Fixture report')
  210. await textPreview.getByRole('button', { name: '关闭预览' }).click()
  211. await page.getByRole('button', {name:'置顶聊天',exact:true}).click()
  212. await expect(page.getByRole('button', {name:'取消置顶',exact:true})).toBeVisible()
  213. const directURL = page.url()
  214. await createAgent(page, '开发工程师')
  215. await expect(page.locator('.message')).toHaveCount(0)
  216. await openCreate(page, '新建群聊')
  217. const dialog = page.getByRole('dialog')
  218. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill('产品开发团队')
  219. await dialog.getByRole('checkbox', { name: '产品经理', exact: true }).check()
  220. await dialog.getByRole('checkbox', { name: '开发工程师', exact: true }).check()
  221. await dialog
  222. .getByRole('combobox', { name: '管理员', exact: true })
  223. .selectOption({ label: '产品经理' })
  224. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  225. await expect(page.getByRole('heading', { name: '产品开发团队', exact: true })).toBeVisible()
  226. await expect(page.locator('.desktop-sidebar .sidebar-item')).toHaveCount(3)
  227. // Every cluster portrait must be the same portrait shown in that member's direct chat.
  228. const groupRow = page.locator('.desktop-sidebar .sidebar-item').filter({ hasText: '产品开发团队' })
  229. await expect(groupRow.locator('.team-avatar__member')).toHaveCount(2)
  230. for (const name of ['产品经理', '开发工程师']) {
  231. const portrait = `.agent-avatar[aria-label="${name} 的头像"]`
  232. const direct = page.locator('.desktop-sidebar .sidebar-item').filter({ hasText: name }).filter({ hasNot: page.locator('.team-avatar') }).locator(portrait)
  233. const member = groupRow.locator(portrait)
  234. await expect(member.locator('[data-part=outline] > :first-child')).toHaveAttribute('fill', await direct.locator('[data-part=outline] > :first-child').getAttribute('fill') || '')
  235. expect(await member.locator('[data-part=outline] > :first-child').evaluate(el => el.tagName))
  236. .toBe(await direct.locator('[data-part=outline] > :first-child').evaluate(el => el.tagName))
  237. }
  238. await expect(page.locator('.desktop-sidebar')).not.toContainText('话题')
  239. await expect(page.locator('.desktop-sidebar .sidebar-account-switcher__main')).not.toContainText(
  240. '通用助手',
  241. )
  242. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  243. await expect(dialog.getByRole('checkbox', { name: '产品经理', exact: true })).toBeDisabled()
  244. await expect(dialog.getByRole('checkbox', { name: '开发工程师', exact: true })).toBeEnabled()
  245. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  246. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill('请@开发工程师给出实现结果')
  247. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  248. await expect(page.locator('.message--assistant')).toHaveCount(3)
  249. await expect(page.locator('.message--assistant').nth(1)).toContainText('我是开发工程师')
  250. await expect(page.locator('.run-status')).toHaveCount(0)
  251. const second = await context.newPage()
  252. await second.goto(page.url())
  253. await expect(second.locator('.message--assistant')).toHaveCount(3)
  254. await page.screenshot({ path: testInfo.outputPath('mixed-chats.png'), fullPage: true })
  255. const groupURL=page.url()
  256. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  257. await dialog.getByRole('checkbox', { name: '开发工程师', exact: true }).uncheck()
  258. await expect(dialog.getByRole('checkbox', { name: '产品经理', exact: true })).toBeDisabled()
  259. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  260. await expect(groupRow.locator('.team-avatar__member')).toHaveCount(1)
  261. await expect(page.locator('.message--assistant')).toHaveCount(3)
  262. await expect(second.locator('.desktop-sidebar .sidebar-item').filter({ hasText: '产品开发团队' }).locator('.team-avatar__member')).toHaveCount(1)
  263. await page.reload()
  264. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  265. await expect(dialog.getByRole('checkbox', { name: '开发工程师', exact: true })).not.toBeChecked()
  266. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  267. await createAgent(page, '新成员')
  268. await page.goto(groupURL)
  269. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  270. await dialog.getByRole('checkbox', { name: '新成员', exact: true }).check()
  271. await dialog.getByRole('checkbox', { name: '产品经理', exact: true }).scrollIntoViewIfNeeded()
  272. await page.screenshot({ path: testInfo.outputPath('member-settings.png'), fullPage: true })
  273. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  274. await expect(groupRow.locator('.team-avatar__member')).toHaveCount(2)
  275. await expect(groupRow.locator('[aria-label="开发工程师 的头像"]')).toHaveCount(0)
  276. await expect(groupRow.locator('[aria-label="新成员 的头像"]')).toHaveCount(1)
  277. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill('请@新成员继续')
  278. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  279. await expect(page.locator('.message--assistant')).toHaveCount(6)
  280. await expect(page.locator('.message--assistant').nth(4)).toContainText('我是新成员')
  281. await page.goto('/files')
  282. await page.getByRole('button',{name:'查看来源',exact:true}).first().click()
  283. await expect(page).toHaveURL(groupURL)
  284. await page.getByRole('button', { name: '归档聊天', exact: true }).click()
  285. await expect(page.locator('.desktop-sidebar .sidebar-item')).toHaveCount(3)
  286. await page.locator('.desktop-sidebar .sidebar-search-trigger').click()
  287. const search = page.getByRole('dialog', { name: '搜索聊天', exact: true })
  288. await expect(search).toBeVisible()
  289. await search.getByRole('searchbox', { name: '搜索聊天', exact: true }).fill('产品开发')
  290. await expect(search.getByRole('option')).toHaveCount(0)
  291. await search.getByRole('tab', { name: '已归档', exact: true }).click()
  292. await expect(search.getByRole('searchbox')).toHaveValue('产品开发')
  293. await expect(search.getByRole('option')).toHaveCount(1)
  294. await page.screenshot({ path: testInfo.outputPath('archived-chat-search.png'), fullPage: true })
  295. await search.getByRole('option').click()
  296. await expect(search).not.toBeVisible()
  297. await page.getByRole('button', { name: '恢复聊天', exact: true }).click()
  298. await page.goto(directURL)
  299. await expect(page.locator('.message--assistant')).toHaveCount(1)
  300. await expect(page.locator('.message--assistant')).toContainText('我是产品经理')
  301. await page.goto('/chat')
  302. await expect(page).toHaveURL(/\/chat/)
  303. expect(pluginCalls).toEqual([])
  304. await second.close()
  305. await page.goto('/conversations')
  306. await openCreate(page, '新建群聊')
  307. await expect(dialog.getByRole('button', { name: '信息收集团队', exact: true })).toBeDisabled()
  308. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  309. await createAgent(page, '模板成员四')
  310. await createAgent(page, '模板成员五')
  311. await openCreate(page, '新建群聊')
  312. for (const name of ['信息收集团队', '产品设计团队', '软件开发团队', '文案书写团队', '项目管理团队', '数据分析团队', '会议协作团队', '运维保障团队']) {
  313. await expect(dialog.getByRole('button', { name, exact: true })).toBeEnabled()
  314. }
  315. await page.screenshot({ path: testInfo.outputPath('restored-team-presets.png'), fullPage: true })
  316. await dialog.getByRole('button', { name: '产品设计团队', exact: true }).click()
  317. await expect(dialog.getByRole('textbox', { name: '名称', exact: true })).toHaveValue('产品设计团队')
  318. await dialog.getByRole('combobox', { name: '产品负责人对应的 Agent', exact: true }).selectOption({ label: '产品经理' })
  319. await dialog.getByRole('combobox', { name: '交互设计对应的 Agent', exact: true }).selectOption({ label: '开发工程师' })
  320. const [submitted] = await Promise.all([
  321. page.waitForRequest(r => r.method() === 'POST' && r.url().endsWith('/api/app/conversations')),
  322. dialog.getByRole('button', { name: '保存', exact: true }).click(),
  323. ])
  324. const payload = submitted.postDataJSON()
  325. expect(new Set(payload.memberIds).size).toBe(5)
  326. expect(Object.keys(payload.memberRoles)).toHaveLength(5)
  327. expect(payload.memberRoles[payload.administratorId].name).toBe('产品负责人')
  328. await expect(page.getByRole('heading', { name: '产品设计团队', exact: true })).toBeVisible()
  329. await page.getByRole('button', { name: '聊天设置', exact: true }).click()
  330. await expect(dialog.getByText('产品经理 · 产品负责人', { exact: false })).toBeVisible()
  331. await expect(dialog.getByText('开发工程师 · 交互设计', { exact: false })).toBeVisible()
  332. await dialog.getByRole('button', { name: '取消', exact: true }).click()
  333. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill('请汇总设计方案')
  334. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  335. await expect(page.locator('.message--assistant')).toContainText('我是产品经理')
  336. const composer = page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行')
  337. await composer.fill('[hold-workspace] 正在执行的任务')
  338. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  339. await expect(page.getByRole('button', { name: '停止生成', exact: true })).toBeVisible()
  340. await composer.fill('后续排队任务')
  341. await expect(page.getByRole('button', { name: '发送消息', exact: true })).toBeEnabled()
  342. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  343. await expect(page.locator('.message--user').last()).toContainText('后续排队任务')
  344. const detail = await (await page.request.get(new URL(page.url()).pathname.replace('/conversations/', '/api/app/conversations/'))).json()
  345. expect(detail.conversation.queuedMessageCount).toBe(1)
  346. await page.request.post('http://127.0.0.1:19120/__release')
  347. await expect.poll(async () => (await (await page.request.get(`/api/app/conversations/${detail.conversation.id}`)).json()).conversation.activeRunId).toBeUndefined()
  348. await page.setViewportSize({ width: 390, height: 844 })
  349. await page.getByRole('button', { name: '返回 Bot 列表', exact: true }).click()
  350. await expect(rail).toBeVisible()
  351. await expect(rail).toHaveCSS('width', '390px')
  352. await expect(page.locator('.mobile-header')).toBeHidden()
  353. await expect(rail.locator('.sidebar-feature-nav')).toHaveCount(0)
  354. await page.screenshot({ path: testInfo.outputPath('mobile-bot-list.png') })
  355. await rail.getByRole('button', { name: '新建', exact: true }).click()
  356. await expect(page.getByRole('menuitem', { name: '新建 Bot', exact: true })).toBeVisible()
  357. await page.keyboard.press('Escape')
  358. await rail.getByRole('button', { name: '搜索', exact: true }).click()
  359. await expect(page.getByRole('dialog', { name: '搜索聊天' })).toBeVisible()
  360. await expect(page.getByRole('tab', { name: '未归档', exact: true })).toBeVisible()
  361. await page.getByRole('button', { name: '关闭搜索', exact: true }).click()
  362. await rail.locator('.sidebar-account-switcher__main').click()
  363. await page.getByRole('button', { name: '进入聊天模式', exact: true }).click()
  364. await expect(page).toHaveURL(/\/chat$/)
  365. await page.goto('/')
  366. await expect(page).toHaveURL(/\/chat$/)
  367. await page.setViewportSize({ width: 1280, height: 800 })
  368. await page.getByRole('button', { name: '设置与模式', exact: true }).click()
  369. await page.getByRole('menuitem', { name: '进入 Bot 模式', exact: true }).click()
  370. await page.goto('/')
  371. await expect(page).toHaveURL(/\/conversations$/)
  372. await page.reload()
  373. await expect(rail.locator('.bot-list-header')).toBeVisible()
  374. expect(pluginCalls).toEqual([])
  375. })
  376. test('subaccount allocation and Bot-only navigation work through the browser', async ({ page, browser }, testInfo) => {
  377. await login(page)
  378. await page.locator('.desktop-sidebar').getByRole('button', { name: '设置与模式', exact: true }).click()
  379. await page.getByRole('menuitem', { name: '进入设置', exact: true }).click()
  380. const settings = page.getByRole('dialog', { name: '设置中心', exact: true })
  381. await settings.getByRole('button', { name: '用户与权限', exact: true }).click()
  382. const username = `child-${Date.now()}`
  383. await settings.locator('input[name=managed-username]').fill(username)
  384. await settings.locator('input[name=managed-temporary-password]').fill('temporary-pass')
  385. await settings.getByRole('listbox', { name: '新子账号的基础 Agent' }).selectOption('default')
  386. await settings.getByRole('button', { name: '创建用户', exact: true }).click()
  387. await expect(settings.getByRole('listbox', { name: `分配给 ${username} 的基础 Agent`, exact: true })).toHaveValues(['default'])
  388. await page.screenshot({ path: testInfo.outputPath('admin-agent-allocation.png'), fullPage: true })
  389. const childContext = await browser.newContext({ viewport: { width: 1280, height: 800 } })
  390. const child = await childContext.newPage()
  391. await child.goto(new URL('/chat', page.url()).href)
  392. await child.getByRole('textbox', { name: '账号', exact: true }).fill(username)
  393. await child.getByRole('textbox', { name: '密码', exact: true }).fill('temporary-pass')
  394. await child.getByRole('button', { name: '登录', exact: true }).click()
  395. await child.getByLabel('当前密码', { exact: true }).fill('temporary-pass')
  396. await child.getByLabel('新密码', { exact: true }).fill('permanent-pass')
  397. await child.getByLabel('确认新密码', { exact: true }).fill('permanent-pass')
  398. await child.getByRole('button', { name: '保存并进入夭夭', exact: true }).click()
  399. await expect(child).toHaveURL(/\/conversations$/)
  400. await expect(child.getByRole('heading', { name: '还没有聊天' })).toBeVisible()
  401. await child.locator('.desktop-sidebar').getByRole('button', { name: '设置与模式', exact: true }).click()
  402. await expect(child.getByRole('menu', { name: '设置与模式' }).getByRole('menuitem')).toHaveText(['进入设置'])
  403. await child.keyboard.press('Escape')
  404. await openCreate(child, '新建 Bot')
  405. const dialog = child.getByRole('dialog')
  406. await expect(dialog.getByLabel('基础 Agent').locator('option')).toHaveCount(1)
  407. await expect(dialog.getByLabel('基础 Agent').locator('option')).toContainText('通用助手')
  408. await dialog.getByLabel('基础 Agent').scrollIntoViewIfNeeded()
  409. await child.screenshot({ path: testInfo.outputPath('subaccount-assigned-source.png'), fullPage: true })
  410. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill('子账号助手')
  411. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  412. await expect(child.getByRole('heading', { name: '子账号助手', exact: true })).toBeVisible()
  413. for (const path of ['/history', '/kanban', '/files', '/chat']) {
  414. await child.goto(new URL(path, page.url()).href)
  415. await expect(child).toHaveURL(/\/conversations$/)
  416. }
  417. await childContext.close()
  418. })
  419. for (const mode of ['普通聊天', 'Bot 单聊', '群聊', '普通长回复'] as const) {
  420. test(`${mode} 在完成前格式化流式 Markdown 并保留最终内容`, async ({ page }, testInfo) => {
  421. await login(page)
  422. if (mode.startsWith('普通')) {
  423. await page.goto('/chat')
  424. await page.getByRole('button', { name: '新建聊天', exact: true }).click()
  425. } else {
  426. await createAgent(page, `${mode}流式验证助手`)
  427. if (mode === '群聊') {
  428. await createAgent(page, '群聊流式协作成员')
  429. await openCreate(page, '新建群聊')
  430. const dialog = page.getByRole('dialog')
  431. await dialog.getByRole('textbox', { name: '名称', exact: true }).fill('流式验证群')
  432. await dialog.getByRole('checkbox', { name: `${mode}流式验证助手`, exact: true }).check()
  433. await dialog.getByRole('checkbox', { name: '群聊流式协作成员', exact: true }).check()
  434. await dialog.getByRole('combobox', { name: '管理员', exact: true }).selectOption({ label: `${mode}流式验证助手` })
  435. await dialog.getByRole('button', { name: '保存', exact: true }).click()
  436. await expect(page.getByRole('heading', { name: '流式验证群', exact: true })).toBeVisible()
  437. }
  438. }
  439. await page.getByPlaceholder('输入消息,Enter 发送,Shift + Enter 换行').fill(`[streaming-markdown] ${mode === '普通长回复' ? '[long]' : ''} 演示排版`)
  440. await page.getByRole('button', { name: '发送消息', exact: true }).click()
  441. const message = page.locator('.message--assistant').last()
  442. await expect(message.locator('.markdown--streaming h1')).toHaveText('流式标题')
  443. await expect(message.locator('strong').filter({ hasText: '即时格式化' })).toBeVisible()
  444. await expect(message.locator('li')).toHaveText(['第一项', '第二项'])
  445. await expect(message.locator('pre code').first()).toContainText('const answer = 42')
  446. await expect(message).not.toContainText('最终完整标记')
  447. await page.screenshot({ path: testInfo.outputPath('streaming-markdown.png'), fullPage: true })
  448. if (mode === '普通长回复') {
  449. await expect(message.getByText('流式末尾标记', { exact: true })).toBeInViewport()
  450. await page.locator('.timeline').evaluate(el => { el.scrollTop = 0; el.dispatchEvent(new Event('scroll')) })
  451. await expect(message.getByRole('heading', { name: '流式标题', exact: true })).toBeInViewport()
  452. }
  453. const released = await page.request.post('http://127.0.0.1:19120/__release')
  454. expect(released.ok()).toBe(true)
  455. await expect(message).toContainText('最终完整标记')
  456. await expect(message.locator('table')).toContainText('项目')
  457. await expect(message.locator('.markdown--streaming')).toHaveCount(0)
  458. if (mode === '普通长回复') {
  459. await expect(message.getByRole('heading', { name: '流式标题', exact: true })).toBeInViewport()
  460. await expect(message.getByText('最终完整标记', { exact: true })).not.toBeInViewport()
  461. }
  462. })
  463. }