homeio

1/* @vitest-environment jsdom */
2
3import { act, fireEvent, render, screen, waitFor } from "@testing-library/react";
4import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
5import type { SystemUpdateStatus } from "@/lib/shared/contracts/system";
6import { UPDATE_REMINDER_SNOOZE_MS } from "@/lib/desktop/update-reminder";
7import {
8  DEFAULT_DESKTOP_PREFERENCES,
9  DESKTOP_PREFERENCES_STORAGE_KEY,
10} from "@/lib/desktop/preferences";
11
12const { statusRef, applyMock, persistMock } = vi.hoisted(() => ({
13  statusRef: { current: undefined as SystemUpdateStatus | undefined },
14  applyMock: vi.fn(),
15  persistMock: vi.fn(),
16}));
17
18vi.mock("@/modules/system/hooks/useSystemUpdateStatus", () => ({
19  useSystemUpdateStatus: () => ({ data: statusRef.current }),
20}));
21
22vi.mock("@/modules/settings/hooks/backend/api", () => ({
23  applySystemUpdateRequest: applyMock,
24}));
25
26vi.mock("@/modules/settings/hooks/useUpdateRecoveryState", () => ({
27  persistPowerActionState: persistMock,
28}));
29
30import { UpdateReminder } from "@/modules/system/components/update-reminder";
31
32function updateStatus(overrides: Partial<SystemUpdateStatus> = {}): SystemUpdateStatus {
33  return {
34    currentVersion: "1.10.0",
35    latestVersion: "1.10.1",
36    updateAvailable: true,
37    checkedAt: "2026-09-27T10:00:00.000Z",
38    canSelfUpdate: true,
39    ...overrides,
40  };
41}
42
43describe("UpdateReminder", () => {
44  beforeEach(() => {
45    window.localStorage.clear();
46    statusRef.current = updateStatus();
47    applyMock.mockReset();
48    persistMock.mockReset();
49  });
50
51  afterEach(() => {
52    vi.useRealTimers();
53  });
54
55  it("shows the new and current version while an update is available", async () => {
56    render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
57
58    const bar = await screen.findByRole("status", { name: "Homeio update available" });
59    expect(bar.textContent).toContain("Homeio 1.10.1 is available");
60    expect(bar.textContent).toContain("You have 1.10.0");
61    expect(screen.getByRole("button", { name: "Update now" })).toBeTruthy();
62  });
63
64  it("stays hidden when there is no update, when disabled, or when update notifications are off", async () => {
65    statusRef.current = updateStatus({ updateAvailable: false });
66    const { rerender } = render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
67    await waitFor(() => expect(screen.queryByRole("status")).toBeNull());
68
69    statusRef.current = updateStatus();
70    rerender(<UpdateReminder enabled={false} onOpenDetails={vi.fn()} />);
71    await waitFor(() => expect(screen.queryByRole("status")).toBeNull());
72  });
73
74  it("respects the existing update notifications preference", async () => {
75    window.localStorage.setItem(
76      DESKTOP_PREFERENCES_STORAGE_KEY,
77      JSON.stringify({
78        ...DEFAULT_DESKTOP_PREFERENCES,
79        notifications: {
80          ...DEFAULT_DESKTOP_PREFERENCES.notifications,
81          updateNotificationsEnabled: false,
82        },
83      }),
84    );
85    render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
86    await waitFor(() => expect(screen.queryByRole("status")).toBeNull());
87  });
88
89  it("hides on Later and comes back an hour later without a reload", async () => {
90    vi.useFakeTimers({ shouldAdvanceTime: true });
91    render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
92
93    fireEvent.click(await screen.findByRole("button", { name: "Later" }));
94    expect(screen.queryByRole("status")).toBeNull();
95
96    await act(async () => {
97      vi.advanceTimersByTime(UPDATE_REMINDER_SNOOZE_MS - 1_000);
98    });
99    expect(screen.queryByRole("status")).toBeNull();
100
101    await act(async () => {
102      vi.advanceTimersByTime(2_000);
103    });
104    expect(screen.getByRole("status", { name: "Homeio update available" })).toBeTruthy();
105  });
106
107  it("keeps the snooze across a reload but shows a newer version straight away", async () => {
108    const first = render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
109    fireEvent.click(await screen.findByRole("button", { name: "Later" }));
110    first.unmount();
111
112    const second = render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
113    await waitFor(() => expect(screen.queryByRole("status")).toBeNull());
114    second.unmount();
115
116    statusRef.current = updateStatus({ latestVersion: "1.10.2" });
117    render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
118    expect((await screen.findByRole("status")).textContent).toContain("1.10.2");
119  });
120
121  it("opens the details and shows the Docker command instead of Update now in the container", async () => {
122    statusRef.current = updateStatus({ canSelfUpdate: false });
123    const onOpenDetails = vi.fn();
124    render(<UpdateReminder enabled onOpenDetails={onOpenDetails} />);
125
126    const bar = await screen.findByRole("status");
127    expect(screen.queryByRole("button", { name: "Update now" })).toBeNull();
128    expect(bar.textContent).toContain("docker compose pull && docker compose up -d");
129
130    fireEvent.click(screen.getByRole("button", { name: "Details" }));
131    expect(onOpenDetails).toHaveBeenCalledOnce();
132    expect(screen.queryByRole("status")).toBeNull();
133  });
134
135  it("shows the error and keeps the bar when the update cannot start", async () => {
136    applyMock.mockRejectedValueOnce(new Error("Failed to schedule Homeio update"));
137    render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
138
139    fireEvent.click(await screen.findByRole("button", { name: "Update now" }));
140
141    expect(await screen.findByText("Failed to schedule Homeio update")).toBeTruthy();
142    expect(persistMock).not.toHaveBeenCalled();
143    expect(screen.getByRole("button", { name: "Update now" })).toBeTruthy();
144  });
145
146  it("hands off to the updating screen like Settings does", async () => {
147    applyMock.mockResolvedValueOnce({ action: "update", accepted: true });
148    const replace = vi.fn();
149    const originalLocation = window.location;
150    Object.defineProperty(window, "location", {
151      configurable: true,
152      value: { ...originalLocation, replace },
153    });
154
155    try {
156      render(<UpdateReminder enabled onOpenDetails={vi.fn()} />);
157      fireEvent.click(await screen.findByRole("button", { name: "Update now" }));
158
159      await waitFor(() => expect(replace).toHaveBeenCalledWith("/updating"));
160      expect(persistMock).toHaveBeenCalledWith(
161        "update",
162        expect.objectContaining({ requestDispatchedAt: expect.any(String) }),
163      );
164    } finally {
165      Object.defineProperty(window, "location", { configurable: true, value: originalLocation });
166    }
167  });
168});
169