mirror of
https://github.com/bitwarden/browser.git
synced 2024-06-25 10:25:36 +02:00
139 lines
3.7 KiB
TypeScript
139 lines
3.7 KiB
TypeScript
|
import { MockProxy } from "jest-mock-extended";
|
||
|
|
||
|
import { KeyDefinitionLike, MigrationHelper } from "../migration-helper";
|
||
|
import { mockMigrationHelper } from "../migration-helper.spec";
|
||
|
|
||
|
import { AuthRequestMigrator } from "./56-move-auth-requests";
|
||
|
|
||
|
function exampleJSON() {
|
||
|
return {
|
||
|
global: {
|
||
|
otherStuff: "otherStuff1",
|
||
|
},
|
||
|
authenticatedAccounts: ["FirstAccount", "SecondAccount"],
|
||
|
FirstAccount: {
|
||
|
settings: {
|
||
|
otherStuff: "otherStuff2",
|
||
|
approveLoginRequests: true,
|
||
|
},
|
||
|
otherStuff: "otherStuff3",
|
||
|
adminAuthRequest: {
|
||
|
id: "id1",
|
||
|
privateKey: "privateKey1",
|
||
|
},
|
||
|
},
|
||
|
SecondAccount: {
|
||
|
settings: {
|
||
|
otherStuff: "otherStuff4",
|
||
|
},
|
||
|
otherStuff: "otherStuff5",
|
||
|
},
|
||
|
};
|
||
|
}
|
||
|
|
||
|
function rollbackJSON() {
|
||
|
return {
|
||
|
user_FirstAccount_authRequestLocal_adminAuthRequest: {
|
||
|
id: "id1",
|
||
|
privateKey: "privateKey1",
|
||
|
},
|
||
|
user_FirstAccount_authRequestLocal_acceptAuthRequests: true,
|
||
|
global: {
|
||
|
otherStuff: "otherStuff1",
|
||
|
},
|
||
|
authenticatedAccounts: ["FirstAccount", "SecondAccount"],
|
||
|
FirstAccount: {
|
||
|
settings: {
|
||
|
otherStuff: "otherStuff2",
|
||
|
},
|
||
|
otherStuff: "otherStuff3",
|
||
|
},
|
||
|
SecondAccount: {
|
||
|
settings: {
|
||
|
otherStuff: "otherStuff4",
|
||
|
},
|
||
|
otherStuff: "otherStuff5",
|
||
|
},
|
||
|
};
|
||
|
}
|
||
|
|
||
|
const ADMIN_AUTH_REQUEST_KEY: KeyDefinitionLike = {
|
||
|
stateDefinition: {
|
||
|
name: "authRequestLocal",
|
||
|
},
|
||
|
key: "adminAuthRequest",
|
||
|
};
|
||
|
|
||
|
const ACCEPT_AUTH_REQUESTS_KEY: KeyDefinitionLike = {
|
||
|
stateDefinition: {
|
||
|
name: "authRequestLocal",
|
||
|
},
|
||
|
key: "acceptAuthRequests",
|
||
|
};
|
||
|
|
||
|
describe("AuthRequestMigrator", () => {
|
||
|
let helper: MockProxy<MigrationHelper>;
|
||
|
let sut: AuthRequestMigrator;
|
||
|
|
||
|
describe("migrate", () => {
|
||
|
beforeEach(() => {
|
||
|
helper = mockMigrationHelper(exampleJSON(), 55);
|
||
|
sut = new AuthRequestMigrator(55, 56);
|
||
|
});
|
||
|
|
||
|
it("removes the existing adminAuthRequest and approveLoginRequests", async () => {
|
||
|
await sut.migrate(helper);
|
||
|
|
||
|
expect(helper.set).toHaveBeenCalledWith("FirstAccount", {
|
||
|
settings: {
|
||
|
otherStuff: "otherStuff2",
|
||
|
},
|
||
|
otherStuff: "otherStuff3",
|
||
|
});
|
||
|
expect(helper.set).not.toHaveBeenCalledWith("SecondAccount");
|
||
|
});
|
||
|
|
||
|
it("sets the adminAuthRequest and approveLoginRequests under the new key definitions", async () => {
|
||
|
await sut.migrate(helper);
|
||
|
|
||
|
expect(helper.setToUser).toHaveBeenCalledWith("FirstAccount", ADMIN_AUTH_REQUEST_KEY, {
|
||
|
id: "id1",
|
||
|
privateKey: "privateKey1",
|
||
|
});
|
||
|
|
||
|
expect(helper.setToUser).toHaveBeenCalledWith("FirstAccount", ACCEPT_AUTH_REQUESTS_KEY, true);
|
||
|
expect(helper.setToUser).not.toHaveBeenCalledWith("SecondAccount");
|
||
|
});
|
||
|
});
|
||
|
|
||
|
describe("rollback", () => {
|
||
|
beforeEach(() => {
|
||
|
helper = mockMigrationHelper(rollbackJSON(), 56);
|
||
|
sut = new AuthRequestMigrator(55, 56);
|
||
|
});
|
||
|
|
||
|
it("nulls the new adminAuthRequest and acceptAuthRequests values", async () => {
|
||
|
await sut.rollback(helper);
|
||
|
|
||
|
expect(helper.setToUser).toHaveBeenCalledWith("FirstAccount", ADMIN_AUTH_REQUEST_KEY, null);
|
||
|
expect(helper.setToUser).toHaveBeenCalledWith("FirstAccount", ACCEPT_AUTH_REQUESTS_KEY, null);
|
||
|
});
|
||
|
|
||
|
it("sets back the adminAuthRequest and approveLoginRequests under old account object", async () => {
|
||
|
await sut.rollback(helper);
|
||
|
|
||
|
expect(helper.set).toHaveBeenCalledWith("FirstAccount", {
|
||
|
adminAuthRequest: {
|
||
|
id: "id1",
|
||
|
privateKey: "privateKey1",
|
||
|
},
|
||
|
settings: {
|
||
|
otherStuff: "otherStuff2",
|
||
|
approveLoginRequests: true,
|
||
|
},
|
||
|
otherStuff: "otherStuff3",
|
||
|
});
|
||
|
});
|
||
|
});
|
||
|
});
|