homeio

1/* @vitest-environment jsdom */
2
3import { act, renderHook } from "@testing-library/react";
4import { beforeEach, describe, expect, it, vi } from "vitest";
5import { queryKeys } from "@/lib/shared/query-keys";
6import { useNetworkEventsSse } from "@/modules/system/hooks/useNetworkEventsSse";
7import { createTestQueryClient, createWrapper } from "@/test/query-client-wrapper";
8
9class MockEventSource {
10  static instances: MockEventSource[] = [];
11
12  onopen: ((event: Event) => void) | null = null;
13  private listeners = new Map<string, Set<(event: MessageEvent) => void>>();
14  close = vi.fn();
15
16  constructor(public readonly url: string) {
17    MockEventSource.instances.push(this);
18  }
19
20  addEventListener(type: string, listener: (event: MessageEvent) => void) {
21    const set = this.listeners.get(type) ?? new Set();
22    set.add(listener);
23    this.listeners.set(type, set);
24  }
25
26  removeEventListener(type: string, listener: (event: MessageEvent) => void) {
27    const set = this.listeners.get(type);
28    if (!set) return;
29    set.delete(listener);
30  }
31
32  emit(type: string, payload: unknown) {
33    const event = {
34      data: typeof payload === "string" ? payload : JSON.stringify(payload),
35    } as MessageEvent;
36
37    const listeners = this.listeners.get(type);
38    listeners?.forEach((listener) => listener(event));
39  }
40}
41
42describe("useNetworkEventsSse", () => {
43  beforeEach(() => {
44    MockEventSource.instances = [];
45    localStorage.clear();
46  });
47
48  it("connects to network event stream and invalidates caches on events", () => {
49    vi.useFakeTimers();
50    vi.stubGlobal("EventSource", MockEventSource as unknown as typeof EventSource);
51
52    const client = createTestQueryClient();
53    const invalidateSpy = vi.spyOn(client, "invalidateQueries");
54    const { result, unmount } = renderHook(() => useNetworkEventsSse(true), {
55      wrapper: createWrapper(client),
56    });
57
58    const source = MockEventSource.instances[0];
59    expect(source.url).toBe("/api/v1/network/events/stream");
60
61    act(() => {
62      source.onopen?.(new Event("open"));
63    });
64    expect(result.current.status).toBe("connected");
65
66    act(() => {
67      source.emit("network.connection.changed", {
68        type: "network.connection.changed",
69        connected: true,
70        iface: "wlan0",
71        ssid: "HomeNet",
72      });
73      source.emit("network.device.state.changed", {
74        type: "network.device.state.changed",
75        connected: true,
76        iface: "wlan0",
77        ssid: "HomeNet",
78      });
79      // Duplicate payloads should be deduplicated.
80      source.emit("network.connection.changed", {
81        type: "network.connection.changed",
82        connected: true,
83        iface: "wlan0",
84        ssid: "HomeNet",
85      });
86      source.emit("network.device.state.changed", {
87        type: "network.device.state.changed",
88        connected: true,
89        iface: "wlan0",
90        ssid: "HomeNet",
91      });
92      vi.runAllTimers();
93    });
94
95    expect(invalidateSpy).toHaveBeenCalledWith({
96      queryKey: queryKeys.networkStatus,
97    });
98    expect(invalidateSpy).toHaveBeenCalledWith({
99      queryKey: queryKeys.networkNetworks,
100    });
101    const statusInvalidations = invalidateSpy.mock.calls.filter(
102      (call) => call[0]?.queryKey === queryKeys.networkStatus,
103    );
104    const networksInvalidations = invalidateSpy.mock.calls.filter(
105      (call) => call[0]?.queryKey === queryKeys.networkNetworks,
106    );
107    expect(statusInvalidations).toHaveLength(1);
108    expect(networksInvalidations).toHaveLength(1);
109
110    unmount();
111    expect(source.close).toHaveBeenCalledTimes(1);
112    vi.useRealTimers();
113  });
114
115  it("does not connect while a system action is active", () => {
116    vi.stubGlobal("EventSource", MockEventSource as unknown as typeof EventSource);
117    localStorage.setItem(
118      "system.power.action.v1",
119      JSON.stringify({
120        action: "update",
121        startedAt: new Date().toISOString(),
122      }),
123    );
124
125    const client = createTestQueryClient();
126
127    const { result } = renderHook(() => useNetworkEventsSse(true), {
128      wrapper: createWrapper(client),
129    });
130
131    expect(result.current.status).toBe("disconnected");
132    expect(MockEventSource.instances).toHaveLength(0);
133  });
134});
135