openidea/src/features/profiler/profiler.test.ts

import { describe, expect, it } from "vitest";
import type { Profile, RunConfig } from "@/platform/ipc";
import { colorSlot, flameLayout, formatWeight, frameLocation, matchingFrames, profilerSupports, share } from "./model";

// all(10) → main(10, self 2) → [work(6), io(2)]
const profile: Profile = {
  profiler: "t",
  unit: "samples",
  total: 10,
  frames: [
    { name: "all", file: null, line: null },
    { name: "main", file: "/p/main.py", line: 4 },
    { name: "work", file: null, line: null },
    { name: "io", file: null, line: null },
  ],
  nodes: [
    { frame: 0, total: 10, selfWeight: 0, children: [1] },
    { frame: 1, total: 10, selfWeight: 2, children: [2, 3] },
    { frame: 2, total: 6, selfWeight: 6, children: [] },
    { frame: 3, total: 2, selfWeight: 2, children: [] },
  ],
  methods: [],
};

function cfg(command: string, args: string[], kind = "shell"): RunConfig {
  return {
    id: "c",
    name: "c",
    kind,
    command,
    args,
    cwd: null,
    env: {},
    detected: true,
    isTest: false,
    debugAdapter: null,
  };
}

describe("profiler model", () => {
  it("knows which configurations can be profiled", () => {
    expect(profilerSupports(cfg("node", ["a.js"]), false)).toBe(true);
    expect(profilerSupports(cfg("java", ["-cp", "x", "A"], "application"), false)).toBe(true);
    expect(profilerSupports(cfg(".venv/bin/python", ["main.py"]), false)).toBe(true);
    expect(profilerSupports(cfg("cargo", ["run"]), false)).toBe(false);
    expect(profilerSupports(cfg("cargo", ["run"]), true)).toBe(true);
    expect(profilerSupports(cfg("./app", [], "native"), true)).toBe(true);
    expect(profilerSupports(cfg("make", []), true)).toBe(false);
  });

  it("lays out the flame graph and zooms into a node", () => {
    const { rects, depth } = flameLayout(profile);
    expect(depth).toBe(3);
    const at = (node: number) => rects.find((r) => r.node === node)!;
    expect(at(0)).toEqual({ node: 0, depth: 0, x: 0, width: 1 });
    expect(at(2)).toMatchObject({ depth: 2, x: 0, width: 0.6 });
    expect(at(3)).toMatchObject({ depth: 2, x: 0.6, width: 0.2 });

    const zoomed = flameLayout(profile, 2);
    expect(zoomed.rects.map((r) => [r.node, r.depth, r.width])).toEqual([
      [0, 0, 1],
      [1, 1, 1],
      [2, 2, 1],
    ]);
    // Too narrow frames are dropped.
    expect(flameLayout(profile, 0, 0.3).rects.some((r) => r.node === 3)).toBe(false);
  });

  it("searches, formats and locates frames", () => {
    expect(matchingFrames(profile, "  ")).toBeNull();
    expect([...matchingFrames(profile, "WO")!]).toEqual([2]);
    expect(formatWeight(1, "samples")).toBe("1 sample");
    expect(formatWeight(12.34, "ms")).toBe("12.3 ms");
    expect(formatWeight(12_340, "ms")).toBe("12.3 s");
    expect(share(6, 10)).toBe("60%");
    expect(share(0.5, 10)).toBe("5.0%");
    expect(frameLocation(profile, 1)).toBe("main.py:5");
    expect(frameLocation(profile, 2)).toBeNull();
    expect(colorSlot("main")).toBe(colorSlot("main"));
    expect(colorSlot("main")).toBeGreaterThanOrEqual(1);
  });
});