Source code
Revision control
Copy as Markdown
Other Tools
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
import { act, renderHook, waitFor } from "@testing-library/react";
import { useWidgetLabels } from "content-src/components/Nova/CustomizeMenu/WidgetsManagementPanel/useWidgetLabels.jsx";
const WIDGETS = [
{ id: "clocks", customizeL10nId: "newtab-custom-widget-clock-toggle" },
{ id: "weather", customizeL10nId: "newtab-custom-widget-weather-toggle" },
];
describe("useWidgetLabels", () => {
afterEach(() => {
delete document.l10n;
});
it("builds the id map on the first render without document.l10n", () => {
const { result } = renderHook(() => useWidgetLabels(WIDGETS));
expect([...result.current]).toEqual([
["clocks", "clocks"],
["weather", "weather"],
]);
});
it("maps every widget to its own id when the lookup rejects", async () => {
document.l10n = {
formatMessages: jest.fn(() => Promise.reject(new Error("no messages"))),
};
const { result } = renderHook(() => useWidgetLabels(WIDGETS));
expect(result.current).toBeNull();
await waitFor(() => expect(result.current).not.toBeNull());
expect([...result.current]).toEqual([
["clocks", "clocks"],
["weather", "weather"],
]);
});
it("ignores a lookup that resolves after the widget list changed", async () => {
const pending = [];
document.l10n = {
formatMessages: jest.fn(
() => new Promise(resolve => pending.push(resolve))
),
};
const { result, rerender } = renderHook(
widgets => useWidgetLabels(widgets),
{ initialProps: WIDGETS }
);
rerender([WIDGETS[0]]);
await act(async () => {
pending[1]([null]);
pending[0]([null, null]);
});
expect([...result.current]).toEqual([["clocks", "clocks"]]);
});
});