From f54840277758d2962f1ca72793ff427d7e89a565 Mon Sep 17 00:00:00 2001
From: Thomas Rittson <trittson@bitwarden.com>
Date: Wed, 15 Dec 2021 11:22:15 +1000
Subject: [PATCH] Add test for existing keyConnector user

---
 spec/common/services/auth.service.spec.ts | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/spec/common/services/auth.service.spec.ts b/spec/common/services/auth.service.spec.ts
index ac94d5ee..24213d94 100644
--- a/spec/common/services/auth.service.spec.ts
+++ b/spec/common/services/auth.service.spec.ts
@@ -136,7 +136,7 @@ describe('Cipher Service', () => {
         return tokenResponse;
     }
 
-    it('logIn: simple call: no 2FA, captcha or password reset', async () => {
+    it('logIn: works in the most simple case (no 2FA, no captcha, no password reset, no KC)', async () => {
         logInSetup();
         commonSetup();
         const tokenResponse = newTokenResponse();
@@ -214,8 +214,7 @@ describe('Cipher Service', () => {
         apiService.postIdentityToken(Arg.any()).resolves(tokenResponse);
 
         // Re-init authService with setCryptoKeys = false
-        authService = new AuthService(cryptoService, apiService, tokenService, appIdService, i18nService,
-            platformUtilsService, messagingService, vaultTimeoutService, logService, cryptoFunctionService,
+        authService = new AuthService(cryptoService, apiService, tokenService, appIdService, i18nService, platformUtilsService, messagingService, vaultTimeoutService, logService, cryptoFunctionService,
             keyConnectorService, environmentService, stateService, false);
         authService.init();
 
@@ -229,4 +228,20 @@ describe('Cipher Service', () => {
         cryptoService.didNotReceive().setEncKey(Arg.any());
         cryptoService.didNotReceive().setEncPrivateKey(Arg.any());
     });
+
+    it('logIn: gets and sets KeyConnector key for enrolled user', async () => {
+        const keyConnectorUrl = 'KEY_CONNECTOR_URL';
+        logInSetup();
+        commonSetup();
+        const tokenResponse = newTokenResponse();
+        tokenResponse.keyConnectorUrl = keyConnectorUrl;
+
+        tokenService.getTwoFactorToken(email).resolves(null);
+        apiService.postIdentityToken(Arg.any()).resolves(tokenResponse);
+
+        const result = await authService.logIn(email, masterPassword);
+
+        commonSuccessAssertions();
+        keyConnectorService.received(1).getAndSetKey(keyConnectorUrl);
+    })
 });