268 lines
9.6 KiB
TypeScript
268 lines
9.6 KiB
TypeScript
import { mock, MockProxy } from "jest-mock-extended";
|
|
|
|
import { CryptoService } from "@bitwarden/common/platform/abstractions/crypto.service";
|
|
import { I18nService } from "@bitwarden/common/platform/abstractions/i18n.service";
|
|
import { Utils } from "@bitwarden/common/platform/misc/utils";
|
|
import { CipherService } from "@bitwarden/common/vault/abstractions/cipher.service";
|
|
import { CollectionService } from "@bitwarden/common/vault/abstractions/collection.service";
|
|
import { FolderService } from "@bitwarden/common/vault/abstractions/folder/folder.service.abstraction";
|
|
import { CipherView } from "@bitwarden/common/vault/models/view/cipher.view";
|
|
import { CollectionView } from "@bitwarden/common/vault/models/view/collection.view";
|
|
import { FolderView } from "@bitwarden/common/vault/models/view/folder.view";
|
|
|
|
import { BitwardenPasswordProtectedImporter } from "../importers/bitwarden/bitwarden-password-protected-importer";
|
|
import { Importer } from "../importers/importer";
|
|
import { ImportResult } from "../models/import-result";
|
|
|
|
import { ImportApiServiceAbstraction } from "./import-api.service.abstraction";
|
|
import { ImportService } from "./import.service";
|
|
|
|
describe("ImportService", () => {
|
|
let importService: ImportService;
|
|
let cipherService: MockProxy<CipherService>;
|
|
let folderService: MockProxy<FolderService>;
|
|
let importApiService: MockProxy<ImportApiServiceAbstraction>;
|
|
let i18nService: MockProxy<I18nService>;
|
|
let collectionService: MockProxy<CollectionService>;
|
|
let cryptoService: MockProxy<CryptoService>;
|
|
|
|
beforeEach(() => {
|
|
cipherService = mock<CipherService>();
|
|
folderService = mock<FolderService>();
|
|
importApiService = mock<ImportApiServiceAbstraction>();
|
|
i18nService = mock<I18nService>();
|
|
collectionService = mock<CollectionService>();
|
|
cryptoService = mock<CryptoService>();
|
|
|
|
importService = new ImportService(
|
|
cipherService,
|
|
folderService,
|
|
importApiService,
|
|
i18nService,
|
|
collectionService,
|
|
cryptoService,
|
|
);
|
|
});
|
|
|
|
describe("getImporterInstance", () => {
|
|
describe("Get bitPasswordProtected importer", () => {
|
|
let importer: Importer;
|
|
const organizationId = Utils.newGuid();
|
|
const password = Utils.newGuid();
|
|
const promptForPassword_callback = async () => {
|
|
return password;
|
|
};
|
|
|
|
beforeEach(() => {
|
|
importer = importService.getImporter(
|
|
"bitwardenpasswordprotected",
|
|
promptForPassword_callback,
|
|
organizationId,
|
|
);
|
|
});
|
|
|
|
it("returns an instance of BitwardenPasswordProtectedImporter", () => {
|
|
expect(importer).toBeInstanceOf(BitwardenPasswordProtectedImporter);
|
|
});
|
|
|
|
it("has the promptForPassword_callback set", async () => {
|
|
// Cast to any to access private property. Note: assumes instance of BitwardenPasswordProtectedImporter
|
|
expect((importer as any).promptForPassword_callback).not.toBeNull();
|
|
expect(await (importer as any).promptForPassword_callback()).toEqual(password);
|
|
});
|
|
|
|
it("has the appropriate organization Id", () => {
|
|
expect(importer.organizationId).toEqual(organizationId);
|
|
});
|
|
});
|
|
});
|
|
|
|
describe("setImportTarget", () => {
|
|
const organizationId = Utils.newGuid();
|
|
|
|
let importResult: ImportResult;
|
|
|
|
beforeEach(() => {
|
|
importResult = new ImportResult();
|
|
});
|
|
|
|
it("empty importTarget does nothing", async () => {
|
|
await importService["setImportTarget"](importResult, null, null);
|
|
expect(importResult.folders.length).toBe(0);
|
|
});
|
|
|
|
const mockImportTargetFolder = new FolderView();
|
|
mockImportTargetFolder.id = "myImportTarget";
|
|
mockImportTargetFolder.name = "myImportTarget";
|
|
|
|
it("passing importTarget adds it to folders", async () => {
|
|
folderService.getAllDecryptedFromState.mockReturnValue(
|
|
Promise.resolve([mockImportTargetFolder]),
|
|
);
|
|
|
|
await importService["setImportTarget"](importResult, null, mockImportTargetFolder);
|
|
expect(importResult.folders.length).toBe(1);
|
|
expect(importResult.folders[0]).toBe(mockImportTargetFolder);
|
|
});
|
|
|
|
const mockFolder1 = new FolderView();
|
|
mockFolder1.id = "folder1";
|
|
mockFolder1.name = "folder1";
|
|
|
|
const mockFolder2 = new FolderView();
|
|
mockFolder2.id = "folder2";
|
|
mockFolder2.name = "folder2";
|
|
|
|
it("passing importTarget sets it as new root for all existing folders", async () => {
|
|
folderService.getAllDecryptedFromState.mockResolvedValue([
|
|
mockImportTargetFolder,
|
|
mockFolder1,
|
|
mockFolder2,
|
|
]);
|
|
|
|
importResult.folders.push(mockFolder1);
|
|
importResult.folders.push(mockFolder2);
|
|
|
|
await importService["setImportTarget"](importResult, null, mockImportTargetFolder);
|
|
expect(importResult.folders.length).toBe(3);
|
|
expect(importResult.folders[0]).toBe(mockImportTargetFolder);
|
|
expect(importResult.folders[1].name).toBe(
|
|
`${mockImportTargetFolder.name}/${mockFolder1.name}`,
|
|
);
|
|
expect(importResult.folders[2].name).toBe(
|
|
`${mockImportTargetFolder.name}/${mockFolder2.name}`,
|
|
);
|
|
});
|
|
|
|
const mockImportTargetCollection = new CollectionView();
|
|
mockImportTargetCollection.id = "myImportTarget";
|
|
mockImportTargetCollection.name = "myImportTarget";
|
|
mockImportTargetCollection.organizationId = organizationId;
|
|
|
|
const mockCollection1 = new CollectionView();
|
|
mockCollection1.id = "collection1";
|
|
mockCollection1.name = "collection1";
|
|
mockCollection1.organizationId = organizationId;
|
|
|
|
const mockCollection2 = new CollectionView();
|
|
mockCollection1.id = "collection2";
|
|
mockCollection1.name = "collection2";
|
|
mockCollection1.organizationId = organizationId;
|
|
|
|
it("passing importTarget adds it to collections", async () => {
|
|
collectionService.getAllDecrypted.mockResolvedValue([
|
|
mockImportTargetCollection,
|
|
mockCollection1,
|
|
]);
|
|
|
|
await importService["setImportTarget"](
|
|
importResult,
|
|
organizationId,
|
|
mockImportTargetCollection,
|
|
);
|
|
expect(importResult.collections.length).toBe(1);
|
|
expect(importResult.collections[0]).toBe(mockImportTargetCollection);
|
|
});
|
|
|
|
it("passing importTarget sets it as new root for all existing collections", async () => {
|
|
collectionService.getAllDecrypted.mockResolvedValue([
|
|
mockImportTargetCollection,
|
|
mockCollection1,
|
|
mockCollection2,
|
|
]);
|
|
|
|
importResult.collections.push(mockCollection1);
|
|
importResult.collections.push(mockCollection2);
|
|
|
|
await importService["setImportTarget"](
|
|
importResult,
|
|
organizationId,
|
|
mockImportTargetCollection,
|
|
);
|
|
expect(importResult.collections.length).toBe(3);
|
|
expect(importResult.collections[0]).toBe(mockImportTargetCollection);
|
|
expect(importResult.collections[1].name).toBe(
|
|
`${mockImportTargetCollection.name}/${mockCollection1.name}`,
|
|
);
|
|
expect(importResult.collections[2].name).toBe(
|
|
`${mockImportTargetCollection.name}/${mockCollection2.name}`,
|
|
);
|
|
});
|
|
|
|
it("passing importTarget as null on setImportTarget with organizationId throws error", async () => {
|
|
const setImportTargetMethod = importService["setImportTarget"](
|
|
null,
|
|
organizationId,
|
|
new Object() as FolderView,
|
|
);
|
|
|
|
await expect(setImportTargetMethod).rejects.toThrow();
|
|
});
|
|
|
|
it("passing importTarget as null on setImportTarget throws error", async () => {
|
|
const setImportTargetMethod = importService["setImportTarget"](
|
|
null,
|
|
"",
|
|
new Object() as CollectionView,
|
|
);
|
|
|
|
await expect(setImportTargetMethod).rejects.toThrow();
|
|
});
|
|
|
|
it("passing importTarget, collectionRelationship has the expected values", async () => {
|
|
collectionService.getAllDecrypted.mockResolvedValue([
|
|
mockImportTargetCollection,
|
|
mockCollection1,
|
|
mockCollection2,
|
|
]);
|
|
|
|
importResult.ciphers.push(createCipher({ name: "cipher1" }));
|
|
importResult.ciphers.push(createCipher({ name: "cipher2" }));
|
|
importResult.collectionRelationships.push([0, 0]);
|
|
importResult.collections.push(mockCollection1);
|
|
importResult.collections.push(mockCollection2);
|
|
|
|
await importService["setImportTarget"](
|
|
importResult,
|
|
organizationId,
|
|
mockImportTargetCollection,
|
|
);
|
|
expect(importResult.collectionRelationships.length).toEqual(2);
|
|
expect(importResult.collectionRelationships[0]).toEqual([1, 0]);
|
|
expect(importResult.collectionRelationships[1]).toEqual([0, 1]);
|
|
});
|
|
|
|
it("passing importTarget, folderRelationship has the expected values", async () => {
|
|
folderService.getAllDecryptedFromState.mockResolvedValue([
|
|
mockImportTargetFolder,
|
|
mockFolder1,
|
|
mockFolder2,
|
|
]);
|
|
|
|
importResult.folders.push(mockFolder1);
|
|
importResult.folders.push(mockFolder2);
|
|
|
|
importResult.ciphers.push(createCipher({ name: "cipher1", folderId: mockFolder1.id }));
|
|
importResult.ciphers.push(createCipher({ name: "cipher2" }));
|
|
importResult.folderRelationships.push([0, 0]);
|
|
|
|
await importService["setImportTarget"](importResult, "", mockImportTargetFolder);
|
|
expect(importResult.folderRelationships.length).toEqual(2);
|
|
expect(importResult.folderRelationships[0]).toEqual([1, 0]);
|
|
expect(importResult.folderRelationships[1]).toEqual([0, 1]);
|
|
});
|
|
});
|
|
});
|
|
|
|
function createCipher(options: Partial<CipherView> = {}) {
|
|
const cipher = new CipherView();
|
|
|
|
cipher.name;
|
|
cipher.type = options.type;
|
|
cipher.folderId = options.folderId;
|
|
cipher.collectionIds = options.collectionIds;
|
|
cipher.organizationId = options.organizationId;
|
|
|
|
return cipher;
|
|
}
|