From 637d4bcf2cc1a20395daa525631ca7a36071faf7 Mon Sep 17 00:00:00 2001
From: Jyoti Puri <jyotipuri@gmail.com>
Date: Fri, 19 Aug 2022 19:13:08 +0530
Subject: [PATCH] Adding unit test case to for idempotent behaviour of
 importAccountWithStrategy idempotent (#15583)

---
 .../metamask-controller.actions.test.js       | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/app/scripts/metamask-controller.actions.test.js b/app/scripts/metamask-controller.actions.test.js
index 7d56007b3..62902e5cb 100644
--- a/app/scripts/metamask-controller.actions.test.js
+++ b/app/scripts/metamask-controller.actions.test.js
@@ -32,6 +32,9 @@ const createLoggerMiddlewareMock = () => (req, res, next) => {
   next();
 };
 
+const TEST_SEED =
+  'debris dizzy just program just float decrease vacant alarm reduce speak stadium';
+
 const MetaMaskController = proxyquire('./metamask-controller', {
   './lib/createLoggerMiddleware': { default: createLoggerMiddlewareMock },
 }).default;
@@ -106,6 +109,63 @@ describe('MetaMaskController', function () {
     });
   });
 
+  describe('#importAccountWithStrategy', function () {
+    it('two sequential calls with same strategy give same result', async function () {
+      let keyringControllerState1;
+      let keyringControllerState2;
+      const importPrivkey =
+        '4cfd3e90fc78b0f86bf7524722150bb8da9c60cd532564d7ff43f5716514f553';
+
+      await metamaskController.createNewVaultAndKeychain('test@123');
+      await Promise.all([
+        metamaskController.importAccountWithStrategy('Private Key', [
+          importPrivkey,
+        ]),
+        Promise.resolve(1).then(() => {
+          keyringControllerState1 = JSON.stringify(
+            metamaskController.keyringController.memStore.getState(),
+          );
+          metamaskController.importAccountWithStrategy('Private Key', [
+            importPrivkey,
+          ]);
+        }),
+        Promise.resolve(2).then(() => {
+          keyringControllerState2 = JSON.stringify(
+            metamaskController.keyringController.memStore.getState(),
+          );
+        }),
+      ]);
+      assert.deepEqual(keyringControllerState1, keyringControllerState2);
+    });
+  });
+
+  describe('#createNewVaultAndRestore', function () {
+    it('two successive calls with same inputs give same result', async function () {
+      const result1 = await metamaskController.createNewVaultAndRestore(
+        'test@123',
+        TEST_SEED,
+      );
+      const result2 = await metamaskController.createNewVaultAndRestore(
+        'test@123',
+        TEST_SEED,
+      );
+      assert.deepEqual(result1, result2);
+    });
+  });
+
+  describe('#createNewVaultAndKeychain', function () {
+    it('two successive calls with same inputs give same result', async function () {
+      const result1 = await metamaskController.createNewVaultAndKeychain(
+        'test@123',
+      );
+      const result2 = await metamaskController.createNewVaultAndKeychain(
+        'test@123',
+      );
+      assert.notEqual(result1, undefined);
+      assert.deepEqual(result1, result2);
+    });
+  });
+
   describe('#addToken', function () {
     const address = '0x514910771af9ca656af840dff83e8264ecf986ca';
     const symbol = 'LINK';