Merge pull request #108 from lidofinance/mm-fix-getTokenBalance

fix MM getTokenBalance()
lidofinance · Nov 29, 2023 · a02d663 · a02d663
2 parents 72ece67 + e7ed7ae
commit a02d663
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/packages/wallets/src/metamask/metamask.page.ts b/packages/wallets/src/metamask/metamask.page.ts
@@ -224,10 +224,21 @@ export class MetamaskPage implements WalletPage {
 
   async getTokenBalance(tokenName: string) {
     await this.navigate();
-    await this.page
-      .getByTestId('multichain-token-list-button')
-      .locator(`p:has-text(${tokenName})`)
-      .textContent();
+    //Cannot find locator by exact text since need to find row by text "stETH"/"ETH" but "stETH" contains "ETH"
+    const elements = await this.page.$$(
+      'data-testid=multichain-token-list-item-value',
+    );
+    let tokenBalance = NaN;
+    for (const element of elements) {
+      await element.waitForElementState('visible');
+      const textContent = await element.textContent();
+      const letterTextContent = textContent.match(/[a-zA-Z]+/g);
+      if (letterTextContent.toString().trim() === tokenName) {
+        tokenBalance = parseFloat(await element.textContent());
+        break;
+      }
+    }
+    return tokenBalance;
   }
 
   async confirmTx(page: Page, setAggressiveGas?: boolean) {

diff --git a/packages/wallets/src/wallet.page.ts b/packages/wallets/src/wallet.page.ts
@@ -19,7 +19,7 @@ export interface WalletPage {
 
   openLastTxInEthplorer?(txIndex?: number): Promise<Page>;
 
-  getTokenBalance?(tokenName: string): Promise<void>;
+  getTokenBalance?(tokenName: string): Promise<number>;
 
   confirmAddTokenToWallet?(page: Page): Promise<void>;