@ricky-stevens/context-guardian 2.1.0

package/test/content.test.mjs

@@ -0,0 +1,86 @@
+import assert from "node:assert/strict";
+import { describe, it } from "node:test";
+import { contentBytesOf, flattenContent } from "../lib/content.mjs";
+
+describe("flattenContent", () => {
+	it("returns empty string for null/undefined", () => {
+		assert.equal(flattenContent(null), "");
+		assert.equal(flattenContent(undefined), "");
+		assert.equal(flattenContent(""), "");
+	});
+
+	it("returns string content as-is", () => {
+		assert.equal(flattenContent("hello world"), "hello world");
+	});
+
+	it("extracts text blocks from array content", () => {
+		const content = [
+			{ type: "text", text: "first" },
+			{ type: "tool_use", id: "t1", name: "Read", input: {} },
+			{ type: "text", text: "second" },
+		];
+		assert.equal(flattenContent(content), "first\nsecond");
+	});
+
+	it("returns empty string for array with no text blocks", () => {
+		const content = [
+			{ type: "tool_use", id: "t1", name: "Read", input: {} },
+			{ type: "tool_result", tool_use_id: "t1", content: "result" },
+		];
+		assert.equal(flattenContent(content), "");
+	});
+
+	it("returns empty string for non-string non-array types", () => {
+		assert.equal(flattenContent(42), "");
+		assert.equal(flattenContent({}), "");
+		assert.equal(flattenContent(true), "");
+	});
+
+	it("handles single text block in array", () => {
+		assert.equal(flattenContent([{ type: "text", text: "only" }]), "only");
+	});
+});
+
+describe("contentBytesOf", () => {
+	it("returns 0 for null/undefined", () => {
+		assert.equal(contentBytesOf(null), 0);
+		assert.equal(contentBytesOf(undefined), 0);
+	});
+
+	it("counts bytes for string content", () => {
+		assert.equal(contentBytesOf("hello"), 5);
+		// Multi-byte: é is 2 bytes in UTF-8
+		assert.equal(contentBytesOf("café"), 5);
+	});
+
+	it("counts text bytes from array blocks", () => {
+		const content = [
+			{ type: "text", text: "hello" },
+			{ type: "text", text: "world" },
+		];
+		assert.equal(contentBytesOf(content), 10);
+	});
+
+	it("counts tool input bytes", () => {
+		const content = [{ type: "tool_use", input: { key: "val" } }];
+		const expected = Buffer.byteLength(JSON.stringify({ key: "val" }), "utf8");
+		assert.equal(contentBytesOf(content), expected);
+	});
+
+	it("handles nested content recursively", () => {
+		const content = [
+			{ type: "tool_result", content: [{ type: "text", text: "nested" }] },
+		];
+		assert.equal(contentBytesOf(content), 6);
+	});
+
+	it("returns 0 for non-string non-array", () => {
+		assert.equal(contentBytesOf(42), 0);
+		assert.equal(contentBytesOf({}), 0);
+	});
+
+	it("skips blocks without text or input", () => {
+		const content = [{ type: "thinking", thinking: "hmm" }];
+		assert.equal(contentBytesOf(content), 0);
+	});
+});

package/test/diagnostics.test.mjs

