homeio

1import { afterEach, describe, expect, it, vi } from "vitest";
2import type * as NetworkActivity from "@/lib/server/modules/system/network-activity";
3
4const { countersMock } = vi.hoisted(() => ({ countersMock: vi.fn() }));
5
6vi.mock("node:os", () => ({
7  default: {
8    loadavg: () => [0.6, 0.5, 0.4],
9    cpus: () => [0, 1, 2].map(() => ({ times: { user: 100, nice: 0, sys: 50, idle: 850, irq: 0 } })),
10    networkInterfaces: () => ({
11      wlan0: [
12        { address: "192.168.1.22", family: "IPv4", internal: false },
13        { address: "fe80::1234", family: "IPv6", internal: false },
14      ],
15    }),
16    totalmem: () => 1024,
17    freemem: () => 256,
18    hostname: () => "pi",
19    platform: () => "linux",
20    release: () => "6.6",
21    uptime: () => 3600,
22    arch: () => "x64",
23  },
24}));
25
26vi.mock("@/lib/server/modules/system/network-activity", async (importOriginal) => ({
27  ...(await importOriginal<typeof NetworkActivity>()),
28  readDefaultRouteInterface: vi.fn(async () => "wlan0"),
29  readInterfaceCounters: countersMock,
30}));
31
32vi.mock("systeminformation", () => ({
33  default: {
34    // Must stay unused by the snapshot: each call spawns cat/grep/nmcli/ip.
35    currentLoad: vi.fn(),
36    networkInterfaces: vi.fn(),
37    networkStats: vi.fn(),
38    cpuTemperature: vi.fn(async () => ({
39      main: 49.2,
40      cores: [48.5, 51.1],
41      max: 51.1,
42    })),
43    battery: vi.fn(async () => ({
44      hasBattery: true,
45      isCharging: false,
46      percent: 78,
47      timeRemaining: 98,
48      acConnected: false,
49      manufacturer: "SONY",
50      cycleCount: 248,
51      designedCapacity: 50,
52      maxCapacity: 45,
53    })),
54    wifiConnections: vi.fn(async () => [
55      {
56        iface: "wlan0",
57        ssid: "HomeNet",
58        bssid: "11:22:33",
59        quality: 66,
60        txRate: 130,
61      },
62    ]),
63    wifiNetworks: vi.fn(async () => [
64      {
65        ssid: "HomeNet",
66        channel: 1,
67        quality: 66,
68        security: ["WPA2"],
69        rsnFlags: [],
70      },
71    ]),
72    fsSize: vi.fn(async () => [
73      {
74        fs: "/dev/sda1",
75        type: "ext4",
76        size: 500 * 1024 * 1024 * 1024,
77        used: 120 * 1024 * 1024 * 1024,
78        use: 24,
79        mount: "/",
80      },
81    ]),
82    diskLayout: vi.fn(async () => [
83      {
84        device: "/dev/sda",
85        type: "SSD",
86        smartStatus: "Ok",
87      },
88    ]),
89  },
90}));
91
92import { getSystemMetricsSnapshot } from "@/lib/server/modules/system/service";
93
94describe("system service", () => {
95  afterEach(() => {
96    vi.restoreAllMocks();
97  });
98
99  it("returns a snapshot with expected shape", async () => {
100    // Network throughput is a rate between two snapshots taken 2s apart.
101    const now = vi.spyOn(Date, "now").mockReturnValue(10_000);
102    countersMock.mockResolvedValueOnce({ rxBytes: 0, txBytes: 0 });
103    await getSystemMetricsSnapshot({ bypassCache: true });
104    now.mockReturnValue(12_000);
105    countersMock.mockResolvedValueOnce({ rxBytes: 6_200_000, txBytes: 2_200_000 });
106    const snapshot = await getSystemMetricsSnapshot({ bypassCache: true });
107
108    expect(snapshot.hostname).toBe("pi");
109    expect(snapshot.timestamp).toBeTypeOf("string");
110    expect(snapshot.uptimeSeconds).toBe(3600);
111    expect(snapshot.cpu.normalizedPercent).toBeGreaterThanOrEqual(0);
112    expect(snapshot.cpu.normalizedPercent).toBeLessThanOrEqual(100);
113    expect(snapshot.memory.totalBytes).toBe(1024);
114    expect(snapshot.temperature.mainCelsius).toBe(49.2);
115    expect(snapshot.battery.percent).toBe(78);
116    expect(snapshot.battery.manufacturer).toBe("SONY");
117    expect(snapshot.battery.cycleCount).toBe(248);
118    expect(snapshot.battery.designToMaxCapacityPercent).toBe(90);
119    expect(snapshot.wifi.connected).toBe(true);
120    expect(snapshot.wifi.ssid).toBe("HomeNet");
121    expect(snapshot.wifi.iface).toBe("wlan0");
122    expect(snapshot.wifi.ipv4).toBe("192.168.1.22");
123    expect(snapshot.wifi.ipv6).toBe("fe80::1234");
124    expect(snapshot.wifi.downloadMbps).toBe(24.8);
125    expect(snapshot.wifi.uploadMbps).toBe(8.8);
126  });
127
128  it("reads network and CPU load from the kernel, without spawning commands", async () => {
129    const si = (await import("systeminformation")).default;
130    countersMock.mockResolvedValue({ rxBytes: 0, txBytes: 0 });
131
132    await getSystemMetricsSnapshot({ bypassCache: true });
133
134    expect(si.currentLoad).not.toHaveBeenCalled();
135    expect(si.networkInterfaces).not.toHaveBeenCalled();
136    expect(si.networkStats).not.toHaveBeenCalled();
137  });
138
139  it("serves cached snapshot unless bypassCache is set", async () => {
140    const cachedOne = await getSystemMetricsSnapshot();
141    const cachedTwo = await getSystemMetricsSnapshot();
142    const fresh = await getSystemMetricsSnapshot({ bypassCache: true });
143
144    expect(cachedTwo).toBe(cachedOne);
145    expect(fresh).not.toBe(cachedOne);
146  });
147});
148