chore(lightspeed): add e2e for stop conversation (#2720)

* test(lightspeed): add Stop conversation e2e and query delay mock

Cover stopping an in-progress reply and prompt restoration. Extend devMode
with delayed /v1/query mock and adjust fixtures and Playwright config.

Made-with: Cursor

* test(lightspeed): POM helpers for Stop flow and restore default Playwright projects

Add chat composer locators and assertions on LightspeedPage. Refactor Stop
e2e to use sendMessage(..., false) and shared helpers. Revert playwright.config
to channel chrome to match origin/main.

Made-with: Cursor

---------

Co-authored-by: HusneShabbir <husneshabbir447@gmail.com>

Author: HusneShabbir

workspaces/lightspeed/packages/app-legacy/e2e-tests/fixtures/responses.ts

@@ -38,10 +38,6 @@ export const models = [
   },
 ];
 
-export const defaultConversation = {
-  conversation_id: '1348e758-15ed-483a-bca5-b8702bbc79fe',
-};
-
 export const conversations = [
   {
     conversation_id: '1348e758-15ed-483a-bca5-b8702bbc79fe',
@@ -56,11 +52,7 @@ export const conversations = [
 ];
 
 export const moreConversations = [
-  {
-    conversation_id: '1348e758-15ed-483a-bca5-b8702bbc79fe',
-    topic_summary: 'Conversation 1',
-    last_message_timestamp: createdAt,
-  },
+  conversations[0],
   {
     conversation_id: '1348e758-15ed-483a-bca5-b8702bbc79fb',
     topic_summary: 'New Conversation',
@@ -125,6 +117,9 @@ export const demoChatContent = [
 
 export const botResponse = `This is a placeholder message`;
 
+/** SSE `start.request_id` in {@link generateQueryResponse}. */
+const mockStreamRequestId = '0e3c4cd7-2817-4c34-91a2-6944550364df';
+
 export const generateQueryResponse = (conversationId: string) => {
   const tokens = botResponse.match(/(\s+|[^\s]+)/g) || [];
 
@@ -136,7 +131,10 @@ export const generateQueryResponse = (conversationId: string) => {
 
   events.push({
     event: 'start',
-    data: { conversation_id: conversationId },
+    data: {
+      conversation_id: conversationId,
+      request_id: mockStreamRequestId,
+    },
   });
 
   tokens.forEach((token, index) => {

workspaces/lightspeed/packages/app-legacy/e2e-tests/lightspeed.test.ts

@@ -43,8 +43,12 @@ import {
   expectChatbotControlsVisible,
   verifyDisplayModeMenuOptions,
   expectChatInputAreaVisible,
+  expectChatInputValue,
+  expectChatStopButtonVisible,
   expectEmptyChatHistory,
   expectConversationArea,
+  chatStopButton,
+  waitForChatMessageLoadingHidden,
 } from './pages/LightspeedPage';
 import {
   uploadFiles,
@@ -107,6 +111,7 @@ import {
   mockFeedbackStatus,
   mockModels,
   mockQuery,
+  mockQueryWithResponseDelay,
   mockShields,
 } from './utils/devMode';
 import {
@@ -443,6 +448,22 @@ test.describe('Lightspeed tests', () => {
       await expect(botMessage).toContainText(assistantResponse);
     });
 
+    test('Stop ends in-progress reply and restores the prompt in the input', async () => {
+      const stopFlowPrompt = 'Stop button restores prompt in input';
+
+      await mockQueryWithResponseDelay(
+        sharedPage,
+        stopFlowPrompt,
+        conversations,
+        5_000,
+      );
+      await sendMessage(stopFlowPrompt, sharedPage, translations, false);
+      await expectChatStopButtonVisible(sharedPage);
+      await chatStopButton(sharedPage).click();
+      await expectChatInputValue(sharedPage, translations, stopFlowPrompt);
+      await waitForChatMessageLoadingHidden(sharedPage);
+    });
+
     test.describe('Chat Management', () => {
       const testChatName = 'Test Rename';
 

workspaces/lightspeed/packages/app-legacy/e2e-tests/pages/LightspeedPage.ts

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { Page, expect } from '@playwright/test';
+import { Page, expect, type Locator } from '@playwright/test';
 import { LightspeedMessages, evaluateMessage } from '../utils/translations';
 
 export type DisplayMode = 'Overlay' | 'Dock to window' | 'Fullscreen';
@@ -87,13 +87,52 @@ export async function verifyDisplayModeMenuOptions(
     `);
 }
 
+/** Chat composer message field (matches sendMessage in testHelper). */
+export function chatMessageTextbox(page: Page, t: LightspeedMessages): Locator {
+  return page.getByRole('textbox', { name: t['chatbox.message.placeholder'] });
+}
+
+export function chatSendButton(page: Page): Locator {
+  return page.getByRole('button', { name: 'Send' });
+}
+
+export function chatStopButton(page: Page): Locator {
+  return page.getByRole('button', { name: 'Stop' });
+}
+
+export function chatMessageLoading(page: Page): Locator {
+  return page.locator('.pf-chatbot__message-loading');
+}
+
+export async function expectChatStopButtonVisible(
+  page: Page,
+  options?: { timeout?: number },
+) {
+  await expect(chatStopButton(page)).toBeVisible({
+    timeout: options?.timeout ?? 10_000,
+  });
+}
+
+export async function expectChatInputValue(
+  page: Page,
+  t: LightspeedMessages,
+  value: string,
+) {
+  await expect(chatMessageTextbox(page, t)).toHaveValue(value);
+}
+
+export async function waitForChatMessageLoadingHidden(
+  page: Page,
+  timeout = 7_000,
+) {
+  await chatMessageLoading(page).waitFor({ state: 'hidden', timeout });
+}
+
 export async function expectChatInputAreaVisible(
   page: Page,
   t: LightspeedMessages,
 ) {
-  await expect(
-    page.getByRole('textbox', { name: t['chatbox.message.placeholder'] }),
-  ).toBeVisible();
+  await expect(chatMessageTextbox(page, t)).toBeVisible();
 }
 
 export async function expectEmptyChatHistory(

workspaces/lightspeed/packages/app-legacy/e2e-tests/utils/devMode.ts

@@ -72,6 +72,32 @@ export async function mockQuery(
   });
 }
 
+/** Delays the mock response so Stop stays visible during the wait. */
+export async function mockQueryWithResponseDelay(
+  page: Page,
+  query: string,
+  conversations: any[],
+  delayMs: number,
+) {
+  await page.unroute(`${modelBaseUrl}/v1/query`);
+  await page.route(`${modelBaseUrl}/v1/query`, async route => {
+    await new Promise<void>(resolve => {
+      setTimeout(resolve, delayMs);
+    });
+    const payload = route.request().postDataJSON();
+
+    if (payload.query === query) {
+      conversations[1].conversation_id = payload.conversation_id;
+    }
+    const body = generateQueryResponse(
+      payload.query === query
+        ? conversations[1].conversation_id
+        : conversations[0].conversation_id,
+    );
+    await route.fulfill({ body });
+  });
+}
+
 export async function mockFeedbackStatus(page: Page, enabled = true) {
   await page.route(`${modelBaseUrl}/v1/feedback/status`, async route => {
     await route.fulfill({