Redredchen01
bb6a1321fb
Multi-line tmux strip (context, quota, tools, todos, git), theme sync with Grok UI, full/essential/minimal presets, one-shot installer, EN+ZH docs, and 40 unit tests.
84 lines
2.7 KiB
TypeScript
84 lines
2.7 KiB
TypeScript
import { describe, it } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import path from "node:path";
|
|
import { fileURLToPath } from "node:url";
|
|
import { loadSnapshotFromDir } from "../src/session.js";
|
|
import {
|
|
formatStatusBlock,
|
|
formatTmuxStatusLines,
|
|
writeStatusFiles,
|
|
} from "../src/status.js";
|
|
import { PRESET_FULL } from "../src/hud-config.js";
|
|
import { THEME_TOKYONIGHT } from "../src/theme.js";
|
|
import fs from "node:fs";
|
|
import os from "node:os";
|
|
|
|
const root = path.join(path.dirname(fileURLToPath(import.meta.url)), "..", "..");
|
|
const fixture = path.join(root, "fixtures", "session");
|
|
|
|
describe("Claude-HUD parity status", () => {
|
|
it("formatStatusBlock has model line + context + usage lines", () => {
|
|
const snap = loadSnapshotFromDir(fixture)!;
|
|
const text = formatStatusBlock(
|
|
snap,
|
|
{
|
|
available: true,
|
|
percent: 23,
|
|
period: "weekly",
|
|
used: 18000,
|
|
limit: 150000,
|
|
resetsIn: "4d",
|
|
message: "GrokBuild 10%",
|
|
},
|
|
PRESET_FULL,
|
|
);
|
|
assert.match(text, /\[Grok 4\.5\]/);
|
|
assert.match(text, /Context/);
|
|
assert.match(text, /37%/);
|
|
assert.match(text, /Usage/);
|
|
assert.match(text, /23%/);
|
|
// multi-line
|
|
assert.ok(text.split("\n").length >= 2);
|
|
});
|
|
|
|
it("formatTmuxStatusLines returns 3 rows in full preset", () => {
|
|
const snap = loadSnapshotFromDir(fixture)!;
|
|
// inject todos/agents for line 3
|
|
snap.todos = [
|
|
{ content: "Ship HUD parity", status: "in_progress" },
|
|
{ content: "Write tests", status: "completed" },
|
|
];
|
|
const lines = formatTmuxStatusLines(
|
|
snap,
|
|
{ available: true, percent: 22, period: "weekly" },
|
|
THEME_TOKYONIGHT,
|
|
PRESET_FULL,
|
|
);
|
|
assert.equal(lines.length, 3);
|
|
assert.match(lines[0]!, /Grok 4\.5|CoachFlow/);
|
|
assert.match(lines[1]!, /Context|Usage|22%|37%/);
|
|
});
|
|
|
|
it("writeStatusFiles emits tmux-lines.txt", () => {
|
|
const snap = loadSnapshotFromDir(fixture)!;
|
|
const tmp = fs.mkdtempSync(path.join(os.tmpdir(), "hud-full-"));
|
|
// point grok home at tmp with config
|
|
fs.mkdirSync(path.join(tmp, "hud"), { recursive: true });
|
|
fs.writeFileSync(
|
|
path.join(tmp, "hud", "config.json"),
|
|
JSON.stringify(PRESET_FULL, null, 2),
|
|
);
|
|
// session needs to be written via writeStatusFiles with grokHome=tmp
|
|
// but session is loaded from fixture; OK
|
|
const r = writeStatusFiles(
|
|
snap,
|
|
{ available: true, percent: 20, period: "weekly" },
|
|
tmp,
|
|
);
|
|
assert.ok(fs.existsSync(path.join(tmp, "hud", "tmux-lines.txt")));
|
|
assert.ok(r.tmuxLines.length >= 2);
|
|
assert.match(r.full, /Context/);
|
|
fs.rmSync(tmp, { recursive: true, force: true });
|
|
});
|
|
});
|