homeio

1import { describe, expect, it } from "vitest";
2import {
3  createDiskRateTracker,
4  parseDiskstats,
5  totalMegabytesPerSecond,
6} from "@/lib/server/modules/system/disk-activity";
7
8// Real layout: major minor name reads merged sectors_read ms writes merged sectors_written ...
9const DISKSTATS = `
10   8       0 sda 1000 0 2000 0 500 0 4000 0 0 0 0 0 0 0 0 0 0
11   8       1 sda1 900 0 1800 0 400 0 3800 0 0 0 0 0 0 0 0 0 0
12 259       0 nvme0n1 10 0 100 0 20 0 200 0 0 0 0 0 0 0 0 0 0
13 259       1 nvme0n1p1 10 0 100 0 20 0 200 0 0 0 0 0 0 0 0 0 0
14 179       0 mmcblk0 1 0 8 0 1 0 8 0 0 0 0 0 0 0 0 0 0
15   7       0 loop0 5 0 50 0 0 0 0 0 0 0 0 0 0 0 0 0 0
16 252       0 zram0 5 0 50 0 5 0 50 0 0 0 0 0 0 0 0 0 0
17 253       0 dm-0 5 0 50 0 5 0 50 0 0 0 0 0 0 0 0 0 0
18`;
19
20describe("disk activity", () => {
21  it("reads whole disks only, in bytes", () => {
22    const counters = parseDiskstats(DISKSTATS);
23
24    expect([...counters.keys()]).toEqual(["sda", "nvme0n1", "mmcblk0"]);
25    expect(counters.get("sda")).toEqual({ readBytes: 2000 * 512, writeBytes: 4000 * 512 });
26  });
27
28  it("turns two readings into per-disk rates", () => {
29    const track = createDiskRateTracker();
30    const first = new Map([["sda", { readBytes: 0, writeBytes: 0 }]]);
31    const second = new Map([["sda", { readBytes: 10_000_000, writeBytes: 2_000_000 }]]);
32
33    expect(track(first, 1_000).size).toBe(0);
34    const rates = track(second, 3_000);
35
36    expect(rates.get("sda")).toEqual({ readBytesPerSec: 5_000_000, writeBytesPerSec: 1_000_000 });
37    expect(totalMegabytesPerSecond(rates)).toEqual({ readMBps: 5, writeMBps: 1 });
38  });
39
40  it("gives no rate after a long gap or when a counter goes backwards", () => {
41    const track = createDiskRateTracker(60_000);
42    track(new Map([["sda", { readBytes: 100, writeBytes: 100 }]]), 0);
43
44    expect(track(new Map([["sda", { readBytes: 200, writeBytes: 200 }]]), 120_000).size).toBe(0);
45    expect(track(new Map([["sda", { readBytes: 50, writeBytes: 300 }]]), 121_000).size).toBe(0);
46    expect(totalMegabytesPerSecond(new Map())).toEqual({ readMBps: null, writeMBps: null });
47  });
48});
49