Add accept reject decrypt message

Synthetixio · Mar 4, 2022 · 491b5b5 · 491b5b5
1 parent c42a2fd
commit 491b5b5
Show file tree

Hide file tree

Showing 5 changed files with 53 additions and 0 deletions.
diff --git a/commands/metamask.js b/commands/metamask.js
@@ -17,6 +17,7 @@ const {
   confirmPageElements,
   signaturePageElements,
   encryptionPublicKeyPageElements,
+  decryptPageElements,
 } = require('../pages/metamask/notification-page');
 const {
   settingsPageElements,
@@ -510,6 +511,24 @@ module.exports = {
     await puppeteer.metamaskWindow().waitForTimeout(3000);
     return true;
   },
+  confirmDecryptionRequest: async () => {
+    const notificationPage = await puppeteer.switchToMetamaskNotification();
+    await puppeteer.waitAndClick(
+      decryptPageElements.confirmDecryptionRequestButton,
+      notificationPage,
+    );
+    await puppeteer.metamaskWindow().waitForTimeout(3000);
+    return true;
+  },
+  rejectDecryptionRequest: async () => {
+    const notificationPage = await puppeteer.switchToMetamaskNotification();
+    await puppeteer.waitAndClick(
+      decryptPageElements.rejectDecryptionRequestButton,
+      notificationPage,
+    );
+    await puppeteer.metamaskWindow().waitForTimeout(3000);
+    return true;
+  },
   allowToAddNetwork: async () => {
     const notificationPage = await puppeteer.switchToMetamaskNotification();
     await puppeteer.waitAndClick(

diff --git a/pages/metamask/notification-page.js b/pages/metamask/notification-page.js
@@ -60,3 +60,9 @@ module.exports.encryptionPublicKeyPageElements = {
   rejectEncryptionPublicKeyButton,
 };
 
+const confirmDecryptionRequestButton = `${notificationPage} .request-decrypt-message__footer__sign-button`;
+const rejectDecryptionRequestButton = `${notificationPage} .request-decrypt-message__footer__cancel-button`;
+module.exports.decryptPageElements = {
+  confirmDecryptionRequestButton,
+  rejectDecryptionRequestButton,
+};
diff --git a/plugins/index.js b/plugins/index.js
@@ -155,6 +155,14 @@ module.exports = (on, config) => {
       const rejected = await metamask.rejectEncryptionPublicKeyRequest();
       return rejected;
     },
+    confirmMetamaskDecryptionRequest: async () => {
+      const confirmed = await metamask.confirmDecryptionRequest();
+      return confirmed;
+    },
+    rejectMetamaskDecryptionRequest: async () => {
+      const rejected = await metamask.rejectDecryptionRequest();
+      return rejected;
+    },
     confirmMetamaskPermissionToSpend: async () => {
       const confirmed = await metamask.confirmPermissionToSpend();
       return confirmed;

diff --git a/support/commands.js b/support/commands.js
@@ -91,6 +91,14 @@ Cypress.Commands.add('rejectMetamaskEncryptionPublicKeyRequest', () => {
   return cy.task('rejectMetamaskEncryptionPublicKeyRequest');
 });
 
+Cypress.Commands.add('confirmMetamaskDecryptionRequest', () => {
+  return cy.task('confirmMetamaskDecryptionRequest');
+});
+
+Cypress.Commands.add('rejectMetamaskDecryptionRequest', () => {
+  return cy.task('rejectMetamaskDecryptionRequest');
+});
+
 Cypress.Commands.add('rejectMetamaskSignatureRequest', () => {
   return cy.task('rejectMetamaskSignatureRequest');
 });

diff --git a/support/index.d.ts b/support/index.d.ts
@@ -144,6 +144,18 @@ declare namespace Cypress {
      * cy.rejectMetamaskEncryptionPublicKeyRequest()
      */
     rejectMetamaskEncryptionPublicKeyRequest(): Chainable<Subject>;
+    /**
+     * Confirm metamask request to decrypt message with private key
+     * @example
+     * cy.confirmMetamaskDecryptionRequest()
+     */
+    confirmMetamaskDecryptionRequest(): Chainable<Subject>;
+    /**
+     * Reject metamask request to decrypt message with private key
+     * @example
+     * cy.rejectMetamaskDecryptionRequest()
+     */
+    rejectMetamaskDecryptionRequest(): Chainable<Subject>;
     /**
      * Confirm metamask permission to spend asset
      * @example