Fix stablecoins in wallet assertion

marsfoundation · Dec 21, 2024 · 70f9f34 · 70f9f34
1 parent 9cc4211
commit 70f9f34
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/packages/app/src/pages/Savings.PageObject.ts b/packages/app/src/pages/Savings.PageObject.ts
@@ -158,7 +158,16 @@ export class SavingsPageObject extends BasePageObject {
 
   async expectStablecoinsInWalletAssetBalance(assetName: string, value: string): Promise<void> {
     const panel = this.locateStablecoinsInWalletPanel()
-    const row = panel.getByRole('row').filter({ has: this.page.getByRole('cell', { name: assetName, exact: true }) })
+    const row = (() => {
+      if (assetName === 'DAI') {
+        return panel
+          .getByRole('row')
+          .filter({ has: this.page.getByTestId(testIds.savings.stablecoinsInWallet.upgradeDaiToUsdsCell) })
+      }
+
+      return panel.getByRole('row').filter({ has: this.page.getByRole('cell', { name: assetName, exact: true }) })
+    })()
+
     await expect(row.getByRole('cell', { name: value })).toBeVisible()
   }