chore: updating selector for getting wallet address and adding test c…

…ases to validate the behavior

commands/keplr.js

@@ -267,20 +267,20 @@ const keplr = {
     return true;
   },
 
-  async getWalletAddress() {
+  async getWalletAddress(chainName) {
     playwright.switchToKeplrWindow();
     await module.exports.goToHome();
-    const newTokensSelctorExists = playwright.waitForAndCheckElementExistence(
+    const newTokensSelctorExists = await playwright.waitForAndCheckElementExistence(
       homePageElements.newTokensFoundSelector,
     );
 
     if (newTokensSelctorExists) {
       await module.exports.addNewTokensFound(false);
     }
 
-    const page = playwright.keplrWindow();
     await playwright.waitAndClickByText(notificationPageElements.copyAddress);
-    await page.click(notificationPageElements.copyWalletAddressSelector);
+    await playwright.waitAndClick(notificationPageElements.walletSelectors[chainName])
+
     walletAddress = clipboardy.readSync();
     await playwright.switchToCypressWindow();
     return walletAddress;

pages/keplr/notification-page.js

@@ -1,9 +1,15 @@
 const approveButton = `button`;
 const copyAddress = 'Copy Address';
-const copyWalletAddressSelector = 'div.sc-dkzDqf div.sc-hKMtZM.sc-kDDrLX.cyoEAq.dkJSBQ'
+const copyWalletAddressSelector =
+  'div.sc-dkzDqf div.sc-hKMtZM.sc-kDDrLX.cyoEAq.dkJSBQ';
+const walletSelectors = {
+  AGORIC: 'img[alt="Agoric local"]',
+  COSMOS: 'img[alt="Cosmos Hub"]',
+};
 
 module.exports.notificationPageElements = {
   approveButton,
   copyAddress,
-  copyWalletAddressSelector
-};
+  copyWalletAddressSelector,
+  walletSelectors
+};

support/commands.js

@@ -455,11 +455,12 @@ Cypress.Commands.add('disconnectWalletFromDapp', () => {
   return cy.task('disconnectWalletFromDapp');
 });
 
-Cypress.Commands.add('getWalletAddress', () => {
-  cy.task('getWalletAddress').then(address => {
+Cypress.Commands.add('getWalletAddress', chainName => {
+  cy.task('getWalletAddress', chainName).then(address => {
     return address;
   });
 });
+
 Cypress.Commands.add('switchWallet', walletName => {
   return cy.task('switchWallet', { walletName });
 });

tests/e2e/specs/keplr/keplr-spec.js

@@ -75,6 +75,16 @@ describe('Keplr', () => {
         expect(setupFinished).to.be.true;
       });
     });
+
+    it(`should get wallet address while running addNewTokensFound flow`, () => {
+      cy.getWalletAddress('AGORIC').then(walletAddress => {
+        expect(walletAddress.length).to.be.equal(45);
+      });
+
+      cy.getWalletAddress('COSMOS').then(walletAddress => {
+        expect(walletAddress.length).to.be.equal(45);
+      });
+    });
     it(`should switch to new wallet by name`, () => {
       cy.switchWallet('My Wallet 2').then(taskCompleted => {
         expect(taskCompleted).to.be.true;
@@ -102,8 +112,12 @@ describe('Keplr', () => {
       });
     });
 
-    it(`should get wallet address`, () => {
-      cy.getWalletAddress().then(walletAddress => {
+    it(`should get wallet address without running addNewTokensFound flow`, () => {
+      cy.getWalletAddress('AGORIC').then(walletAddress => {
+        expect(walletAddress.length).to.be.equal(45);
+      });
+
+      cy.getWalletAddress('COSMOS').then(walletAddress => {
         expect(walletAddress.length).to.be.equal(45);
       });
     });