riot-web/test/CreateCrossSigning-test.ts

/*
Copyright 2024 New Vector Ltd.
Copyright 2018-2022 The Matrix.org Foundation C.I.C.

SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only
Please see LICENSE files in the repository root for full details.
*/

import { MatrixClient, MatrixError } from "matrix-js-sdk/src/matrix";
import { mocked } from "jest-mock";

import { createCrossSigning } from "../src/CreateCrossSigning";
import { createTestClient } from "./test-utils";
import Modal from "../src/Modal";

describe("CreateCrossSigning", () => {
    let client: MatrixClient;

    beforeEach(() => {
        client = createTestClient();
    });

    it("should call bootstrapCrossSigning with an authUploadDeviceSigningKeys function", async () => {
        await createCrossSigning(client, false, "password");

        expect(client.getCrypto()?.bootstrapCrossSigning).toHaveBeenCalledWith({
            authUploadDeviceSigningKeys: expect.any(Function),
        });
    });

    it("should upload with password auth if possible", async () => {
        client.uploadDeviceSigningKeys = jest.fn().mockRejectedValueOnce(
            new MatrixError({
                flows: [
                    {
                        stages: ["m.login.password"],
                    },
                ],
            }),
        );

        await createCrossSigning(client, false, "password");

        const { authUploadDeviceSigningKeys } = mocked(client.getCrypto()!).bootstrapCrossSigning.mock.calls[0][0];

        const makeRequest = jest.fn();
        await authUploadDeviceSigningKeys!(makeRequest);
        expect(makeRequest).toHaveBeenCalledWith({
            type: "m.login.password",
            identifier: {
                type: "m.id.user",
                user: client.getUserId(),
            },
            password: "password",
        });
    });

    it("should attempt to upload keys without auth if using token login", async () => {
        await createCrossSigning(client, true, undefined);

        const { authUploadDeviceSigningKeys } = mocked(client.getCrypto()!).bootstrapCrossSigning.mock.calls[0][0];

        const makeRequest = jest.fn();
        await authUploadDeviceSigningKeys!(makeRequest);
        expect(makeRequest).toHaveBeenCalledWith({});
    });

    it("should prompt user if password upload not possible", async () => {
        const createDialog = jest.spyOn(Modal, "createDialog").mockReturnValue({
            finished: Promise.resolve([true]),
            close: jest.fn(),
        });

        client.uploadDeviceSigningKeys = jest.fn().mockRejectedValueOnce(
            new MatrixError({
                flows: [
                    {
                        stages: ["dummy.mystery_flow_nobody_knows"],
                    },
                ],
            }),
        );

        await createCrossSigning(client, false, "password");

        const { authUploadDeviceSigningKeys } = mocked(client.getCrypto()!).bootstrapCrossSigning.mock.calls[0][0];

        const makeRequest = jest.fn();
        await authUploadDeviceSigningKeys!(makeRequest);
        expect(makeRequest).not.toHaveBeenCalledWith();
        expect(createDialog).toHaveBeenCalled();
    });
});