From 28c07b5020f8305804a383efe4ebfe8b12684012 Mon Sep 17 00:00:00 2001
From: Alex Donesky <adonesky@gmail.com>
Date: Thu, 24 Feb 2022 10:37:37 -0600
Subject: [PATCH] add e2e test: It should be possible to remove an account
 imported with a private key, but should not be possible to remove an account
 generated from the SRP imported in onboarding (#13741)

---
 test/e2e/tests/add-account.spec.js | 60 ++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/test/e2e/tests/add-account.spec.js b/test/e2e/tests/add-account.spec.js
index 3c38f8e72..5c0d3fcb9 100644
--- a/test/e2e/tests/add-account.spec.js
+++ b/test/e2e/tests/add-account.spec.js
@@ -195,4 +195,64 @@ describe('Add account', function () {
       },
     );
   });
+
+  it('It should be possible to remove an account imported with a private key, but should not be possible to remove an account generated from the SRP imported in onboarding', async function () {
+    const testPrivateKey =
+      '14abe6f4aab7f9f626fe981c864d0adeb5685f289ac9270c27b8fd790b4235d6';
+
+    await withFixtures(
+      {
+        fixtures: 'imported-account',
+        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.delay(regularDelayMs);
+
+        await driver.clickElement('.account-menu__icon');
+        await driver.clickElement({ text: 'Create Account', tag: 'div' });
+        await driver.fill('.new-account-create-form input', '2nd account');
+        await driver.clickElement({ text: 'Create', tag: 'button' });
+
+        await driver.clickElement(
+          '[data-testid="account-options-menu-button"]',
+        );
+
+        const menuItems = await driver.findElements('.menu-item');
+        assert.equal(menuItems.length, 3);
+
+        // click out of menu
+        await driver.clickElement('.menu__background');
+
+        // import with private key
+        await driver.clickElement('.account-menu__icon');
+        await driver.clickElement({ text: 'Import Account', tag: 'div' });
+
+        // enter private key',
+        await driver.fill('#private-key-box', testPrivateKey);
+        await driver.clickElement({ text: 'Import', tag: 'button' });
+
+        // should show the correct account name
+        const importedAccountName = await driver.findElement(
+          '.selected-account__name',
+        );
+        assert.equal(await importedAccountName.getText(), 'Account 3');
+
+        await driver.clickElement(
+          '[data-testid="account-options-menu-button"]',
+        );
+
+        const menuItems2 = await driver.findElements('.menu-item');
+        assert.equal(menuItems2.length, 4);
+
+        await driver.findElement(
+          '[data-testid="account-options-menu__remove-account"]',
+        );
+      },
+    );
+  });
 });