fix(chat-e2e): fixed flaky tests for header tooltip (epam#1814)

WEAIDiscovery · Jul 23, 2024 · ee1500a · ee1500a
1 parent 57905f5
commit ee1500a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/apps/chat-e2e/src/tests/chatHeader.test.ts b/apps/chat-e2e/src/tests/chatHeader.test.ts
@@ -1,7 +1,8 @@
 import { Conversation } from '@/chat/types/chat';
 import { DialAIEntityModel } from '@/chat/types/models';
 import dialTest from '@/src/core/dialFixtures';
-import { ExpectedMessages } from '@/src/testData';
+import { API, ExpectedMessages } from '@/src/testData';
+import { responseThrottlingTimeout } from '@/src/ui/pages';
 import { ModelsUtil } from '@/src/utils';
 import { expect } from '@playwright/test';
 
@@ -55,6 +56,10 @@ dialTest(
       async () => {
         await dialHomePage.openHomePage();
         await dialHomePage.waitForPageLoaded();
+        await dialHomePage.throttleAPIResponse(
+          API.chatHost,
+          responseThrottlingTimeout * 2,
+        );
         const requestsData = await chat.sendRequestWithKeyboard(request, false);
 
         expect

diff --git a/apps/chat-e2e/src/tests/replay.test.ts b/apps/chat-e2e/src/tests/replay.test.ts
@@ -3,6 +3,7 @@ import { FolderInterface } from '@/chat/types/folder';
 import { DialAIEntityModel } from '@/chat/types/models';
 import dialTest from '@/src/core/dialFixtures';
 import {
+  API,
   ExpectedConstants,
   ExpectedMessages,
   Import,
@@ -284,6 +285,7 @@ dialTest(
         await talkToSelector.selectModel(bison);
         await entitySettings.setSystemPrompt(replayPrompt);
         await temperatureSlider.setTemperature(replayTemp);
+        await dialHomePage.throttleAPIResponse(API.chatHost);
         replayRequest = await chat.startReplay();
       },
     );
@@ -585,6 +587,7 @@ dialTest(
           iconsToBeLoaded: [gpt35Model.iconUrl],
         });
         await dialHomePage.waitForPageLoaded();
+        await dialHomePage.throttleAPIResponse(API.chatHost);
         replayRequest = await chat.startReplay(
           conversation.messages[0].content,
         );

diff --git a/apps/chat-e2e/src/ui/pages/basePage.ts b/apps/chat-e2e/src/ui/pages/basePage.ts
@@ -11,7 +11,7 @@ export interface UploadDownloadData {
 }
 
 const apiTimeout = 35000;
-const responseThrottlingTimeout = 2500;
+export const responseThrottlingTimeout = 2500;
 
 export class BasePage {
   protected page: Page;