import { mock, MockProxy } from "jest-mock-extended"; import { PinServiceAbstraction } from "@bitwarden/auth/common"; import { AccountService } from "@bitwarden/common/auth/abstractions/account.service"; import { CryptoService } from "@bitwarden/common/platform/abstractions/crypto.service"; import { EncryptService } from "@bitwarden/common/platform/abstractions/encrypt.service"; import { I18nService } from "@bitwarden/common/platform/abstractions/i18n.service"; import { KdfType } from "@bitwarden/common/platform/enums"; import { Utils } from "@bitwarden/common/platform/misc/utils"; import { CipherService } from "@bitwarden/common/vault/abstractions/cipher.service"; import { BitwardenPasswordProtectedImporter, BitwardenJsonImporter, } from "../src/importers/bitwarden"; import { emptyAccountEncrypted } from "./test-data/bitwarden-json/account-encrypted.json"; import { emptyUnencryptedExport } from "./test-data/bitwarden-json/unencrypted.json"; describe("BitwardenPasswordProtectedImporter", () => { let importer: BitwardenPasswordProtectedImporter; let cryptoService: MockProxy; let encryptService: MockProxy; let i18nService: MockProxy; let cipherService: MockProxy; let pinService: MockProxy; let accountService: MockProxy; const password = Utils.newGuid(); const promptForPassword_callback = async () => { return password; }; beforeEach(() => { cryptoService = mock(); encryptService = mock(); i18nService = mock(); cipherService = mock(); pinService = mock(); accountService = mock(); importer = new BitwardenPasswordProtectedImporter( cryptoService, encryptService, i18nService, cipherService, pinService, accountService, promptForPassword_callback, ); }); describe("Unencrypted", () => { beforeAll(() => { jest.spyOn(BitwardenJsonImporter.prototype, "parse"); }); it("Should call BitwardenJsonImporter", async () => { expect((await importer.parse(emptyUnencryptedExport)).success).toEqual(true); expect(BitwardenJsonImporter.prototype.parse).toHaveBeenCalledWith(emptyUnencryptedExport); }); }); describe("Account encrypted", () => { beforeAll(() => { jest.spyOn(BitwardenJsonImporter.prototype, "parse"); }); it("Should call BitwardenJsonImporter", async () => { expect((await importer.parse(emptyAccountEncrypted)).success).toEqual(true); expect(BitwardenJsonImporter.prototype.parse).toHaveBeenCalledWith(emptyAccountEncrypted); }); }); describe("Password protected", () => { let jDoc: { encrypted?: boolean; passwordProtected?: boolean; salt?: string; kdfIterations?: any; kdfType?: any; encKeyValidation_DO_NOT_EDIT?: string; data?: string; }; beforeEach(() => { jDoc = { encrypted: true, passwordProtected: true, salt: "c2FsdA==", kdfIterations: 100000, kdfType: KdfType.PBKDF2_SHA256, encKeyValidation_DO_NOT_EDIT: Utils.newGuid(), data: Utils.newGuid(), }; }); it("succeeds with default jdoc", async () => { encryptService.decryptToUtf8.mockReturnValue(Promise.resolve(emptyUnencryptedExport)); expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(true); }); it("fails if salt === null", async () => { jDoc.salt = null; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); it("fails if kdfIterations === null", async () => { jDoc.kdfIterations = null; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); it("fails if kdfIterations is not a number", async () => { jDoc.kdfIterations = "not a number"; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); it("fails if kdfType === null", async () => { jDoc.kdfType = null; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); it("fails if kdfType is not a string", async () => { jDoc.kdfType = "not a valid kdf type"; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); it("fails if kdfType is not a known kdfType", async () => { jDoc.kdfType = -1; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); it("fails if encKeyValidation_DO_NOT_EDIT === null", async () => { jDoc.encKeyValidation_DO_NOT_EDIT = null; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); it("fails if data === null", async () => { jDoc.data = null; expect((await importer.parse(JSON.stringify(jDoc))).success).toEqual(false); }); }); });