feat(chat-e2e): fix replay flaky test (#1858)

epam · Aug 1, 2024 · 75fd22c · 75fd22c
1 parent 1e2edfc
commit 75fd22c
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/apps/chat-e2e/src/testData/conversationHistory/conversationData.ts b/apps/chat-e2e/src/testData/conversationHistory/conversationData.ts
@@ -232,6 +232,7 @@ export class ConversationData extends FolderData {
         (m) => (m.model!.id = updatedModel.id),
       );
       defaultReplayConversation.replay!.replayAsIs = false;
+      defaultReplayConversation.selectedAddons = [];
     }
     return defaultReplayConversation;
   }

diff --git a/apps/chat-e2e/src/tests/abortedReplay.test.ts b/apps/chat-e2e/src/tests/abortedReplay.test.ts
@@ -46,6 +46,7 @@ dialTest(
     sendMessageAssertion,
     page,
   }) => {
+    dialTest.slow();
     setTestIds('EPMRTC-512', 'EPMRTC-3451', 'EPMRTC-1448', 'EPMRTC-1132');
     let firstConversation: Conversation;
     let secondConversation: Conversation;
@@ -101,8 +102,14 @@ dialTest(
     await dialTest.step(
       'Verify no "Share" option is available in dropdown menu for partially replayed conversation',
       async () => {
-        await dialHomePage.openHomePage();
+        await dialHomePage.openHomePage({
+          iconsToBeLoaded: [
+            firstRandomModel.iconUrl,
+            secondRandomModel.iconUrl,
+          ],
+        });
         await dialHomePage.waitForPageLoaded();
+        await conversations.getEntityByName(replayConversation.name).waitFor();
         await conversations.openEntityDropdownMenu(replayConversation.name);
         await conversationDropdownMenuAssertion.assertMenuExcludesOptions(
           MenuOptions.share,
@@ -247,8 +254,10 @@ dialTest(
     iconApiHelper,
     chatMessagesAssertion,
     recentEntitiesAssertion,
+    conversations,
     page,
   }) => {
+    dialTest.slow();
     setTestIds('EPMRTC-1132');
     let firstConversation: Conversation;
     let secondConversation: Conversation;
@@ -316,8 +325,11 @@ dialTest(
     await dialTest.step(
       'Open conversation settings, select "Replay as is" option and verify it is highlighted',
       async () => {
-        await dialHomePage.openHomePage();
+        await dialHomePage.openHomePage({
+          iconsToBeLoaded: [thirdRandomModel.iconUrl, newRandomModel.iconUrl],
+        });
         await dialHomePage.waitForPageLoaded();
+        await conversations.getEntityByName(replayConversation.name).waitFor();
         await chatHeader.openConversationSettingsPopup();
         await recentEntities.replayAsIsButton.click();
         await page.mouse.move(0, 0);

diff --git a/apps/chat-e2e/src/ui/webElements/chat.ts b/apps/chat-e2e/src/ui/webElements/chat.ts
@@ -187,8 +187,10 @@ export class Chat extends BaseElement {
   }
 
   public async proceedReplaying(waitForAnswer = false) {
+    const proceedGeneratingButton = this.getSendMessage().proceedGenerating;
+    await proceedGeneratingButton.waitForState();
     const requestPromise = this.page.waitForRequest(API.chatHost);
-    await this.getSendMessage().proceedGenerating.click();
+    await proceedGeneratingButton.click();
     const request = await requestPromise;
     await this.waitForResponse(waitForAnswer);
     return request.postDataJSON();