From 4e210caf8505e4571936d323c132c90bf45022bf Mon Sep 17 00:00:00 2001
From: Brian Botha <brian.botha@matrix.ai>
Date: Fri, 15 Jul 2022 18:38:06 +1000
Subject: [PATCH] fixin: testing privateKeyOverride

---
 tests/keys/KeyManager.test.ts | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/tests/keys/KeyManager.test.ts b/tests/keys/KeyManager.test.ts
index c1aaa345e4..c2cbab1886 100644
--- 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({