@onlooker-community/ecosystem 0.15.2 → 0.17.0

package/test/bats/governor-ledger.bats

@@ -0,0 +1,220 @@
+#!/usr/bin/env bats
+
+setup() {
+	source "${BATS_TEST_DIRNAME}/../helpers/setup.bash"
+	setup_test_env
+
+	PLUGIN_ROOT="${REPO_ROOT}/plugins/governor"
+	export CLAUDE_PLUGIN_ROOT="$PLUGIN_ROOT"
+	export ONLOOKER_EVENTS_LOG="${ONLOOKER_DIR}/logs/onlooker-events.jsonl"
+	mkdir -p "$(dirname "$ONLOOKER_EVENTS_LOG")"
+
+	export _ONLOOKER_EVENT_JS="${REPO_ROOT}/scripts/lib/onlooker-event.mjs"
+	export CLAUDE_SESSION_ID="bats-ledger-$$"
+
+	# shellcheck disable=SC1091
+	source "${REPO_ROOT}/scripts/lib/portable-lock.sh"
+	# shellcheck disable=SC1091
+	source "${PLUGIN_ROOT}/scripts/lib/governor-config.sh"
+	# shellcheck disable=SC1091
+	source "${PLUGIN_ROOT}/scripts/lib/governor-events.sh"
+	# shellcheck disable=SC1091
+	source "${PLUGIN_ROOT}/scripts/lib/governor-ledger.sh"
+
+	export ONLOOKER_DIR="${TEST_HOME}/.onlooker"
+	SID="ledger-test-session-$$"
+}
+
+_make_record() {
+	local tokens="${1:-1000}"
+	local cost="${2:-0.009}"
+	jq -n \
+		--arg ts "2026-01-01T00:00:00Z" \
+		--arg sid "$SID" \
+		--arg aid "test-agent" \
+		--argjson est "$tokens" \
+		--argjson cost "$cost" \
+		'{
+			ts: $ts,
+			session_id: $sid,
+			agent_id: $aid,
+			agent_type: "Task",
+			estimated_tokens: $est,
+			cost_usd_estimated: $cost,
+			duration_ms: 1000
+		}'
+}
+
+_make_reservation() {
+	local tokens="${1:-1000}"
+	local cost="${2:-0.009}"
+	jq -n \
+		--arg ts "2026-01-01T00:00:00Z" \
+		--arg sid "$SID" \
+		--arg aid "test-agent" \
+		--argjson est "$tokens" \
+		--argjson cost "$cost" \
+		'{
+			ts: $ts,
+			session_id: $sid,
+			agent_id: $aid,
+			agent_type: "Task",
+			estimated_tokens: $est,
+			cost_usd_estimated: $cost,
+			record_type: "reservation"
+		}'
+}
+
+_make_completion() {
+	local neg_tokens="${1:--1000}"
+	local actual_tokens="${2:-900}"
+	local cost="${3:-0.009}"
+	jq -n \
+		--arg ts "2026-01-01T00:00:00Z" \
+		--arg sid "$SID" \
+		--arg aid "test-agent" \
+		--argjson est "$neg_tokens" \
+		--argjson actual "$actual_tokens" \
+		--argjson cost "$cost" \
+		'{
+			ts: $ts,
+			session_id: $sid,
+			agent_id: $aid,
+			agent_type: "Task",
+			estimated_tokens: $est,
+			actual_tokens: $actual,
+			cost_usd_estimated: $cost,
+			duration_ms: 1500
+		}'
+}
+
+@test "governor_ledger_path returns a .jsonl path under ONLOOKER_DIR" {
+	local p
+	p=$(governor_ledger_path "test-session")
+	[[ "$p" == *".jsonl" ]]
+	[[ "$p" == *"test-session"* ]]
+}
+
+@test "governor_ledger_total_tokens returns 0 for missing ledger" {
+	local total
+	total=$(governor_ledger_total_tokens "no-such-session")
+	[ "$total" = "0" ]
+}
+
+@test "governor_ledger_total_cost returns 0 for missing ledger" {
+	local total
+	total=$(governor_ledger_total_cost "no-such-session")
+	[ "$total" = "0" ]
+}
+
+@test "governor_ledger_call_count returns 0 for missing ledger" {
+	local count
+	count=$(governor_ledger_call_count "no-such-session")
+	[ "$count" = "0" ]
+}
+
+@test "governor_ledger_append writes a record and it is readable" {
+	local record
+	record=$(_make_record 5000 0.045)
+	governor_ledger_append "$SID" "$record"
+
+	local ledger_path
+	ledger_path=$(governor_ledger_path "$SID")
+	[ -f "$ledger_path" ]
+
+	local lines
+	lines=$(awk 'END{print NR}' "$ledger_path")
+	[ "$lines" = "1" ]
+}
+
+@test "governor_ledger_total_tokens sums across multiple records" {
+	governor_ledger_append "$SID" "$(_make_record 3000 0.027)"
+	governor_ledger_append "$SID" "$(_make_record 4000 0.036)"
+	governor_ledger_append "$SID" "$(_make_record 2500 0.023)"
+
+	local total
+	total=$(governor_ledger_total_tokens "$SID")
+	[ "$total" = "9500" ]
+}
+
+@test "governor_ledger_total_cost sums across multiple records" {
+	governor_ledger_append "$SID" "$(_make_record 1000 0.01)"
+	governor_ledger_append "$SID" "$(_make_record 1000 0.02)"
+
+	local total
+	total=$(governor_ledger_total_cost "$SID")
+	# Allow slight floating-point representation variance
+	[[ "$total" =~ ^0\.03 ]]
+}
+
+@test "governor_ledger_call_count counts records" {
+	governor_ledger_append "$SID" "$(_make_record 1000 0.009)"
+	governor_ledger_append "$SID" "$(_make_record 1000 0.009)"
+	governor_ledger_append "$SID" "$(_make_record 1000 0.009)"
+
+	local count
+	count=$(governor_ledger_call_count "$SID")
+	[ "$count" = "3" ]
+}
+
+@test "governor_ledger_is_poisoned returns false for healthy ledger" {
+	governor_ledger_append "$SID" "$(_make_record 1000 0.009)"
+	run governor_ledger_is_poisoned "$SID"
+	[ "$status" -ne 0 ]
+}
+
+@test "governor_ledger_is_poisoned returns true after poison sentinel" {
+	local ledger_path
+	ledger_path=$(governor_ledger_path "$SID")
+	mkdir -p "$(dirname "$ledger_path")"
+	touch "$(governor_ledger_poison_path "$ledger_path")"
+
+	run governor_ledger_is_poisoned "$SID"
+	[ "$status" -eq 0 ]
+}
+
+@test "governor_ledger_write_direct writes a record without acquiring the write lock" {
+	local ledger_path
+	ledger_path=$(governor_ledger_path "$SID")
+	mkdir -p "$(dirname "$ledger_path")"
+
+	local record
+	record=$(_make_reservation 2000 0.018)
+	governor_ledger_write_direct "$ledger_path" "$record"
+
+	local lines
+	lines=$(awk 'END{print NR}' "$ledger_path")
+	[ "$lines" = "1" ]
+}
+
+@test "two-phase: reservation plus completion converges to actual tokens" {
+	# Phase 1: PreToolUse writes reservation with N_est = 5000
+	local reservation
+	reservation=$(_make_reservation 5000 0.045)
+	governor_ledger_append "$SID" "$reservation"
+
+	local mid_total
+	mid_total=$(governor_ledger_total_tokens "$SID")
+	[ "$mid_total" = "5000" ]
+
+	# Phase 2: PostToolUse writes completion with estimated=-5000, actual=4200
+	local completion
+	completion=$(_make_completion -5000 4200 0.045)
+	governor_ledger_append "$SID" "$completion"
+
+	# Net: 5000 + (-5000) + 4200 = 4200
+	local final_total
+	final_total=$(governor_ledger_total_tokens "$SID")
+	[ "$final_total" = "4200" ]
+}
+
+@test "governor_ledger_call_count excludes reservation records" {
+	governor_ledger_append "$SID" "$(_make_reservation 1000 0.009)"
+	governor_ledger_append "$SID" "$(_make_completion -1000 900 0.009)"
+	governor_ledger_append "$SID" "$(_make_completion -1000 800 0.009)"
+
+	local count
+	count=$(governor_ledger_call_count "$SID")
+	# 2 completions, 1 reservation — only completions count
+	[ "$count" = "2" ]
+}

