mirror of
https://github.com/andrewyng/openworker.git
synced 2026-09-13 15:50:02 +00:00
Rohit C PrasadandDevika
2b45018ffa
Imported from andrewyng/aisuite@1b4bbf303e (contents of its platform/ directory, hoisted to the repo root). Development history prior to this commit lives in that repository. Co-authored-by: Devika <devikaverma11@gmail.com>
98 lines
3.5 KiB
TypeScript
98 lines
3.5 KiB
TypeScript
// Auto-update banner: periodic check + per-version "Later" + background pre-download,
|
|
// driven through a mocked __TAURI__ global (the browser build renders nothing, so the
|
|
// e2e harness never sees this — these unit tests are the coverage).
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
import { act, cleanup, fireEvent, render, screen } from "@testing-library/react";
|
|
import { UpdateBanner } from "./UpdateBanner";
|
|
|
|
const FIRST_CHECK_MS = 15_000;
|
|
const RECHECK_MS = 30 * 60_000;
|
|
|
|
let invoke: ReturnType<typeof vi.fn>;
|
|
let available: { version: string; notes: string } | null;
|
|
let download: () => Promise<void>;
|
|
|
|
beforeEach(() => {
|
|
vi.useFakeTimers();
|
|
available = { version: "1.2.0", notes: "" };
|
|
download = async () => {};
|
|
invoke = vi.fn(async (cmd: string) => {
|
|
if (cmd === "check_for_update") return available;
|
|
if (cmd === "download_update") return download();
|
|
if (cmd === "install_update") return null;
|
|
return null;
|
|
});
|
|
(globalThis as any).__TAURI__ = { core: { invoke } };
|
|
});
|
|
|
|
afterEach(() => {
|
|
cleanup();
|
|
vi.useRealTimers();
|
|
delete (globalThis as any).__TAURI__;
|
|
});
|
|
|
|
const advance = (ms: number) => act(() => vi.advanceTimersByTimeAsync(ms));
|
|
|
|
describe("UpdateBanner", () => {
|
|
it("shows after the boot-settle check finds an update", async () => {
|
|
render(<UpdateBanner />);
|
|
expect(screen.queryByTestId("update-banner")).toBeNull();
|
|
|
|
await advance(FIRST_CHECK_MS);
|
|
expect(screen.getByTestId("update-banner").textContent).toContain("v1.2.0");
|
|
// Pre-download resolved immediately → the button is ready and enabled.
|
|
const btn = screen.getByTestId("update-install") as HTMLButtonElement;
|
|
expect(btn.textContent).toBe("Restart to update");
|
|
expect(btn.disabled).toBe(false);
|
|
});
|
|
|
|
it("Later hides the banner and a same-version re-check keeps it hidden", async () => {
|
|
render(<UpdateBanner />);
|
|
await advance(FIRST_CHECK_MS);
|
|
|
|
fireEvent.click(screen.getByTestId("update-later"));
|
|
expect(screen.queryByTestId("update-banner")).toBeNull();
|
|
|
|
await advance(RECHECK_MS);
|
|
expect(screen.queryByTestId("update-banner")).toBeNull();
|
|
});
|
|
|
|
it("a NEWER version found by a later check overrides the dismissal", async () => {
|
|
render(<UpdateBanner />);
|
|
await advance(FIRST_CHECK_MS);
|
|
fireEvent.click(screen.getByTestId("update-later"));
|
|
|
|
available = { version: "1.3.0", notes: "" };
|
|
await advance(RECHECK_MS);
|
|
expect(screen.getByTestId("update-banner").textContent).toContain("v1.3.0");
|
|
});
|
|
|
|
it("button reads Downloading… (disabled) until the pre-download resolves", async () => {
|
|
let finish!: () => void;
|
|
download = () => new Promise((resolve) => (finish = resolve));
|
|
render(<UpdateBanner />);
|
|
await advance(FIRST_CHECK_MS);
|
|
|
|
const btn = screen.getByTestId("update-install") as HTMLButtonElement;
|
|
expect(btn.textContent).toBe("Downloading…");
|
|
expect(btn.disabled).toBe(true);
|
|
|
|
await act(async () => finish());
|
|
expect(btn.textContent).toBe("Restart to update");
|
|
expect(btn.disabled).toBe(false);
|
|
});
|
|
|
|
it("a failed pre-download falls back to the enabled download-on-click path", async () => {
|
|
download = () => Promise.reject(new Error("offline"));
|
|
render(<UpdateBanner />);
|
|
await advance(FIRST_CHECK_MS);
|
|
|
|
const btn = screen.getByTestId("update-install") as HTMLButtonElement;
|
|
expect(btn.textContent).toBe("Restart to update");
|
|
expect(btn.disabled).toBe(false);
|
|
|
|
fireEvent.click(btn);
|
|
expect(invoke).toHaveBeenCalledWith("install_update", undefined);
|
|
});
|
|
});
|