@@ -0,0 +1,188 @@
+import assert from "node:assert/strict";
+import { execFileSync } from "node:child_process";
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import { afterEach, beforeEach, describe, it } from "node:test";
+
+// diagnostics.mjs is a CLI script (not a module with exports).
+// It reads env/args and writes JSON to stdout. We test it by spawning.
+
+const DIAG_PATH = path.resolve("lib/diagnostics.mjs");
+
+let tmpDir;
+let dataDir;
+
+beforeEach(() => {
+	tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "cg-diag-"));
+	dataDir = path.join(tmpDir, "data");
+	fs.mkdirSync(dataDir, { recursive: true });
+});
+
+afterEach(() => {
+	fs.rmSync(tmpDir, { recursive: true, force: true });
+});
+
+function runDiag(args = [], env = {}) {
+	const stdout = execFileSync("node", [DIAG_PATH, ...args], {
+		encoding: "utf8",
+		timeout: 5000,
+		env: {
+			...process.env,
+			CLAUDE_PLUGIN_DATA: dataDir,
+			HOME: os.homedir(),
+			...env,
+		},
+	});
+	return JSON.parse(stdout);
+}
+
+// ===========================================================================
+// Basic output structure
+// ===========================================================================
+
+describe("diagnostics output", () => {
+	it("returns valid JSON with checks array", () => {
+		const result = runDiag(["test-session"]);
+		assert.ok(Array.isArray(result.checks));
+		assert.ok(result.checks.length > 0);
+	});
+
+	it("each check has name, ok, and detail fields", () => {
+		const result = runDiag(["test-session"]);
+		for (const check of result.checks) {
+			assert.ok(
+				typeof check.name === "string",
+				`name should be string: ${JSON.stringify(check)}`,
+			);
+			assert.ok(
+				typeof check.ok === "boolean",
+				`ok should be boolean: ${JSON.stringify(check)}`,
+			);
+			assert.ok(
+				typeof check.detail === "string",
+				`detail should be string: ${JSON.stringify(check)}`,
+			);
+		}
+	});
+});
+
+// ===========================================================================
+// data_dir check
+// ===========================================================================
+
+describe("data_dir check", () => {
+	it("passes when data dir is writable", () => {
+		const result = runDiag(["test-session"]);
+		const check = result.checks.find((c) => c.name === "data_dir");
+		assert.ok(check);
+		assert.equal(check.ok, true);
+		// detail contains whichever data dir the diagnostics resolved to
+		assert.ok(typeof check.detail === "string" && check.detail.length > 0);
+	});
+});
+
+// ===========================================================================
+// state_file check
+// ===========================================================================
+
+describe("state_file check", () => {
+	it("fails when state file is missing", () => {
+		const result = runDiag(["nonexistent-session"]);
+		const check = result.checks.find((c) => c.name === "state_file");
+		assert.ok(check);
+		assert.equal(check.ok, false);
+	});
+
+	it("passes when state file exists", () => {
+		fs.writeFileSync(
+			path.join(dataDir, "state-my-session.json"),
+			JSON.stringify({ transcript_path: "/tmp/fake.jsonl" }),
+		);
+		const result = runDiag(["my-session"]);
+		const check = result.checks.find((c) => c.name === "state_file");
+		assert.ok(check);
+		assert.equal(check.ok, true);
+	});
+});
+
+// ===========================================================================
+// transcript check
+// ===========================================================================
+
+describe("transcript check", () => {
+	it("fails when no state file exists (skipped)", () => {
+		const result = runDiag(["no-state"]);
+		const check = result.checks.find((c) => c.name === "transcript");
+		assert.ok(check);
+		assert.equal(check.ok, false);
+		assert.ok(
+			check.detail.includes("no state file") ||
+				check.detail.includes("Skipped"),
+		);
+	});
+
+	it("passes when transcript file exists", () => {
+		const transcriptPath = path.join(tmpDir, "transcript.jsonl");
+		fs.writeFileSync(transcriptPath, "{}");
+		fs.writeFileSync(
+			path.join(dataDir, "state-tx-session.json"),
+			JSON.stringify({ transcript_path: transcriptPath }),
+		);
+		const result = runDiag(["tx-session"]);
+		const check = result.checks.find((c) => c.name === "transcript");
+		assert.ok(check);
+		assert.equal(check.ok, true);
+	});
+
+	it("fails when transcript path in state does not exist on disk", () => {
+		fs.writeFileSync(
+			path.join(dataDir, "state-bad-tx.json"),
+			JSON.stringify({ transcript_path: "/nonexistent/path.jsonl" }),
+		);
+		const result = runDiag(["bad-tx"]);
+		const check = result.checks.find((c) => c.name === "transcript");
+		assert.ok(check);
+		assert.equal(check.ok, false);
+		assert.ok(check.detail.includes("Not found"));
+	});
+});
+
+// ===========================================================================
+// plugin_root check
+// ===========================================================================
+
+describe("plugin_root check", () => {
+	it("passes (inferred from diagnostics.mjs location)", () => {
+		const result = runDiag(["test-session"]);
+		const check = result.checks.find((c) => c.name === "plugin_root");
+		assert.ok(check);
+		assert.equal(check.ok, true);
+	});
+});
+
+// ===========================================================================
+// hooks check
+// ===========================================================================
+
+describe("hooks check", () => {
+	it("passes when all 4 hook files exist", () => {
+		const result = runDiag(["test-session"]);
+		const check = result.checks.find((c) => c.name === "hooks");
+		assert.ok(check);
+		assert.equal(check.ok, true);
+		assert.ok(check.detail.includes("4 hook files"));
+	});
+});
+
+// ===========================================================================
+// Always exits 0
+// ===========================================================================
+
+describe("exit behaviour", () => {
+	it("always exits 0 even with missing session", () => {
+		// If it threw, execFileSync would throw too
+		const result = runDiag(["completely-bogus-session-id"]);
+		assert.ok(Array.isArray(result.checks));
+	});
+});