homeio

1import { describe, expect, it, vi } from "vitest";
2
3const { inventoryMock, countersMock, snapshotMock } = vi.hoisted(() => ({
4  inventoryMock: vi.fn(),
5  countersMock: vi.fn(),
6  snapshotMock: vi.fn(),
7}));
8
9vi.mock("@/lib/server/modules/system/disk-service", () => ({ getDiskInventory: inventoryMock }));
10vi.mock("@/lib/server/modules/system/service", () => ({ getSystemMetricsSnapshot: snapshotMock }));
11vi.mock("@/lib/server/modules/system/disk-activity", async (importOriginal) => ({
12  ...(await importOriginal<typeof import("@/lib/server/modules/system/disk-activity")>()),
13  readDiskCounters: countersMock,
14}));
15
16import { getDiskMonitor } from "@/lib/server/modules/system/disk-monitor";
17
18const sda = {
19  name: "sda",
20  device: "/dev/sda",
21  model: "Samsung SSD 870",
22  vendor: "ATA",
23  serial: "S1",
24  sizeBytes: 500_000_000_000,
25  mediaType: "ssd",
26  transport: "sata",
27  isRemovable: false,
28  partitions: [
29    { name: "sda1", device: "/dev/sda1", number: 1, fstype: "ext4", label: null, uuid: "u", sizeBytes: 500_000_000_000, mountpoint: "/", type: "part", ro: false },
30  ],
31};
32
33describe("getDiskMonitor", () => {
34  it("adds live throughput and SMART data to each disk", async () => {
35    vi.useFakeTimers();
36    inventoryMock.mockResolvedValue({ disks: [sda], unavailableReason: null, readOnly: false });
37    snapshotMock.mockResolvedValue({
38      storage: { smart: { disks: [{ device: "/dev/sda", status: "healthy", temperatureCelsius: 34, powerOnHours: 1200 }] } },
39    });
40
41    vi.setSystemTime(10_000);
42    countersMock.mockResolvedValueOnce(new Map([["sda", { readBytes: 0, writeBytes: 0 }]]));
43    const first = await getDiskMonitor();
44    expect(first.disks[0].readBytesPerSec).toBeNull();
45
46    vi.setSystemTime(12_000);
47    countersMock.mockResolvedValueOnce(new Map([["sda", { readBytes: 4_000_000, writeBytes: 1_000_000 }]]));
48    const second = await getDiskMonitor();
49
50    expect(second.disks[0]).toMatchObject({
51      device: "/dev/sda",
52      readBytesPerSec: 2_000_000,
53      writeBytesPerSec: 500_000,
54      health: "healthy",
55      temperatureCelsius: 34,
56      powerOnHours: 1200,
57      partitions: [{ name: "sda1", mountpoint: "/", fstype: "ext4", sizeBytes: 500_000_000_000 }],
58    });
59    vi.useRealTimers();
60  });
61
62  it("passes on why disks cannot be listed", async () => {
63    inventoryMock.mockResolvedValue({ disks: [], unavailableReason: "Disk management needs Linux; this machine runs darwin.", readOnly: false });
64    countersMock.mockResolvedValue(null);
65    snapshotMock.mockRejectedValue(new Error("no metrics"));
66
67    const result = await getDiskMonitor();
68
69    expect(result.disks).toEqual([]);
70    expect(result.unavailableReason).toContain("needs Linux");
71  });
72
73  it("leaves out the container's own mount points in Docker", async () => {
74    inventoryMock.mockResolvedValue({ disks: [sda], unavailableReason: null, readOnly: true });
75    countersMock.mockResolvedValue(null);
76    snapshotMock.mockResolvedValue(null);
77
78    const result = await getDiskMonitor();
79
80    expect(result.inContainer).toBe(true);
81    expect(result.disks[0].partitions[0].mountpoint).toBeNull();
82  });
83});
84