From c1e53b3330c78c369956a292888edecb77834730 Mon Sep 17 00:00:00 2001
From: Bowen Sanders <bowensanders@gmail.com>
Date: Wed, 27 Apr 2022 11:21:40 -0700
Subject: [PATCH] Create new e2e test for bip-44 snap in test-snaps (#14440)

* created new test for bip-44 snap test

* added driver.Key.SPACE to allow sending spacebar

* made changes to final result check

* fixed expected pk result to proper value

* fixed to use npm package instead of local

* removed comment

* removed const delay - not needed (lint error)
---
 test/e2e/snaps/test-snap-bip-44.spec.js | 94 +++++++++++++++++++++++++
 test/e2e/webdriver/driver.js            |  1 +
 2 files changed, 95 insertions(+)
 create mode 100644 test/e2e/snaps/test-snap-bip-44.spec.js

diff --git a/test/e2e/snaps/test-snap-bip-44.spec.js b/test/e2e/snaps/test-snap-bip-44.spec.js
new file mode 100644
index 000000000..689e62b03
--- /dev/null
+++ b/test/e2e/snaps/test-snap-bip-44.spec.js
@@ -0,0 +1,94 @@
+const { strict: assert } = require('assert');
+const { withFixtures } = require('../helpers');
+
+describe('Test Snap bip-44', function () {
+  it('can pop up bip-44 snap and get private key result', async function () {
+    const ganacheOptions = {
+      accounts: [
+        {
+          secretKey:
+            '0x7C9529A67102755B7E6102D6D950AC5D5863C98713805CEC576B945B15B71EAC',
+          balance: 25000000000000000000,
+        },
+      ],
+    };
+    await withFixtures(
+      {
+        fixtures: 'imported-account',
+        ganacheOptions,
+        title: this.test.title,
+        driverOptions: {
+          type: 'flask',
+        },
+      },
+      async ({ driver }) => {
+        await driver.navigate();
+
+        // enter pw into extension
+        await driver.fill('#password', 'correct horse battery staple');
+        await driver.press('#password', driver.Key.ENTER);
+
+        // navigate to test snaps page and connect
+        await driver.driver.get('https://metamask.github.io/test-snaps/0.1.3/');
+        await driver.fill('.snapId3', 'npm:@metamask/test-snap-bip44');
+        await driver.clickElement({
+          text: 'Connect BIP-44 Snap',
+          tag: 'button',
+        });
+
+        // switch to metamask extension and click connect
+        await driver.waitUntilXWindowHandles(2, 5000, 10000);
+        let windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle(
+          'MetaMask Notification',
+          windowHandles,
+        );
+        await driver.clickElement(
+          {
+            text: 'Connect',
+            tag: 'button',
+          },
+          10000,
+        );
+
+        await driver.delay(2000);
+
+        // approve install of snap
+        await driver.waitUntilXWindowHandles(2, 5000, 10000);
+        windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle(
+          'MetaMask Notification',
+          windowHandles,
+        );
+        await driver.clickElement({
+          text: 'Approve & Install',
+          tag: 'button',
+        });
+        // deal with permissions popover
+        await driver.delay(1000);
+        await driver.press('#warning-accept', driver.Key.SPACE);
+        await driver.clickElement({
+          text: 'Confirm',
+          tag: 'button',
+        });
+
+        // click send inputs on test snap page
+        await driver.waitUntilXWindowHandles(1, 5000, 10000);
+        windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle('Test Snaps', windowHandles);
+        await driver.clickElement({
+          text: 'Send Test to BIP-44 Snap',
+          tag: 'button',
+        });
+
+        // check the results of the public key test
+        await driver.delay(2000);
+        const bip44Result = await driver.findElement('.bip44Result');
+        assert.equal(
+          await bip44Result.getText(),
+          'Public key: "0x86debb44fb3a984d93f326131d4c1db0bc39644f1a67b673b3ab45941a1cea6a385981755185ac4594b6521e4d1e8d1"',
+        );
+      },
+    );
+  });
+});
diff --git a/test/e2e/webdriver/driver.js b/test/e2e/webdriver/driver.js
index 068478741..01a85457a 100644
--- a/test/e2e/webdriver/driver.js
+++ b/test/e2e/webdriver/driver.js
@@ -53,6 +53,7 @@ class Driver {
     this.Key = {
       BACK_SPACE: '\uE003',
       ENTER: '\uE007',
+      SPACE: '\uE00D',
     };
   }