import type { DesktopUpdateState } from "@t3tools/contracts";import * as AsyncResult from "effect/unstable/reactivity/AsyncResult";import { AtomRegistry } from "effect/unstable/reactivity";import { afterEach, describe, expect, it, vi } from "vite-plus/test"; import { createDesktopUpdateStateAtom, DesktopUpdateStateReadError } from "./desktopUpdate"; const baseState: DesktopUpdateState = { enabled: true, status: "idle", channel: "latest", currentVersion: "1.0.0", hostArch: "x64", appArch: "x64", runningUnderArm64Translation: false, availableVersion: null, downloadedVersion: null, releaseNotes: [], omittedReleaseCount: 0, downloadPercent: null, checkedAt: null, message: null, errorContext: null, canRetry: false,}; afterEach(() => { vi.restoreAllMocks();}); describe("desktopUpdateStateAtom", () => { it("loads once, retains state, and follows desktop update events", async () => { let listener: ((state: DesktopUpdateState) => void) | undefined; const unsubscribe = vi.fn(); const getUpdateState = vi.fn(async () => baseState); const onUpdateState = vi.fn((nextListener: (state: DesktopUpdateState) => void) => { listener = nextListener; return unsubscribe; }); const atom = createDesktopUpdateStateAtom(() => ({ getUpdateState, onUpdateState })); const registry = AtomRegistry.make(); const unmount = registry.mount(atom); await vi.waitFor(() => { expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(baseState); }); const downloadedState: DesktopUpdateState = { ...baseState, status: "downloaded", availableVersion: "1.1.0", downloadedVersion: "1.1.0", }; listener?.(downloadedState); await vi.waitFor(() => { expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(downloadedState); }); unmount(); const remount = registry.mount(atom); expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(downloadedState); expect(getUpdateState).toHaveBeenCalledTimes(1); expect(onUpdateState).toHaveBeenCalledTimes(1); remount(); registry.dispose(); expect(unsubscribe).toHaveBeenCalledTimes(1); }); it("does not let a slower initial read overwrite a newer update event", async () => { let resolveInitial: ((state: DesktopUpdateState) => void) | undefined; let listener: ((state: DesktopUpdateState) => void) | undefined; const atom = createDesktopUpdateStateAtom(() => ({ getUpdateState: () => new Promise<DesktopUpdateState>((resolve) => { resolveInitial = resolve; }), onUpdateState: (nextListener) => { listener = nextListener; return () => undefined; }, })); const registry = AtomRegistry.make(); registry.mount(atom); await vi.waitFor(() => expect(listener).toBeDefined()); const newerState: DesktopUpdateState = { ...baseState, status: "checking" }; listener?.(newerState); resolveInitial?.(baseState); await vi.waitFor(() => { expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(newerState); }); registry.dispose(); }); it("keeps listening when the initial desktop state read fails", async () => { let listener: ((state: DesktopUpdateState) => void) | undefined; const cause = new Error("IPC unavailable"); const reportError = vi.spyOn(console, "log").mockImplementation(() => undefined); const getUpdateState = vi.fn(async () => Promise.reject(cause)); const atom = createDesktopUpdateStateAtom(() => ({ getUpdateState, onUpdateState: (nextListener) => { listener = nextListener; return () => undefined; }, })); const registry = AtomRegistry.make(); registry.mount(atom); await vi.waitFor(() => expect(listener).toBeDefined()); await vi.waitFor(() => expect(reportError).toHaveBeenCalledOnce()); expect(getUpdateState).toHaveBeenCalledTimes(3); const [, errorMessage, errorContext] = reportError.mock.calls[0] ?? []; expect(errorMessage).toBe("Failed to read the initial desktop update state after 3 attempts."); expect(errorContext).toMatchObject({ errorTag: "DesktopUpdateStateReadError", attemptCount: 3, }); const loggedError = (errorContext as { readonly error: unknown }).error; expect(loggedError).toBeInstanceOf(DesktopUpdateStateReadError); expect(loggedError).toMatchObject({ _tag: "DesktopUpdateStateReadError", attemptCount: 3, }); expect((loggedError as DesktopUpdateStateReadError).cause).toBe(cause); listener?.(baseState); await vi.waitFor(() => { expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(baseState); }); registry.dispose(); });});
import type { DesktopUpdateState } from "@t3tools/contracts";
import * as AsyncResult from "effect/unstable/reactivity/AsyncResult";
import { AtomRegistry } from "effect/unstable/reactivity";
import { afterEach, describe, expect, it, vi } from "vite-plus/test";
import { createDesktopUpdateStateAtom, DesktopUpdateStateReadError } from "./desktopUpdate";
const baseState: DesktopUpdateState = {
enabled: true,
status: "idle",
channel: "latest",
currentVersion: "1.0.0",
hostArch: "x64",
appArch: "x64",
runningUnderArm64Translation: false,
availableVersion: null,
downloadedVersion: null,
releaseNotes: [],
omittedReleaseCount: 0,
downloadPercent: null,
checkedAt: null,
message: null,
errorContext: null,
canRetry: false,
};
afterEach(() => {
vi.restoreAllMocks();
});
describe("desktopUpdateStateAtom", () => {
it("loads once, retains state, and follows desktop update events", async () => {
let listener: ((state: DesktopUpdateState) => void) | undefined;
const unsubscribe = vi.fn();
const getUpdateState = vi.fn(async () => baseState);
const onUpdateState = vi.fn((nextListener: (state: DesktopUpdateState) => void) => {
listener = nextListener;
return unsubscribe;
});
const atom = createDesktopUpdateStateAtom(() => ({ getUpdateState, onUpdateState }));
const registry = AtomRegistry.make();
const unmount = registry.mount(atom);
await vi.waitFor(() => {
expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(baseState);
});
const downloadedState: DesktopUpdateState = {
...baseState,
status: "downloaded",
availableVersion: "1.1.0",
downloadedVersion: "1.1.0",
};
listener?.(downloadedState);
await vi.waitFor(() => {
expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(downloadedState);
});
unmount();
const remount = registry.mount(atom);
expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(downloadedState);
expect(getUpdateState).toHaveBeenCalledTimes(1);
expect(onUpdateState).toHaveBeenCalledTimes(1);
remount();
registry.dispose();
expect(unsubscribe).toHaveBeenCalledTimes(1);
});
it("does not let a slower initial read overwrite a newer update event", async () => {
let resolveInitial: ((state: DesktopUpdateState) => void) | undefined;
let listener: ((state: DesktopUpdateState) => void) | undefined;
const atom = createDesktopUpdateStateAtom(() => ({
getUpdateState: () =>
new Promise<DesktopUpdateState>((resolve) => {
resolveInitial = resolve;
}),
onUpdateState: (nextListener) => {
listener = nextListener;
return () => undefined;
},
}));
const registry = AtomRegistry.make();
registry.mount(atom);
await vi.waitFor(() => expect(listener).toBeDefined());
const newerState: DesktopUpdateState = { ...baseState, status: "checking" };
listener?.(newerState);
resolveInitial?.(baseState);
await vi.waitFor(() => {
expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(newerState);
});
registry.dispose();
});
it("keeps listening when the initial desktop state read fails", async () => {
let listener: ((state: DesktopUpdateState) => void) | undefined;
const cause = new Error("IPC unavailable");
const reportError = vi.spyOn(console, "log").mockImplementation(() => undefined);
const getUpdateState = vi.fn(async () => Promise.reject(cause));
const atom = createDesktopUpdateStateAtom(() => ({
getUpdateState,
onUpdateState: (nextListener) => {
listener = nextListener;
return () => undefined;
},
}));
const registry = AtomRegistry.make();
registry.mount(atom);
await vi.waitFor(() => expect(listener).toBeDefined());
await vi.waitFor(() => expect(reportError).toHaveBeenCalledOnce());
expect(getUpdateState).toHaveBeenCalledTimes(3);
const [, errorMessage, errorContext] = reportError.mock.calls[0] ?? [];
expect(errorMessage).toBe("Failed to read the initial desktop update state after 3 attempts.");
expect(errorContext).toMatchObject({
errorTag: "DesktopUpdateStateReadError",
attemptCount: 3,
});
const loggedError = (errorContext as { readonly error: unknown }).error;
expect(loggedError).toBeInstanceOf(DesktopUpdateStateReadError);
expect(loggedError).toMatchObject({
_tag: "DesktopUpdateStateReadError",
attemptCount: 3,
});
expect((loggedError as DesktopUpdateStateReadError).cause).toBe(cause);
listener?.(baseState);
await vi.waitFor(() => {
expect(AsyncResult.getOrElse(registry.get(atom), () => null)).toEqual(baseState);
});
registry.dispose();
});
});