From 290dbd77ce7fae24e5dbea9b041744695dc2ca1a Mon Sep 17 00:00:00 2001
From: Bowen Sanders <bowensanders@gmail.com>
Date: Tue, 23 May 2023 02:54:31 -0700
Subject: [PATCH] [FLASK] Create E2E test for snaps network_access endowment
 (#19213)

* create networkaccess e2e

* final changes

* fixed enums

* requested changes

* fixed result test
---
 test/e2e/snaps/enums.js                       |   2 +-
 .../e2e/snaps/test-snap-networkaccess.spec.js | 106 ++++++++++++++++++
 2 files changed, 107 insertions(+), 1 deletion(-)
 create mode 100644 test/e2e/snaps/test-snap-networkaccess.spec.js

diff --git a/test/e2e/snaps/enums.js b/test/e2e/snaps/enums.js
index 10c49428d..1e5558cf2 100644
--- a/test/e2e/snaps/enums.js
+++ b/test/e2e/snaps/enums.js
@@ -1,3 +1,3 @@
 module.exports = {
-  TEST_SNAPS_WEBSITE_URL: 'https://metamask.github.io/test-snaps/5.4.0/',
+  TEST_SNAPS_WEBSITE_URL: 'https://metamask.github.io/test-snaps/5.5.0/',
 };
diff --git a/test/e2e/snaps/test-snap-networkaccess.spec.js b/test/e2e/snaps/test-snap-networkaccess.spec.js
new file mode 100644
index 000000000..8ea925712
--- /dev/null
+++ b/test/e2e/snaps/test-snap-networkaccess.spec.js
@@ -0,0 +1,106 @@
+const { strict: assert } = require('assert');
+const { withFixtures } = require('../helpers');
+const FixtureBuilder = require('../fixture-builder');
+const { TEST_SNAPS_WEBSITE_URL } = require('./enums');
+
+describe('Test Snap networkAccess', function () {
+  it('test the network-access endowment', async function () {
+    const ganacheOptions = {
+      accounts: [
+        {
+          secretKey:
+            '0x7C9529A67102755B7E6102D6D950AC5D5863C98713805CEC576B945B15B71EAC',
+          balance: 25000000000000000000,
+        },
+      ],
+    };
+    await withFixtures(
+      {
+        fixtures: new FixtureBuilder().build(),
+        ganacheOptions,
+        failOnConsoleError: false,
+        title: this.test.title,
+      },
+      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 to dialog snap
+        await driver.openNewPage(TEST_SNAPS_WEBSITE_URL);
+        await driver.delay(1000);
+        const dialogButton = await driver.findElement(
+          '#connectNetworkAccessSnap',
+        );
+        await driver.scrollToElement(dialogButton);
+        await driver.delay(1000);
+        await driver.clickElement('#connectNetworkAccessSnap');
+        await driver.delay(1000);
+
+        // switch to metamask extension and click connect
+        let windowHandles = await driver.waitUntilXWindowHandles(
+          3,
+          1000,
+          10000,
+        );
+        await driver.switchToWindowWithTitle(
+          'MetaMask Notification',
+          windowHandles,
+        );
+        await driver.clickElement({
+          text: 'Connect',
+          tag: 'button',
+        });
+
+        await driver.waitForSelector({ text: 'Approve & install' });
+
+        await driver.clickElement({
+          text: 'Approve & install',
+          tag: 'button',
+        });
+
+        await driver.waitForSelector({ text: 'Ok' });
+
+        await driver.clickElement({
+          text: 'Ok',
+          tag: 'button',
+        });
+
+        // switch to test snaps tab
+        await driver.switchToWindowWithTitle('Test Snaps', windowHandles);
+
+        // wait for npm installation success
+        await driver.waitForSelector({
+          css: '#connectNetworkAccessSnap',
+          text: 'Reconnect to networkAccess Snap',
+        });
+
+        // click on alert dialog
+        await driver.clickElement('#sendNetworkAccessTest');
+        await driver.delay(500);
+
+        // switch to dialog popup
+        windowHandles = await driver.waitUntilXWindowHandles(3, 1000, 10000);
+        await driver.switchToWindowWithTitle(
+          'MetaMask Notification',
+          windowHandles,
+        );
+        await driver.delay(500);
+
+        // check dialog contents
+        const result = await driver.findElement('.snap-ui-renderer__panel');
+        await driver.scrollToElement(result);
+        await driver.delay(500);
+        assert.equal(await result.getText(), 'FETCHED_SUCCESSFULLY');
+
+        // click ok button
+        await driver.clickElement({
+          text: 'Ok',
+          tag: 'button',
+        });
+      },
+    );
+  });
+});