64 lines
2.3 KiB
TypeScript
64 lines
2.3 KiB
TypeScript
/*
|
|
* Copyright 2024 New Vector Ltd.
|
|
* Copyright 2024 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 React from "react";
|
|
import { mocked } from "jest-mock";
|
|
import { render, fireEvent, waitFor } from "jest-matrix-react";
|
|
import fetchMock from "fetch-mock-jest";
|
|
|
|
import ImageView from "../../../../../src/components/views/elements/ImageView";
|
|
import { FileDownloader } from "../../../../../src/utils/FileDownloader";
|
|
import Modal from "../../../../../src/Modal";
|
|
import ErrorDialog from "../../../../../src/components/views/dialogs/ErrorDialog";
|
|
|
|
jest.mock("../../../../../src/utils/FileDownloader");
|
|
|
|
describe("<ImageView />", () => {
|
|
beforeEach(() => {
|
|
jest.resetAllMocks();
|
|
fetchMock.reset();
|
|
});
|
|
|
|
it("renders correctly", () => {
|
|
const { container } = render(<ImageView src="https://example.com/image.png" onFinished={jest.fn()} />);
|
|
expect(container).toMatchSnapshot();
|
|
});
|
|
|
|
it("should download on click", async () => {
|
|
fetchMock.get("https://example.com/image.png", "TESTFILE");
|
|
const { getByRole } = render(
|
|
<ImageView src="https://example.com/image.png" name="filename.png" onFinished={jest.fn()} />,
|
|
);
|
|
fireEvent.click(getByRole("button", { name: "Download" }));
|
|
await waitFor(() =>
|
|
expect(mocked(FileDownloader).mock.instances[0].download).toHaveBeenCalledWith({
|
|
blob: expect.anything(),
|
|
name: "filename.png",
|
|
}),
|
|
);
|
|
expect(fetchMock).toHaveFetched("https://example.com/image.png");
|
|
});
|
|
|
|
it("should handle download errors", async () => {
|
|
const modalSpy = jest.spyOn(Modal, "createDialog");
|
|
fetchMock.get("https://example.com/image.png", { status: 500 });
|
|
const { getByRole } = render(
|
|
<ImageView src="https://example.com/image.png" name="filename.png" onFinished={jest.fn()} />,
|
|
);
|
|
fireEvent.click(getByRole("button", { name: "Download" }));
|
|
await waitFor(() =>
|
|
expect(modalSpy).toHaveBeenCalledWith(
|
|
ErrorDialog,
|
|
expect.objectContaining({
|
|
title: "Download failed",
|
|
}),
|
|
),
|
|
);
|
|
});
|
|
});
|