@blackbelt-technology/pi-agent-dashboard 0.2.0

package/packages/extension/src/__tests__/source-detector.test.ts

@@ -0,0 +1,73 @@
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+import { detectSessionSource } from "../source-detector.js";
+import { writeSessionMeta } from "@blackbelt-technology/pi-dashboard-shared/session-meta.js";
+
+describe("detectSessionSource", () => {
+  const originalEnv = { ...process.env };
+  let tmpDir: string;
+
+  beforeEach(() => {
+    delete process.env.ZED_TERM;
+    delete process.env.TMUX;
+    tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "source-detect-test-"));
+  });
+
+  afterEach(() => {
+    process.env = { ...originalEnv };
+    fs.rmSync(tmpDir, { recursive: true, force: true });
+  });
+
+  it("should return 'dashboard' when .meta.json has source dashboard", () => {
+    const sessionFile = path.join(tmpDir, "test.jsonl");
+    writeSessionMeta(sessionFile, { source: "dashboard" });
+    expect(detectSessionSource(undefined, sessionFile)).toBe("dashboard");
+  });
+
+  it("should return 'zed' when ZED_TERM is set and hasUI is false", () => {
+    process.env.ZED_TERM = "1";
+    expect(detectSessionSource(false)).toBe("zed");
+  });
+
+  it("should return 'zed' when ZED_TERM is set and hasUI is not provided", () => {
+    process.env.ZED_TERM = "1";
+    expect(detectSessionSource()).toBe("zed");
+  });
+
+  it("should return 'tui' when ZED_TERM is set but hasUI is true", () => {
+    process.env.ZED_TERM = "1";
+    expect(detectSessionSource(true)).toBe("tui");
+  });
+
+  it("should return 'tmux' when TMUX is set", () => {
+    process.env.TMUX = "/tmp/tmux-1000/default,12345,0";
+    expect(detectSessionSource()).toBe("tmux");
+  });
+
+  it("should return 'tui' when no env vars or meta file", () => {
+    expect(detectSessionSource()).toBe("tui");
+  });
+
+  it("should return 'tui' when no session file provided", () => {
+    expect(detectSessionSource(undefined, undefined)).toBe("tui");
+  });
+
+  it("should prioritize zed over tmux when hasUI is false", () => {
+    process.env.ZED_TERM = "1";
+    process.env.TMUX = "something";
+    expect(detectSessionSource(false)).toBe("zed");
+  });
+
+  it("should return 'tui' over tmux when ZED_TERM is set and hasUI is true", () => {
+    process.env.ZED_TERM = "1";
+    process.env.TMUX = "something";
+    expect(detectSessionSource(true)).toBe("tui");
+  });
+
+  it("should return 'tui' when meta file does not exist", () => {
+    const sessionFile = path.join(tmpDir, "nonexistent.jsonl");
+    expect(detectSessionSource(undefined, sessionFile)).toBe("tui");
+  });
+});

package/packages/extension/src/__tests__/stats-extractor.test.ts

@@ -0,0 +1,92 @@
+import { describe, it, expect } from "vitest";
+import { extractTurnStats } from "@blackbelt-technology/pi-dashboard-shared/stats-extractor.js";
+
+describe("extractTurnStats", () => {
+  it("extracts stats from event.message.usage", () => {
+    const event = {
+      type: "turn_end",
+      turnIndex: 0,
+      message: {
+        usage: {
+          input: 1500,
+          output: 300,
+          cacheRead: 800,
+          cacheWrite: 200,
+          totalTokens: 2800,
+          cost: { input: 0.001, output: 0.002, cacheRead: 0.0005, cacheWrite: 0.0005, total: 0.004 },
+        },
+      },
+      toolResults: [],
+    };
+
+    const result = extractTurnStats(event);
+
+    expect(result).toEqual({
+      tokensIn: 1500,
+      tokensOut: 300,
+      cost: 0.004,
+      turnUsage: {
+        input: 1500,
+        output: 300,
+        cacheRead: 800,
+        cacheWrite: 200,
+      },
+    });
+  });
+
+  it("returns null when message.usage is undefined", () => {
+    const event = {
+      type: "turn_end",
+      turnIndex: 0,
+      message: { role: "assistant", content: [] },
+      toolResults: [],
+    };
+
+    expect(extractTurnStats(event)).toBeNull();
+  });
+
+  it("returns null when message is undefined", () => {
+    const event = { type: "turn_end", turnIndex: 0 };
+    expect(extractTurnStats(event)).toBeNull();
+  });
+
+  it("includes contextUsage when provided", () => {
+    const event = {
+      type: "turn_end",
+      message: {
+        usage: {
+          input: 100,
+          output: 50,
+          cacheRead: 0,
+          cacheWrite: 0,
+          totalTokens: 150,
+          cost: { total: 0.001 },
+        },
+      },
+    };
+
+    const result = extractTurnStats(event, { tokens: 5000, contextWindow: 128000 });
+
+    expect(result?.contextUsage).toEqual({ tokens: 5000, contextWindow: 128000 });
+  });
+
+  it("omits contextUsage when not provided", () => {
+    const event = {
+      type: "turn_end",
+      message: {
+        usage: {
+          input: 100,
+          output: 50,
+          cacheRead: 0,
+          cacheWrite: 0,
+          totalTokens: 150,
+          cost: { total: 0.001 },
+        },
+      },
+    };
+
+    const result = extractTurnStats(event);
+
+    expect(result?.contextUsage).toBeUndefined();
+  });
+});