From 5543023f48392952f424cd49d886e5755b4c48d5 Mon Sep 17 00:00:00 2001
From: Christine Pinto <chrissie2788@gmail.com>
Date: Fri, 31 May 2024 12:43:33 +0200
Subject: [PATCH] Add clickOnSpecificToken function

---
 tests/pages/wallet.ts              | 7 +++++++
 tests/specs/transactionBTC.spec.ts | 6 +++---
 tests/specs/transactionSTX.spec.ts | 9 +++++----
 3 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/tests/pages/wallet.ts b/tests/pages/wallet.ts
index e4f8af655..5e83ac8c5 100644
--- a/tests/pages/wallet.ts
+++ b/tests/pages/wallet.ts
@@ -472,6 +472,13 @@ export default class Wallet {
     return numericValue;
   }
 
+  async clickOnSpecificToken(tokenname) {
+    const specificToken = this.page
+      .getByRole('button')
+      .filter({ has: this.labelTokenSubtitle.getByText(tokenname, { exact: true }) });
+    await specificToken.last().click();
+  }
+
   async checkNetworkSettingVisuals() {
     await expect(this.buttonSave).toBeVisible();
     await expect(this.buttonBack).toBeVisible();
diff --git a/tests/specs/transactionBTC.spec.ts b/tests/specs/transactionBTC.spec.ts
index 2bb1a4fc0..c5463ba95 100644
--- a/tests/specs/transactionBTC.spec.ts
+++ b/tests/specs/transactionBTC.spec.ts
@@ -43,7 +43,7 @@ test.describe('Transaction BTC', () => {
 
     await expect(await wallet.divTokenRow.count()).toBeGreaterThanOrEqual(2);
     // Send BTC
-    await wallet.divTokenRow.nth(2).click();
+    await wallet.clickOnSpecificToken('Bitcoin');
     await expect(page.url()).toContain('send-btc');
     await expect(wallet.buttonNext).toBeVisible();
     await expect(wallet.buttonNext).toBeDisabled();
@@ -87,7 +87,7 @@ test.describe('Transaction BTC', () => {
     await wallet.allupperButtons.nth(0).click();
 
     await expect(await wallet.divTokenRow.count()).toBeGreaterThanOrEqual(2);
-    await wallet.divTokenRow.nth(2).click();
+    await wallet.clickOnSpecificToken('Bitcoin');
     await expect(page.url()).toContain('send-btc');
     await expect(wallet.buttonNext).toBeVisible();
     await expect(wallet.buttonNext).toBeDisabled();
@@ -151,7 +151,7 @@ test.describe('Transaction BTC', () => {
     await wallet.allupperButtons.nth(0).click();
 
     await expect(await wallet.divTokenRow.count()).toBeGreaterThanOrEqual(2);
-    await wallet.divTokenRow.nth(2).click();
+    await wallet.clickOnSpecificToken('Bitcoin');
     await expect(page.url()).toContain('send-btc');
     await expect(wallet.buttonNext).toBeVisible();
     await expect(wallet.buttonNext).toBeDisabled();
diff --git a/tests/specs/transactionSTX.spec.ts b/tests/specs/transactionSTX.spec.ts
index 8f955cb9c..6de8db345 100644
--- a/tests/specs/transactionSTX.spec.ts
+++ b/tests/specs/transactionSTX.spec.ts
@@ -37,7 +37,8 @@ test.describe('Transaction STX', () => {
 
     await expect(await wallet.divTokenRow.count()).toBeGreaterThanOrEqual(2);
     // Send STX
-    await wallet.divTokenRow.nth(3).click();
+
+    await wallet.clickOnSpecificToken('Stacks');
     await wallet.checkVisualsSendSTXPage();
 
     // Recipient address invalid check
@@ -80,7 +81,7 @@ test.describe('Transaction STX', () => {
     await wallet.allupperButtons.nth(0).click();
 
     await expect(await wallet.divTokenRow.count()).toBeGreaterThanOrEqual(2);
-    await wallet.divTokenRow.nth(3).click();
+    await wallet.clickOnSpecificToken('Stacks');
     await wallet.checkVisualsSendSTXPage();
 
     // Fill in Receiver Address
@@ -140,7 +141,7 @@ test.describe('Transaction STX', () => {
 
     // Needed to add this to avoid loading issues with the token list to be displayed
     await expect(await wallet.divTokenRow.count()).toBeGreaterThanOrEqual(2);
-    await wallet.divTokenRow.nth(3).click();
+    await wallet.clickOnSpecificToken('Stacks');
 
     // Check visuals of sending page
     await wallet.checkVisualsSendSTXPage();
@@ -187,7 +188,7 @@ test.describe('Transaction STX', () => {
     await wallet.navigationDashboard.click();
     // Check if switch to testnet was successfull and all visuals are correct
     await wallet.checkVisualsStartpage('testnet');
-    await wallet.divTokenRow.nth(2).click();
+    await wallet.clickOnSpecificToken('Stacks');
     await expect(page.url()).toContain('STX');
     // Check token detail page for token image and coin title
     await expect(wallet.imageToken).toBeVisible();