e2e: fixed tests after icon caching (#550)

Co-authored-by: Irina_Kartun <[email protected]>
epam · Jan 17, 2024 · 377f1dc · 377f1dc
1 parent 4b18543
commit 377f1dc
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 24 deletions.
diff --git a/e2e/src/tests/replay.test.ts b/e2e/src/tests/replay.test.ts
@@ -527,10 +527,7 @@ test(
     await test.step('Send new request with preselected "Replay as is" option and verify message icons correspond models', async () => {
       await dialHomePage.openHomePage();
       await dialHomePage.waitForPageLoaded();
-      await chat.startReplayForDifferentModels([
-        conversation.messages[0].content,
-        conversation.messages[2].content,
-      ]);
+      await chat.startReplayForDifferentModels();
 
       const expectedFirstModelIcon = await apiHelper.getEntityIcon(firstModel);
       const expectedSecondModelIcon =
@@ -827,7 +824,6 @@ test(
     replayAsIs,
   }) => {
     setTestIds('EPMRTC-1330', 'EPMRTC-1332');
-    const newMessages: string[] = [];
     const filename = GeneratorUtil.randomArrayElement([
       Import.v14AppImportedFilename,
       Import.v19AppImportedFilename,
@@ -855,7 +851,6 @@ test(
         );
         await chat.applyChanges().click();
         const newMessage = `${i}*2=`;
-        newMessages.push(newMessage);
         await chat.sendRequestWithButton(newMessage);
       }
     });
@@ -892,10 +887,7 @@ test(
     });
 
     await test.step('Start replaying and verify old requests are replayed using gpt-4 model', async () => {
-      const requests = await chat.startReplayForDifferentModels([
-        Import.oldVersionAppGpt35Message,
-        ...newMessages,
-      ]);
+      const requests = await chat.startReplayForDifferentModels();
       for (let i = 0; i < requests.length; i++) {
         const modelId = i === 1 ? ModelIds.BISON_001 : ModelIds.GPT_4;
         expect

diff --git a/e2e/src/ui/pages/basePage.ts b/e2e/src/ui/pages/basePage.ts
@@ -53,7 +53,7 @@ export class BasePage {
       for (const iconHost of options.iconsToBeLoaded) {
         const resp = this.page.waitForResponse(
           (response) =>
-            response.url() === iconHost && response.status() === 200,
+            response.url().includes(iconHost!) && response.status() === 200,
           { timeout: apiTimeout },
         );
         responses.push(resp);

diff --git a/e2e/src/ui/webElements/chat.ts b/e2e/src/ui/webElements/chat.ts
@@ -114,22 +114,17 @@ export class Chat extends BaseElement {
     return request.postDataJSON();
   }
 
-  public async startReplayForDifferentModels(userRequests: string[]) {
+  public async startReplayForDifferentModels() {
     await this.replay.waitForState();
-    const requestPromises = [];
-    for (const req of userRequests) {
-      const requestPromise = this.waitForRequestSent(req);
-      requestPromises.push(requestPromise);
-    }
+    const requests: string[] = [];
+    this.page.on('request', (data) => {
+      if (data.url().includes(API.chatHost)) {
+        requests.push(data.postData()!);
+      }
+    });
     await this.replay.click();
-
-    const requests = [];
-    for (const req of requestPromises) {
-      const request = await req;
-      requests.push(request);
-    }
     await this.waitForResponse(true);
-    return requests.map((r) => r.postDataJSON());
+    return requests.map((r) => JSON.parse(r));
   }
 
   public async sendRequestInCompareMode(