package/test/bats/scribe-extract.bats

@@ -0,0 +1,102 @@
+#!/usr/bin/env bats
+
+setup() {
+  # shellcheck source=../helpers/setup.bash
+  source "${BATS_TEST_DIRNAME}/../helpers/setup.bash"
+  setup_test_env
+
+  PLUGIN_ROOT="${REPO_ROOT}/plugins/scribe"
+  export CLAUDE_PLUGIN_ROOT="$PLUGIN_ROOT"
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-config.sh"
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-extract.sh"
+}
+
+# ---------------------------------------------------------------------------
+# scribe_count_turns
+# ---------------------------------------------------------------------------
+
+@test "count_turns returns 0 for missing file" {
+  run scribe_count_turns "/nonexistent/path.jsonl"
+  [ "$status" -eq 0 ]
+  [ "$output" = "0" ]
+}
+
+@test "count_turns returns 0 for empty file" {
+  local f="${BATS_TEST_TMPDIR}/empty.jsonl"
+  touch "$f"
+  run scribe_count_turns "$f"
+  [ "$status" -eq 0 ]
+  [ "$output" = "0" ]
+}
+
+@test "count_turns counts only user-role entries" {
+  local f="${BATS_TEST_TMPDIR}/transcript.jsonl"
+  printf '%s\n' \
+    '{"role":"user","content":"hello"}' \
+    '{"role":"assistant","content":"hi"}' \
+    '{"role":"user","content":"thanks"}' \
+    '{"role":"assistant","content":"sure"}' \
+    > "$f"
+  run scribe_count_turns "$f"
+  [ "$status" -eq 0 ]
+  [ "$output" = "2" ]
+}
+
+@test "count_turns skips blank lines" {
+  local f="${BATS_TEST_TMPDIR}/sparse.jsonl"
+  printf '%s\n' \
+    '{"role":"user","content":"a"}' \
+    '' \
+    '{"role":"user","content":"b"}' \
+    > "$f"
+  run scribe_count_turns "$f"
+  [ "$status" -eq 0 ]
+  [ "$output" = "2" ]
+}
+
+# ---------------------------------------------------------------------------
+# min_turns gate (via scribe_distill return code)
+# ---------------------------------------------------------------------------
+
+@test "scribe_distill exits 2 when transcript has fewer turns than min_turns" {
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-events.sh"
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-project-key.sh"
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-distill.sh"
+
+  # Load config so min_turns is available (default: 3).
+  scribe_config_load ""
+
+  # Transcript with only 1 user turn — below default min_turns of 3.
+  local f="${BATS_TEST_TMPDIR}/short.jsonl"
+  printf '%s\n' \
+    '{"role":"user","content":"hey"}' \
+    '{"role":"assistant","content":"yo"}' \
+    > "$f"
+
+  run scribe_distill "test-session-id" "${BATS_TEST_TMPDIR}" "$f"
+  [ "$status" -eq 2 ]
+}
+
+@test "scribe_distill min_turns skip produces no stderr output" {
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-events.sh"
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-project-key.sh"
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-distill.sh"
+
+  scribe_config_load ""
+
+  local f="${BATS_TEST_TMPDIR}/short2.jsonl"
+  printf '%s\n' '{"role":"user","content":"hi"}' > "$f"
+
+  # Run in a subshell capturing stderr separately.
+  local stderr_out
+  stderr_out=$(scribe_distill "test-sid" "${BATS_TEST_TMPDIR}" "$f" 2>&1 >/dev/null) || true
+  [ -z "$stderr_out" ]
+}

