88 lines
2.4 KiB
TypeScript
88 lines
2.4 KiB
TypeScript
 
|
import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
|
||
|
|
import { act, renderHook } from "@testing-library/react";
|
||
|
|
import { useDocker } from "./useDocker";
|
||
|
|
|
||
|
|
const checkDocker = vi.fn();
|
||
|
|
const checkImageExists = vi.fn();
|
||
|
|
|
||
|
|
vi.mock("../lib/tauri-commands", () => ({
|
||
|
|
checkDocker: () => checkDocker(),
|
||
|
|
checkImageExists: () => checkImageExists(),
|
||
|
|
buildImage: vi.fn(),
|
||
|
|
pullImage: vi.fn(),
|
||
|
|
}));
|
||
|
|
|
||
|
|
vi.mock("@tauri-apps/api/event", () => ({ listen: vi.fn(async () => vi.fn()) }));
|
||
|
|
|
||
|
|
const setDockerAvailable = vi.fn();
|
||
|
|
const setImageExists = vi.fn();
|
||
|
|
|
||
|
|
vi.mock("../store/appState", () => ({
|
||
|
|
useAppState: (selector: (s: unknown) => unknown) =>
|
||
|
|
selector({
|
||
|
|
dockerAvailable: false,
|
||
|
|
setDockerAvailable,
|
||
|
|
imageExists: false,
|
||
|
|
setImageExists,
|
||
|
|
}),
|
||
|
|
}));
|
||
|
|
|
||
|
|
/** Let the interval fire and its awaited body settle. */
|
||
|
|
const tick = async () => {
|
||
|
|
await act(async () => {
|
||
|
|
vi.advanceTimersByTime(5000);
|
||
|
|
await Promise.resolve();
|
||
|
|
await Promise.resolve();
|
||
|
|
await Promise.resolve();
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
describe("useDocker.startDockerPolling", () => {
|
||
|
|
beforeEach(() => {
|
||
|
|
vi.clearAllMocks();
|
||
|
|
vi.useFakeTimers();
|
||
|
|
checkImageExists.mockResolvedValue(true);
|
||
|
|
});
|
||
|
|
|
||
|
|
afterEach(() => {
|
||
|
|
vi.useRealTimers();
|
||
|
|
});
|
||
|
|
|
||
|
|
it("runs onAvailable once, after Docker is marked available and the image re-checked", async () => {
|
||
|
|
checkDocker.mockResolvedValueOnce(false).mockResolvedValue(true);
|
||
|
|
const onAvailable = vi.fn();
|
||
|
|
|
||
|
|
const { result } = renderHook(() => useDocker());
|
||
|
|
act(() => {
|
||
|
|
result.current.startDockerPolling(onAvailable);
|
||
|
|
});
|
||
|
|
|
||
|
|
await tick();
|
||
|
|
expect(onAvailable).not.toHaveBeenCalled();
|
||
|
|
|
||
|
|
await tick();
|
||
|
|
expect(setDockerAvailable).toHaveBeenCalledWith(true);
|
||
|
|
expect(setImageExists).toHaveBeenCalledWith(true);
|
||
|
|
expect(onAvailable).toHaveBeenCalledTimes(1);
|
||
|
|
|
||
|
|
// Polling stopped, so no second invocation.
|
||
|
|
await tick();
|
||
|
|
expect(onAvailable).toHaveBeenCalledTimes(1);
|
||
|
|
});
|
||
|
|
|
||
|
|
it("still works without a callback and can be cancelled by its cleanup", async () => {
|
||
|
|
checkDocker.mockResolvedValue(true);
|
||
|
|
|
||
|
|
const { result } = renderHook(() => useDocker());
|
||
|
|
let stop: () => void = () => {};
|
||
|
|
act(() => {
|
||
|
|
stop = result.current.startDockerPolling();
|
||
|
|
});
|
||
|
|
act(() => stop());
|
||
|
|
|
||
|
|
await tick();
|
||
|
|
expect(checkDocker).not.toHaveBeenCalled();
|
||
|
|
expect(setDockerAvailable).not.toHaveBeenCalled();
|
||
|
|
});
|
||
|
|
});
|