diff --git a/apps/web/tests/chat-long-interactions.e2e.ts b/apps/web/tests/chat-long-interactions.e2e.ts index 44bc6f0012..b9bebbd897 100644 --- a/apps/web/tests/chat-long-interactions.e2e.ts +++ b/apps/web/tests/chat-long-interactions.e2e.ts @@ -260,7 +260,9 @@ describe('web e2e: long Chat interaction contract', () => { expect(await composer.inputValue()).toBe('') expect(await composer.isEnabled()).toBe(true) expect(source.session.events.some(event => carries(event, CONTINUE_PROMPT))).toBe(false) - expect(child.session.events.filter(event => carries(event, CONTINUE_PROMPT))).toHaveLength(1) + expect(child.session.events.filter(event => ( + event.type === 'user/message' && carries(event, CONTINUE_PROMPT) + ))).toHaveLength(1) const lastTurnEnd = child.session.events.findLast((event): event is SessionEvent<'turn/end'> => ( event.type === 'turn/end' )) diff --git a/apps/web/tests/queue-actions.e2e.ts b/apps/web/tests/queue-actions.e2e.ts index 5dd5788eb5..a10ca7188e 100644 --- a/apps/web/tests/queue-actions.e2e.ts +++ b/apps/web/tests/queue-actions.e2e.ts @@ -159,6 +159,8 @@ describe('web e2e: queue row actions', () => { await page.getByRole('button', { name: 'Stop generating' }).click() await firstSettled + await expect.poll(() => page.getByRole('button', { name: 'Stop generating' }).count()) + .toBe(0) await expect.poll(() => page.getByRole('button', { name: 'Remove queued message' }).count()) .toBe(2)