package/test/bats/scribe-project-key.bats

@@ -0,0 +1,75 @@
+#!/usr/bin/env bats
+
+setup() {
+  # shellcheck source=../helpers/setup.bash
+  source "${BATS_TEST_DIRNAME}/../helpers/setup.bash"
+  setup_test_env
+
+  PLUGIN_ROOT="${REPO_ROOT}/plugins/scribe"
+  # shellcheck disable=SC1091
+  source "${PLUGIN_ROOT}/scripts/lib/scribe-project-key.sh"
+}
+
+@test "non-git directory returns empty key" {
+  local d="${BATS_TEST_TMPDIR}/non-git"
+  mkdir -p "$d"
+  run scribe_project_key "$d"
+  [ "$status" -eq 0 ]
+  [ -z "$output" ]
+}
+
+@test "git repo without remote falls back to repo-root hash" {
+  local d="${BATS_TEST_TMPDIR}/local-only-repo"
+  mkdir -p "$d"
+  git -C "$d" init -q
+  git -C "$d" config user.email t@example.com
+  git -C "$d" config user.name "Test"
+
+  local k1
+  k1=$(scribe_project_key "$d")
+  [ -n "$k1" ]
+  [ "${#k1}" -eq 12 ]
+
+  # Stability: a second call returns the same key.
+  local k2
+  k2=$(scribe_project_key "$d")
+  [ "$k1" = "$k2" ]
+}
+
+@test "git repo with remote uses remote hash, ignores local path" {
+  local a="${BATS_TEST_TMPDIR}/clone-a"
+  local b="${BATS_TEST_TMPDIR}/clone-b"
+  mkdir -p "$a" "$b"
+  for d in "$a" "$b"; do
+    git -C "$d" init -q
+    git -C "$d" config user.email t@example.com
+    git -C "$d" config user.name "Test"
+    git -C "$d" remote add origin git@github.com:org/proj.git
+  done
+
+  local ka kb
+  ka=$(scribe_project_key "$a")
+  kb=$(scribe_project_key "$b")
+  [ -n "$ka" ]
+  [ "$ka" = "$kb" ]
+}
+
+@test "different remotes yield different keys" {
+  local a="${BATS_TEST_TMPDIR}/proj-a"
+  local b="${BATS_TEST_TMPDIR}/proj-b"
+  mkdir -p "$a" "$b"
+  for d in "$a" "$b"; do
+    git -C "$d" init -q
+    git -C "$d" config user.email t@example.com
+    git -C "$d" config user.name "Test"
+  done
+  git -C "$a" remote add origin git@github.com:org/proj-a.git
+  git -C "$b" remote add origin git@github.com:org/proj-b.git
+
+  local ka kb
+  ka=$(scribe_project_key "$a")
+  kb=$(scribe_project_key "$b")
+  [ -n "$ka" ]
+  [ -n "$kb" ]
+  [ "$ka" != "$kb" ]
+}