test: Allow token balance to populate after swap

test/e2e/playwright/shared/pageObjects/wallet-page.ts

@@ -1,7 +1,7 @@
 import { type Locator, type Page } from '@playwright/test';
 
 export class WalletPage {
-  private page: Page;
+  readonly page: Page;
 
   readonly importTokensButton: Locator;
 
@@ -21,6 +21,8 @@ export class WalletPage {
 
   readonly importAccountConfirmBtn: Locator;
 
+  readonly tokenBalance: Locator;
+
   constructor(page: Page) {
     this.page = page;
     this.swapButton = this.page.getByTestId('token-overview-button-swap');
@@ -29,6 +31,9 @@ export class WalletPage {
     this.importAccountButton = this.page.getByText('Import account');
     this.importButton = this.page.getByText('Import (');
     this.tokenTab = this.page.getByTestId('account-overview__asset-tab');
+    this.tokenBalance = this.page.getByTestId(
+      'multichain-token-list-item-value',
+    );
     this.addAccountButton = this.page.getByTestId(
       'multichain-account-menu-popover-action-button',
     );
@@ -54,6 +59,8 @@ export class WalletPage {
     await this.importAccountButton.click();
     await this.page.fill('#private-key-box', accountPK);
     await this.importAccountConfirmBtn.click();
+    await this.page.waitForTimeout(2000);
+    return await this.accountMenu.textContent();
   }
 
   async selectTokenWallet() {
@@ -68,4 +75,12 @@ export class WalletPage {
   async selectActivityList() {
     await this.activityListTab.click();
   }
+
+  async getTokenBalance() {
+    return await this.tokenBalance.first().textContent();
+  }
+
+  async waitforTokenBalance(balance: string) {
+    await this.page.waitForSelector(`text=/${balance}/`, { timeout: 60000 });
+  }
 }

test/e2e/playwright/swap/specs/swap.spec.ts

@@ -1,5 +1,5 @@
 import { ethers } from 'ethers';
-import { test } from '@playwright/test';
+import { test, expect } from '@playwright/test';
 import log from 'loglevel';
 
 import { ChromeExtensionPage } from '../../shared/pageObjects/extension-page';
@@ -73,7 +73,9 @@ test.beforeAll(
     walletPage = new WalletPage(page);
 
     await networkController.addCustomNetwork(Tenderly.Mainnet);
-    await walletPage.importAccount(wallet.privateKey);
+    const accountName = await walletPage.importAccount(wallet.privateKey);
+    expect(accountName).toEqual('Account 2');
+    await walletPage.waitforTokenBalance('1 ETH');
   },
 );
 
@@ -82,6 +84,8 @@ testSet.forEach((options) => {
     await walletPage.selectTokenWallet();
     await networkController.selectNetwork(options.network);
     await walletPage.selectSwapAction();
+    // Allow balance label to populate
+    await walletPage.page.waitForTimeout(3000);
     const quoteEntered = await swapPage.enterQuote({
       from: options.source,
       to: options.destination,