86 lines
3.0 KiB
TypeScript
86 lines
3.0 KiB
TypeScript
|
/*
|
||
|
|
Copyright 2024 New Vector Ltd.
|
||
|
|
|
||
|
|
SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only
|
||
|
|
Please see LICENSE files in the repository root for full details.
|
||
|
|
*/
|
||
|
|
|
||
|
|
import { mocked } from "jest-mock";
|
||
|
|
import { MatrixClient } from "matrix-js-sdk/src/matrix";
|
||
|
|
import { waitFor } from "jest-matrix-react";
|
||
|
|
|
||
|
|
import { createCrossSigning } from "../../../src/CreateCrossSigning";
|
||
|
|
import { InitialCryptoSetupStore } from "../../../src/stores/InitialCryptoSetupStore";
|
||
|
|
import { SdkContextClass } from "../../../src/contexts/SDKContext";
|
||
|
|
import { createTestClient } from "../../test-utils";
|
||
|
|
import { AccountPasswordStore } from "../../../src/stores/AccountPasswordStore";
|
||
|
|
|
||
|
|
jest.mock("../../../src/CreateCrossSigning", () => ({
|
||
|
|
createCrossSigning: jest.fn(),
|
||
|
|
}));
|
||
|
|
|
||
|
|
describe("InitialCryptoSetupStore", () => {
|
||
|
|
let testStore: InitialCryptoSetupStore;
|
||
|
|
let client: MatrixClient;
|
||
|
|
let stores: SdkContextClass;
|
||
|
|
|
||
|
|
let createCrossSigningResolve: () => void;
|
||
|
|
let createCrossSigningReject: (e: Error) => void;
|
||
|
|
|
||
|
|
beforeEach(() => {
|
||
|
|
testStore = new InitialCryptoSetupStore();
|
||
|
|
client = createTestClient();
|
||
|
|
stores = {
|
||
|
|
accountPasswordStore: {
|
||
|
|
getPassword: jest.fn(),
|
||
|
|
} as unknown as AccountPasswordStore,
|
||
|
|
} as unknown as SdkContextClass;
|
||
|
|
|
||
|
|
mocked(createCrossSigning).mockImplementation(() => {
|
||
|
|
return new Promise<void>((resolve, reject) => {
|
||
|
|
createCrossSigningResolve = resolve;
|
||
|
|
createCrossSigningReject = reject;
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
it("should call createCrossSigning when startInitialCryptoSetup is called", async () => {
|
||
|
|
testStore.startInitialCryptoSetup(client, false, stores, jest.fn());
|
||
|
|
|
||
|
|
await waitFor(() => expect(createCrossSigning).toHaveBeenCalled());
|
||
|
|
});
|
||
|
|
|
||
|
|
it("emits an update event when createCrossSigning resolves", async () => {
|
||
|
|
const updateSpy = jest.fn();
|
||
|
|
testStore.on("update", updateSpy);
|
||
|
|
|
||
|
|
testStore.startInitialCryptoSetup(client, false, stores, jest.fn());
|
||
|
|
createCrossSigningResolve();
|
||
|
|
|
||
|
|
await waitFor(() => expect(updateSpy).toHaveBeenCalled());
|
||
|
|
expect(testStore.getStatus()).toBe("complete");
|
||
|
|
});
|
||
|
|
|
||
|
|
it("emits an update event when createCrossSigning rejects", async () => {
|
||
|
|
const updateSpy = jest.fn();
|
||
|
|
testStore.on("update", updateSpy);
|
||
|
|
|
||
|
|
testStore.startInitialCryptoSetup(client, false, stores, jest.fn());
|
||
|
|
createCrossSigningReject(new Error("Test error"));
|
||
|
|
|
||
|
|
await waitFor(() => expect(updateSpy).toHaveBeenCalled());
|
||
|
|
expect(testStore.getStatus()).toBe("error");
|
||
|
|
});
|
||
|
|
|
||
|
|
it("should ignore failures if tokenLogin is true", async () => {
|
||
|
|
const updateSpy = jest.fn();
|
||
|
|
testStore.on("update", updateSpy);
|
||
|
|
|
||
|
|
testStore.startInitialCryptoSetup(client, true, stores, jest.fn());
|
||
|
|
createCrossSigningReject(new Error("Test error"));
|
||
|
|
|
||
|
|
await waitFor(() => expect(updateSpy).toHaveBeenCalled());
|
||
|
|
expect(testStore.getStatus()).toBe("complete");
|
||
|
|
});
|
||
|
|
});
|