import Substitute, { Arg } from "@fluffy-spoon/substitute"; import { CryptoService } from "jslib-common/abstractions/crypto.service"; import { EncryptionType } from "jslib-common/enums/encryptionType"; import { EncString } from "jslib-common/models/domain/encString"; import { SymmetricCryptoKey } from "jslib-common/models/domain/symmetricCryptoKey"; import { ContainerService } from "jslib-common/services/container.service"; describe("EncString", () => { afterEach(() => { (window as any).bitwardenContainerService = undefined; }); describe("Rsa2048_OaepSha256_B64", () => { it("constructor", () => { const encString = new EncString(EncryptionType.Rsa2048_OaepSha256_B64, "data"); expect(encString).toEqual({ data: "data", encryptedString: "3.data", encryptionType: 3, }); }); describe("parse existing", () => { it("valid", () => { const encString = new EncString("3.data"); expect(encString).toEqual({ data: "data", encryptedString: "3.data", encryptionType: 3, }); }); it("invalid", () => { const encString = new EncString("3.data|test"); expect(encString).toEqual({ encryptedString: "3.data|test", encryptionType: 3, }); }); }); describe("decrypt", () => { const encString = new EncString(EncryptionType.Rsa2048_OaepSha256_B64, "data"); const cryptoService = Substitute.for(); cryptoService.getOrgKey(null).resolves(null); cryptoService.decryptToUtf8(encString, Arg.any()).resolves("decrypted"); beforeEach(() => { (window as any).bitwardenContainerService = new ContainerService(cryptoService); }); it("decrypts correctly", async () => { const decrypted = await encString.decrypt(null); expect(decrypted).toBe("decrypted"); }); it("result should be cached", async () => { const decrypted = await encString.decrypt(null); cryptoService.received(1).decryptToUtf8(Arg.any(), Arg.any()); expect(decrypted).toBe("decrypted"); }); }); }); describe("AesCbc256_B64", () => { it("constructor", () => { const encString = new EncString(EncryptionType.AesCbc256_B64, "data", "iv"); expect(encString).toEqual({ data: "data", encryptedString: "0.iv|data", encryptionType: 0, iv: "iv", }); }); describe("parse existing", () => { it("valid", () => { const encString = new EncString("0.iv|data"); expect(encString).toEqual({ data: "data", encryptedString: "0.iv|data", encryptionType: 0, iv: "iv", }); }); it("invalid", () => { const encString = new EncString("0.iv|data|mac"); expect(encString).toEqual({ encryptedString: "0.iv|data|mac", encryptionType: 0, }); }); }); }); describe("AesCbc256_HmacSha256_B64", () => { it("constructor", () => { const encString = new EncString(EncryptionType.AesCbc256_HmacSha256_B64, "data", "iv", "mac"); expect(encString).toEqual({ data: "data", encryptedString: "2.iv|data|mac", encryptionType: 2, iv: "iv", mac: "mac", }); }); it("valid", () => { const encString = new EncString("2.iv|data|mac"); expect(encString).toEqual({ data: "data", encryptedString: "2.iv|data|mac", encryptionType: 2, iv: "iv", mac: "mac", }); }); it("invalid", () => { const encString = new EncString("2.iv|data"); expect(encString).toEqual({ encryptedString: "2.iv|data", encryptionType: 2, }); }); }); it("Exit early if null", () => { const encString = new EncString(null); expect(encString).toEqual({ encryptedString: null, }); }); describe("decrypt", () => { it("throws exception when bitwarden container not initialized", async () => { const encString = new EncString(null); expect.assertions(1); try { await encString.decrypt(null); } catch (e) { expect(e.message).toEqual("global bitwardenContainerService not initialized."); } }); it("handles value it can't decrypt", async () => { const encString = new EncString(null); const cryptoService = Substitute.for(); cryptoService.getOrgKey(null).resolves(null); cryptoService.decryptToUtf8(encString, Arg.any()).throws("error"); (window as any).bitwardenContainerService = new ContainerService(cryptoService); const decrypted = await encString.decrypt(null); expect(decrypted).toBe("[error: cannot decrypt]"); expect(encString).toEqual({ decryptedValue: "[error: cannot decrypt]", encryptedString: null, }); }); it("passes along key", async () => { const encString = new EncString(null); const key = Substitute.for(); const cryptoService = Substitute.for(); cryptoService.getOrgKey(null).resolves(null); (window as any).bitwardenContainerService = new ContainerService(cryptoService); await encString.decrypt(null, key); cryptoService.received().decryptToUtf8(encString, key); }); }); });