From 950e3274775263252b39a57c91f0197c989e5c5b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Carlos=20Gon=C3=A7alves?= <cgoncalves@bitwarden.com>
Date: Tue, 24 Oct 2023 14:42:25 +0100
Subject: [PATCH] PM-2365 Added a helper function (#6508)

---
 .../src/vault/services/cipher.service.spec.ts | 26 ++++++++-----------
 1 file changed, 11 insertions(+), 15 deletions(-)

diff --git a/libs/common/src/vault/services/cipher.service.spec.ts b/libs/common/src/vault/services/cipher.service.spec.ts
index 2c9adce553..755397ebfb 100644
--- a/libs/common/src/vault/services/cipher.service.spec.ts
+++ b/libs/common/src/vault/services/cipher.service.spec.ts
@@ -149,9 +149,7 @@ describe("Cipher Service", () => {
       );
 
       configService.checkServerMeetsVersionRequirement$.mockReturnValue(of(false));
-      process.env.FLAGS = JSON.stringify({
-        enableCipherKeyEncryption: false,
-      });
+      setEncryptionKeyFlag(false);
 
       const spy = jest.spyOn(cipherFileUploadService, "upload");
 
@@ -262,9 +260,7 @@ describe("Cipher Service", () => {
 
     describe("cipher.key", () => {
       it("is null when enableCipherKeyEncryption flag is false", async () => {
-        process.env.FLAGS = JSON.stringify({
-          enableCipherKeyEncryption: false,
-        });
+        setEncryptionKeyFlag(false);
 
         const cipher = await cipherService.encrypt(cipherView);
 
@@ -272,9 +268,7 @@ describe("Cipher Service", () => {
       });
 
       it("is defined when enableCipherKeyEncryption flag is true", async () => {
-        process.env.FLAGS = JSON.stringify({
-          enableCipherKeyEncryption: true,
-        });
+        setEncryptionKeyFlag(true);
 
         const cipher = await cipherService.encrypt(cipherView);
 
@@ -288,9 +282,7 @@ describe("Cipher Service", () => {
       });
 
       it("is not called when enableCipherKeyEncryption is false", async () => {
-        process.env.FLAGS = JSON.stringify({
-          enableCipherKeyEncryption: false,
-        });
+        setEncryptionKeyFlag(false);
 
         await cipherService.encrypt(cipherView);
 
@@ -298,9 +290,7 @@ describe("Cipher Service", () => {
       });
 
       it("is called when enableCipherKeyEncryption is true", async () => {
-        process.env.FLAGS = JSON.stringify({
-          enableCipherKeyEncryption: true,
-        });
+        setEncryptionKeyFlag(true);
 
         await cipherService.encrypt(cipherView);
 
@@ -309,3 +299,9 @@ describe("Cipher Service", () => {
     });
   });
 });
+
+function setEncryptionKeyFlag(value: boolean) {
+  process.env.FLAGS = JSON.stringify({
+    enableCipherKeyEncryption: value,
+  });
+}