Browse Source

test(chat): collapse live thinking before checking its tail

_Kerman 3 weeks ago
parent
commit
a9b5faef0b
1 changed files with 6 additions and 1 deletions
  1. 6 1
      apps/web/tests/lifecycle-chrome.e2e.ts

+ 6 - 1
apps/web/tests/lifecycle-chrome.e2e.ts

@@ -227,7 +227,12 @@ describe('web e2e: lifecycle & chrome (workspace flow / reload / dark mode)', ()
       try {
         await input.press('Enter')
         if (MODE !== 'record') {
-          const liveTail = page.locator('[data-variant="think"][data-state="running"] [data-follow-end]')
+          const thinking = page.locator('[data-variant="think"][data-state="running"]')
+          const disclosure = thinking.getByRole('button')
+          await expect.poll(() => disclosure.getAttribute('aria-expanded')).toBe('true')
+          await disclosure.click()
+          await expect.poll(() => disclosure.getAttribute('aria-expanded')).toBe('false')
+          const liveTail = thinking.locator('[data-follow-end]')
           await expect.poll(async () => {
             if (await liveTail.count() !== 1) return false
             return await liveTail.evaluate((element) => {