(wip) add envelope encryption tests

Author: matthiasgeihs

packages/keyring-controller/src/KeyringController.test.ts

@@ -796,6 +796,60 @@ describe('KeyringController', () => {
     );
   });
 
+  describe('envelope encryption mode', () => {
+    it('should create new vault with encryption key', async () => {
+      await withController(
+        { cacheEncryptionKey: true, skipVaultCreation: true },
+        async ({ controller }) => {
+          await controller.createNewVaultAndKeychain(
+            password,
+            MOCK_ENCRYPTION_KEY,
+          );
+
+          expect(controller.state.encryptionKey).toBeDefined();
+          // expect(controller.state.encryptionSalt).toBeDefined();
+          expect(controller.state.encryptedEncryptionKey).toBeDefined();
+        },
+      );
+    });
+
+    it('should unlock with password', async () => {
+      await withController(
+        { cacheEncryptionKey: true, skipVaultCreation: true },
+        async ({ controller }) => {
+          await controller.createNewVaultAndKeychain(
+            password,
+            MOCK_ENCRYPTION_KEY,
+          );
+
+          expect(controller.isUnlocked()).toBe(true);
+          expect(controller.state.isUnlocked).toBe(true);
+
+          await controller.setLocked();
+
+          expect(controller.isUnlocked()).toBe(false);
+          expect(controller.state.isUnlocked).toBe(false);
+
+          await controller.submitPassword(password);
+
+          expect(controller.isUnlocked()).toBe(true);
+          expect(controller.state.isUnlocked).toBe(true);
+        },
+      );
+    });
+
+    it('should throw error if creating new vault with encryption key and cacheEncryptionKey is false', async () => {
+      await withController(
+        { cacheEncryptionKey: false, skipVaultCreation: true },
+        async ({ controller }) => {
+          await expect(
+            controller.createNewVaultAndKeychain(password, MOCK_ENCRYPTION_KEY),
+          ).rejects.toThrow(KeyringControllerError.CacheEncryptionKeyDisabled);
+        },
+      );
+    });
+  });
+
   describe('setLocked', () => {
     it('should set locked correctly', async () => {
       await withController(async ({ controller }) => {