Skip to content
This repository has been archived by the owner on Sep 11, 2024. It is now read-only.

Commit

Permalink
Use the non-deprecated importRoomKeys method on CryptoApi
Browse files Browse the repository at this point in the history
  • Loading branch information
andybalaam committed Feb 6, 2024
1 parent a664172 commit 0c56025
Show file tree
Hide file tree
Showing 2 changed files with 30 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,7 @@ export default class ImportE2eKeysDialog extends React.Component<IProps, IState>
return MegolmExportEncryption.decryptMegolmKeyFile(arrayBuffer, passphrase);
})
.then((keys) => {
return this.props.matrixClient.importRoomKeys(JSON.parse(keys));
return this.props.matrixClient.getCrypto()?.importRoomKeys(JSON.parse(keys));
})
.then(() => {
// TODO: it would probably be nice to give some feedback about what we've imported here.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@ limitations under the License.
*/

import React from "react";
import { fireEvent, render } from "@testing-library/react";
import { fireEvent, render, waitFor } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import { CryptoApi } from "matrix-js-sdk/src/matrix";

import ImportE2eKeysDialog from "../../../../../src/async-components/views/dialogs/security/ImportE2eKeysDialog";
import * as MegolmExportEncryption from "../../../../../src/utils/MegolmExportEncryption";
import { createTestClient } from "../../../../test-utils";

describe("ImportE2eKeysDialog", () => {
Expand Down Expand Up @@ -64,4 +66,30 @@ describe("ImportE2eKeysDialog", () => {
await userEvent.paste("passphrase");
expect(container.querySelector("[type=submit]")!).toBeEnabled();
});

it("should import exported keys on submit", async () => {
const cli = createTestClient();
const onFinished = jest.fn();
const file = new File(["test"], "file.txt", { type: "text/plain" });
const importRoomKeys = jest.fn();
cli.getCrypto = () => {
return {
importRoomKeys,
} as unknown as CryptoApi;
};

// Mock the result of decrypting the sessions, to avoid needing to
// create encrypted input data.
jest.spyOn(MegolmExportEncryption, "decryptMegolmKeyFile").mockResolvedValue("[]");

const { container } = render(<ImportE2eKeysDialog matrixClient={cli} onFinished={onFinished} />);
fireEvent.change(container.querySelector("[type=file]")!, {
target: { files: [file] },
});
await userEvent.click(container.querySelector("[type=password]")!);
await userEvent.paste("passphrase");
fireEvent.click(container.querySelector("[type=submit]")!);

await waitFor(() => expect(importRoomKeys).toHaveBeenCalled());
});
});

0 comments on commit 0c56025

Please sign in to comment.