|
| 1 | +// Qa Lab tests cover Windows process tree sampling command selection. |
| 2 | +import path from "node:path"; |
| 3 | +import { afterEach, describe, expect, it, vi } from "vitest"; |
| 4 | + |
| 5 | +const spawnSyncMock = vi.hoisted(() => vi.fn()); |
| 6 | + |
| 7 | +vi.mock("node:child_process", async () => { |
| 8 | +const actual = await vi.importActual<typeof import("node:child_process")>("node:child_process"); |
| 9 | +return { |
| 10 | + ...actual, |
| 11 | +spawnSync: spawnSyncMock, |
| 12 | +}; |
| 13 | +}); |
| 14 | + |
| 15 | +import { readProcessTreeCpuMs } from "./process-tree-cpu.js"; |
| 16 | + |
| 17 | +afterEach(() => { |
| 18 | +vi.restoreAllMocks(); |
| 19 | +vi.unstubAllEnvs(); |
| 20 | +spawnSyncMock.mockReset(); |
| 21 | +}); |
| 22 | + |
| 23 | +describe("readProcessTreeCpuMs on Windows", () => { |
| 24 | +it("uses the trusted Windows PowerShell path", () => { |
| 25 | +vi.spyOn(process, "platform", "get").mockReturnValue("win32"); |
| 26 | +vi.stubEnv("SystemRoot", "D:\\Windows"); |
| 27 | +spawnSyncMock.mockReturnValue({ |
| 28 | +status: 0, |
| 29 | +stdout: JSON.stringify([ |
| 30 | +{ |
| 31 | +ProcessId: 100, |
| 32 | +ParentProcessId: 50, |
| 33 | +KernelModeTime: "10000", |
| 34 | +UserModeTime: "20000", |
| 35 | +WorkingSetSize: "1000", |
| 36 | +}, |
| 37 | +{ |
| 38 | +ProcessId: 101, |
| 39 | +ParentProcessId: 100, |
| 40 | +KernelModeTime: "30000", |
| 41 | +UserModeTime: "40000", |
| 42 | +WorkingSetSize: "2000", |
| 43 | +}, |
| 44 | +]), |
| 45 | +}); |
| 46 | + |
| 47 | +expect(readProcessTreeCpuMs(100)).toBe(10); |
| 48 | +expect(spawnSyncMock.mock.calls[0]?.[0]).toBe( |
| 49 | +path.win32.join("D:\\Windows", "System32", "WindowsPowerShell", "v1.0", "powershell.exe"), |
| 50 | +); |
| 51 | +}); |
| 52 | +}); |