[FLASK] Create E2E test for snap management (#16628)

* added management test

* added id and dismiss alert function

* changed xwindow handling

* fixed linting

test/e2e/snaps/test-snap-management.spec.js

@@ -0,0 +1,154 @@
+const { strict: assert } = require('assert');
+const { withFixtures } = require('../helpers');
+const FixtureBuilder = require('../fixture-builder');
+const { TEST_SNAPS_WEBSITE_URL } = require('./enums');
+
+describe('Test Snap onOff', function () {
+  it('tests install disable enable and removal of a snap', 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);
+
+        // open a new tab and navigate to test snaps page and connect
+        await driver.openNewPage(TEST_SNAPS_WEBSITE_URL);
+
+        // find and scroll to the correct card and click first
+        const snapButton = await driver.findElement('#connectNotification');
+        await driver.scrollToElement(snapButton);
+        await driver.delay(500);
+        await driver.clickElement('#connectNotification');
+
+        // 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',
+          },
+          10000,
+        );
+
+        await driver.delay(1000);
+
+        // approve install of snap
+        windowHandles = await driver.getAllWindowHandles();
+        await driver.switchToWindowWithTitle(
+          'MetaMask Notification',
+          windowHandles,
+        );
+        await driver.clickElement({
+          text: 'Approve & install',
+          tag: 'button',
+        });
+
+        // switch to the original MM tab
+        const extensionPage = windowHandles[0];
+        await driver.switchToWindow(extensionPage);
+        await driver.delay(1000);
+
+        // click on the account menu icon
+        await driver.clickElement('.account-menu__icon');
+        await driver.delay(1000);
+
+        // try to click on the notification item
+        await driver.clickElement({
+          text: 'Settings',
+          tag: 'div',
+        });
+        await driver.delay(1000);
+
+        // try to click on the snaps item
+        await driver.clickElement({
+          text: 'Snaps',
+          tag: 'div',
+        });
+        await driver.delay(1000);
+
+        // try to disable the snap
+        await driver.clickElement('.toggle-button > div');
+
+        // switch back to test-snaps window
+        windowHandles = await driver.waitUntilXWindowHandles(2, 1000, 10000);
+        await driver.switchToWindowWithTitle('Test Snaps', windowHandles);
+
+        // wait then try the notification test
+        await driver.delay(1000);
+        await driver.clickElement('#sendInAppNotification');
+
+        // click OK on the popup
+        await driver.delay(1000);
+        await driver.closeAlertPopup();
+        await driver.delay(1000);
+
+        // switch back to snaps page
+        await driver.switchToWindow(extensionPage);
+        await driver.delay(1000);
+
+        // try to re-enaable the snap
+        await driver.clickElement('.toggle-button > div');
+        await driver.delay(1000);
+
+        // switch back to test snaps page
+        await driver.switchToWindowWithTitle('Test Snaps', windowHandles);
+
+        // wait then try the notification test
+        await driver.delay(1000);
+        await driver.clickElement('#sendInAppNotification');
+
+        // check to see that there is one notification
+        await driver.switchToWindow(extensionPage);
+        await driver.delay(1000);
+        const notificationResult = await driver.findElement(
+          '.account-menu__icon__notification-count',
+        );
+        assert.equal(await notificationResult.getText(), '1');
+
+        // click on see details
+        await driver.clickElement({ text: 'See details', tag: 'button' });
+        await driver.delay(1000);
+
+        // try to remove snap
+        await driver.clickElement({ text: 'Remove Snap', tag: 'button' });
+        await driver.delay(1000);
+
+        // try to click remove on popover
+        await driver.clickElement('#popoverRemoveSnapButton');
+        await driver.delay(1000);
+
+        // check the results of the removal
+        await driver.delay(2000);
+        const removeResult = await driver.findElement(
+          '.snap-list-tab__container--no-snaps',
+        );
+        assert.equal(await removeResult.getText(), 'No Snaps installed');
+      },
+    );
+  });
+});

test/e2e/webdriver/driver.js

@@ -395,6 +395,11 @@ class Driver {
     throw new Error(`No window with title: ${title}`);
   }
 
+  // Close Alert Popup
+  async closeAlertPopup() {
+    return await this.driver.switchTo().alert().accept();
+  }
+
   /**
    * Closes all windows except those in the given list of exceptions
    *

ui/components/app/flask/snap-remove-warning/snap-remove-warning.js

@@ -21,6 +21,7 @@ export default function SnapRemoveWarning({ onCancel, onSubmit, snapName }) {
           {t('nevermind')}
         </Button>
         <Button
+          id="popoverRemoveSnapButton"
           className="snap-remove-warning__footer-button"
           type="danger-primary"
           onClick={onSubmit}