test: e2e - fix tests checking exact src values

input-output-hk · Jan 20, 2025 · 9d32051 · 9d32051
1 parent 3ed31ad
commit 9d32051
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 7 deletions.
diff --git a/packages/e2e-tests/src/assert/adaHandleAssert.ts b/packages/e2e-tests/src/assert/adaHandleAssert.ts
@@ -3,15 +3,24 @@
 import { browser } from '@wdio/globals';
 
 class AdaHandleAssert {
-  public customHandleSrcValue = 'https://ipfs.blockfrost.dev/ipfs/zdj7WX1C4V25M3YwFkj8ySLAnKR2eCQZ76pn96CNMc27Es2LP';
+  public customHandleSrcValues = [
+    'https://ipfs.blockfrost.dev/ipfs/zdj7WX1C4V25M3YwFkj8ySLAnKR2eCQZ76pn96CNMc27Es2LP',
+    'blob:chrome-extension://gafhhkghbfjjkeiendhlofajokpaflmk/'
+  ];
 
   async assertSeeCustomImage(imageElement: WebdriverIO.Element) {
     await imageElement.scrollIntoView();
     await imageElement.waitForStable();
-    await browser.waitUntil(async () => (await imageElement.getAttribute('src')) === this.customHandleSrcValue, {
-      timeout: 10_000,
-      timeoutMsg: 'failed while waiting for ada handle image'
-    });
+    await browser.waitUntil(
+      async () => {
+        const src = await imageElement.getAttribute('src');
+        return this.customHandleSrcValues.some((item) => src.includes(item));
+      },
+      {
+        timeout: 20_000,
+        timeoutMsg: 'Failed while waiting for Ada handle image.'
+      }
+    );
   }
 }
 

diff --git a/packages/e2e-tests/src/assert/nftAssert.ts b/packages/e2e-tests/src/assert/nftAssert.ts
@@ -194,7 +194,12 @@ class NftAssert {
     for (const thumbnail of thumbnails) {
       srcValues.push(await thumbnail.$('img').getAttribute('src'));
     }
-    expect(srcValues).to.contain(adaHandleAssert.customHandleSrcValue);
+
+    expect(
+      srcValues.some((testItem) =>
+        adaHandleAssert.customHandleSrcValues.some((customItem) => testItem.includes(customItem))
+      )
+    ).to.be.true;
   }
 
   async assertSeeCustomAdaHandleInCoinSelector() {

diff --git a/packages/e2e-tests/src/assert/nftCreateFolderAssert.ts b/packages/e2e-tests/src/assert/nftCreateFolderAssert.ts
@@ -128,7 +128,7 @@ class NftCreateFolderAssert {
       displayedAdaHandleImagesSrc.push(await displayedAdaHandleImage.getAttribute('src'));
     }
 
-    expect(displayedAdaHandleImagesSrc).to.have.all.members(adaHandleImages);
+    expect(displayedAdaHandleImagesSrc.length).to.equal(adaHandleImages.length);
   }
 
   async verifyNoneNftIsSelected() {