From dc775b598e3be43f6aebf703a911b08c03549dc3 Mon Sep 17 00:00:00 2001
From: PeterYinusa <53189696+PeterYinusa@users.noreply.github.com>
Date: Mon, 10 Oct 2022 12:43:18 +0100
Subject: [PATCH] add decrypt e2e (#16122)

---
 test/e2e/tests/encrypt-decrypt.spec.js | 89 ++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 test/e2e/tests/encrypt-decrypt.spec.js

diff --git a/test/e2e/tests/encrypt-decrypt.spec.js b/test/e2e/tests/encrypt-decrypt.spec.js
new file mode 100644
index 000000000000..059d93ca9ee2
--- /dev/null
+++ b/test/e2e/tests/encrypt-decrypt.spec.js
@@ -0,0 +1,89 @@
+const { strict: assert } = require('assert');
+const { convertToHexValue, withFixtures } = require('../helpers');
+
+describe('Encrypt Decrypt', function () {
+  const ganacheOptions = {
+    accounts: [
+      {
+        secretKey:
+          '0x7C9529A67102755B7E6102D6D950AC5D5863C98713805CEC576B945B15B71EAC',
+        balance: convertToHexValue(25000000000000000000),
+      },
+    ],
+  };
+  const encryptionKey = 'fxYXfCbun026g5zcCQh7Ia+O0urAEVZWLG8H4Jzu7Xs=';
+  const message = 'Hello, Bob!';
+  it('should decrypt an encrypted message', async function () {
+    await withFixtures(
+      {
+        dapp: true,
+        fixtures: 'connected-state',
+        ganacheOptions,
+        title: this.test.title,
+      },
+      async ({ driver }) => {
+        await driver.navigate();
+        await driver.fill('#password', 'correct horse battery staple');
+        await driver.press('#password', driver.Key.ENTER);
+        await driver.openNewPage('http://127.0.0.1:8080');
+
+        // Get Encryption key
+        await driver.clickElement('#getEncryptionKeyButton');
+        await driver.waitUntilXWindowHandles(3);
+        let windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle(
+          'MetaMask Notification',
+          windowHandles,
+        );
+        await driver.waitForSelector({
+          css: '.request-encryption-public-key__header__text',
+          text: 'Request encryption public key',
+        });
+        await driver.clickElement({ text: 'Provide', tag: 'button' });
+        await driver.waitUntilXWindowHandles(2);
+        windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle('E2E Test Dapp', windowHandles);
+        const encryptionKeyLabel = await driver.findElement(
+          '#encryptionKeyDisplay',
+        );
+        assert.equal(await encryptionKeyLabel.getText(), encryptionKey);
+
+        // Encrypt
+        await driver.fill('#encryptMessageInput', message);
+        await driver.clickElement('#encryptButton');
+        await driver.waitForSelector({
+          css: '#ciphertextDisplay',
+          text: '0x',
+        });
+
+        // Decrypt
+        await driver.clickElement('#decryptButton');
+        await driver.waitUntilXWindowHandles(3);
+        windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle(
+          'MetaMask Notification',
+          windowHandles,
+        );
+        await driver.waitForSelector({
+          css: '.request-decrypt-message__header__text',
+          text: 'Decrypt request',
+        });
+
+        // Verify message in MetaMask Notification
+        await driver.clickElement({ text: 'Decrypt message', tag: 'div' });
+        const notificationMessage = await driver.findElement(
+          '.request-decrypt-message__message-text',
+        );
+        assert.equal(await notificationMessage.getText(), message);
+        await driver.clickElement({ text: 'Decrypt', tag: 'button' });
+
+        // Verify message in Test Dapp
+        await driver.waitUntilXWindowHandles(2);
+        windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle('E2E Test Dapp', windowHandles);
+        const clearTextLabel = await driver.findElement('#cleartextDisplay');
+        assert.equal(await clearTextLabel.getText(), message);
+      },
+    );
+  });
+});