86 lines
2.9 KiB
TypeScript
86 lines
2.9 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 { sleep } from "matrix-js-sdk/src/utils";
|
|
|
|
import { createCrossSigning } from "../../../src/CreateCrossSigning";
|
|
import { InitialCryptoSetupStore } from "../../../src/stores/InitialCryptoSetupStore";
|
|
import { createTestClient } from "../../test-utils";
|
|
|
|
jest.mock("../../../src/CreateCrossSigning", () => ({
|
|
createCrossSigning: jest.fn(),
|
|
}));
|
|
|
|
describe("InitialCryptoSetupStore", () => {
|
|
let testStore: InitialCryptoSetupStore;
|
|
let client: MatrixClient;
|
|
|
|
let createCrossSigningResolve: () => void;
|
|
let createCrossSigningReject: (e: Error) => void;
|
|
|
|
beforeEach(() => {
|
|
testStore = new InitialCryptoSetupStore();
|
|
client = createTestClient();
|
|
|
|
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, 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, 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, jest.fn());
|
|
createCrossSigningReject(new Error("Test error"));
|
|
|
|
await waitFor(() => expect(updateSpy).toHaveBeenCalled());
|
|
expect(testStore.getStatus()).toBe("error");
|
|
});
|
|
|
|
it("should fail to retry once complete", async () => {
|
|
testStore.startInitialCryptoSetup(client, jest.fn());
|
|
|
|
await waitFor(() => expect(createCrossSigning).toHaveBeenCalled());
|
|
createCrossSigningResolve();
|
|
await sleep(0); // await the next tick
|
|
expect(testStore.retry()).toBeFalsy();
|
|
});
|
|
|
|
it("should retry if initial attempt failed", async () => {
|
|
testStore.startInitialCryptoSetup(client, jest.fn());
|
|
|
|
await waitFor(() => expect(createCrossSigning).toHaveBeenCalled());
|
|
createCrossSigningReject(new Error("Test error"));
|
|
await sleep(0); // await the next tick
|
|
expect(testStore.retry()).toBeTruthy();
|
|
});
|
|
});
|