oh-my-opencode/packages/omo-codex/plugin/components/ultrawork/hooks/ultrawork-hooks.test.mjs

import assert from "node:assert/strict";
import { spawn } from "node:child_process";
import { lstat, mkdtemp, readFile, rm, stat } from "node:fs/promises";
import { tmpdir } from "node:os";
import { dirname, join } from "node:path";
import { fileURLToPath } from "node:url";
import test from "node:test";

const hookDir = dirname(fileURLToPath(import.meta.url));
const pluginRoot = dirname(hookDir);
const detectorPath = join(hookDir, "ultrawork-detector.py");
const syncAgentsPath = join(hookDir, "sync-agents.py");

async function makeTempDir() {
	return mkdtemp(join(tmpdir(), "codex-ultrawork-"));
}

async function runPython(scriptPath, input, env = {}) {
	return new Promise((resolve, reject) => {
		const child = spawn("python3", [scriptPath], {
			env: {
				...process.env,
				...env,
			},
		});
		let stdout = "";
		let stderr = "";
		child.stdout.setEncoding("utf8");
		child.stderr.setEncoding("utf8");
		child.stdout.on("data", (chunk) => {
			stdout += chunk;
		});
		child.stderr.on("data", (chunk) => {
			stderr += chunk;
		});
		child.once("error", reject);
		child.once("close", (code, signal) => {
			resolve({ code, signal, stdout, stderr });
		});
		child.stdin.end(input);
	});
}

test("#given session start #when sync hook runs #then installs bundled reviewer agent", async () => {
	const codexHome = await makeTempDir();
	try {
		const result = await runPython(
			syncAgentsPath,
			'{"hook_event_name":"SessionStart"}',
			{ CODEX_HOME: codexHome },
		);

		assert.equal(result.code, 0);
		assert.equal(result.signal, null);
		assert.equal(result.stdout, "");
		assert.equal(result.stderr, "");

		const targetPath = join(codexHome, "agents", "codex-ultrawork-reviewer.toml");
		const targetStat = await lstat(targetPath);
		assert.equal(targetStat.isFile(), true);
		assert.equal(targetStat.isSymbolicLink(), false);
		const syncedAgent = await readFile(targetPath, "utf8");
		assert.match(syncedAgent, /^name = "codex-ultrawork-reviewer"$/m);
		assert.match(syncedAgent, /^model = "gpt-5.2"$/m);
		assert.match(syncedAgent, /^model_reasoning_effort = "xhigh"$/m);
		assert.match(syncedAgent, /^developer_instructions = """/m);
	} finally {
		await rm(codexHome, { recursive: true, force: true });
	}
});

test("#given malformed session payload #when sync hook runs #then exits zero without output", async () => {
	const codexHome = await makeTempDir();
	try {
		const result = await runPython(syncAgentsPath, "{", { CODEX_HOME: codexHome });

		assert.equal(result.code, 0);
		assert.equal(result.signal, null);
		assert.equal(result.stdout, "");
		assert.equal(result.stderr, "");
		await assert.rejects(
			stat(join(codexHome, "agents", "codex-ultrawork-reviewer.toml")),
			/code: 'ENOENT'|ENOENT/,
		);
	} finally {
		await rm(codexHome, { recursive: true, force: true });
	}
});

test("#given ultrawork prompt #when detector runs #then emits directive", async () => {
	const payload = JSON.stringify({
		hook_event_name: "UserPromptSubmit",
		prompt: "please ulw this change",
	});

	const result = await runPython(detectorPath, payload);

	assert.equal(result.code, 0);
	assert.equal(result.signal, null);
	assert.equal(result.stderr, "");
	assert.match(result.stdout, /^<ultrawork-mode>/);
	assert.match(result.stdout, /First user-visible line this turn MUST be exactly:/);
});

test("#given ultrawork prompt #when detector runs #then directive keeps goal budget unlimited", async () => {
	const payload = JSON.stringify({
		hook_event_name: "UserPromptSubmit",
		prompt: "please ultrawork this change",
	});

	const result = await runPython(detectorPath, payload);

	assert.equal(result.code, 0);
	assert.equal(result.signal, null);
	assert.equal(result.stderr, "");
	assert.match(result.stdout, /Goals are\s+unlimited/);
	assert.match(result.stdout, /exactly `objective` and `status` fields/);
	assert.doesNotMatch(result.stdout, /token[_-]?budget/i);
	assert.doesNotMatch(result.stdout, /200000/i);
});

test("#given identifier-like ulw #when detector runs #then does not emit directive", async () => {
	const payload = JSON.stringify({
		hook_event_name: "UserPromptSubmit",
		prompt: "refactor ulw_helper.ts",
	});

	const result = await runPython(detectorPath, payload);

	assert.equal(result.code, 0);
	assert.equal(result.signal, null);
	assert.equal(result.stdout, "");
	assert.equal(result.stderr, "");
});

test("#given hook manifest #when read #then registers prompt and session hooks", async () => {
	const manifest = JSON.parse(await readFile(join(hookDir, "hooks.json"), "utf8"));

	assert.match(
		manifest.hooks.UserPromptSubmit[0].hooks[0].command,
		/ultrawork-detector\.py/,
	);
	assert.match(
		manifest.hooks.SessionStart[0].hooks[0].command,
		/sync-agents\.py/,
	);
	assert.equal(pluginRoot.endsWith("components/ultrawork"), true);
});

test("#given component package #when inspected #then plugin identity is owned by aggregate root", async () => {
	const pkg = JSON.parse(await readFile(join(pluginRoot, "package.json"), "utf8"));

	assert.equal(pkg.files.includes(".codex-plugin"), false);
	await assert.rejects(
		readFile(join(pluginRoot, ".codex-plugin", "plugin.json"), "utf8"),
		/code: 'ENOENT'|ENOENT/,
	);
});