fixin: testing privateKeyOverride

MatrixAI · Jul 15, 2022 · 4e210ca · 4e210ca
1 parent 2e295da
commit 4e210ca
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/keys/KeyManager.test.ts b/tests/keys/KeyManager.test.ts
@@ -161,6 +161,23 @@ describe('KeyManager', () => {
     },
     global.defaultTimeout * 2,
   );
+  test('override key generation with privateKeyOverride', async () => {
+    const keysPath = `${dataDir}/keys`;
+    const keyPair = await keysUtils.generateKeyPair(4096);
+    const mockedGenerateKeyPair = jest.spyOn(keysUtils, 'generateDeterministicKeyPair');
+    const keyManager = await KeyManager.createKeyManager({
+      keysPath,
+      password,
+      privateKeyOverride: keyPair.privateKey,
+      logger,
+    });
+    expect(mockedGenerateKeyPair).not.toHaveBeenCalled()
+    const keysPathContents = await fs.promises.readdir(keysPath);
+    expect(keysPathContents).toContain('root.pub');
+    expect(keysPathContents).toContain('root.key');
+    expect(keysUtils.publicKeyToPem(keyManager.getRootKeyPair().publicKey)).toEqual(keysUtils.publicKeyToPem(keyPair.publicKey));
+    await keyManager.stop();
+  })
   test('uses WorkerManager for generating root key pair', async () => {
     const keysPath = `${dataDir}/keys`;
     const keyManager = await KeyManager.createKeyManager({