use std::process::{Command, Output};
fn agentsight_output(args: &[&str]) -> Output {
agentsight_output_with_env(args, &[])
}
fn agentsight_output_with_env(args: &[&str], envs: &[(&str, &std::ffi::OsStr)]) -> Output {
let output = Command::new(env!("CARGO_BIN_EXE_agentsight"))
.args(args)
.envs(envs.iter().copied())
.output()
.expect("agentsight command should run");
assert!(
output.status.success(),
"agentsight {:?} failed\nstdout:\n{}\nstderr:\n{}",
args,
String::from_utf8_lossy(&output.stdout),
String::from_utf8_lossy(&output.stderr)
);
output
}
fn agentsight_stdout(args: &[&str]) -> String {
String::from_utf8(agentsight_output(args).stdout).expect("stdout should be UTF-8")
}
fn agentsight_stdout_with_env(args: &[&str], envs: &[(&str, &std::ffi::OsStr)]) -> String {
String::from_utf8(agentsight_output_with_env(args, envs).stdout)
.expect("stdout should be UTF-8")
}
#[test]
fn top_level_help_surfaces_perf_strace_flow() {
let help = agentsight_stdout(&["--help"]);
assert!(
help.contains("stat/top/record/report for AI agent runs"),
"{help}"
);
assert!(help.contains("stat"), "{help}");
assert!(help.contains("top"), "{help}");
assert!(help.contains("record"), "{help}");
assert!(help.contains("report"), "{help}");
assert!(help.contains("prompts"), "{help}");
assert!(help.contains("list"), "{help}");
}
#[test]
fn agent_native_summary_reads_codex_session_jsonl() {
let temp = tempfile::tempdir().expect("tempdir");
let session_dir = temp.path().join(".codex/sessions/2026/06/02");
std::fs::create_dir_all(&session_dir).expect("session dir");
std::fs::write(
session_dir.join("rollout-test.jsonl"),
concat!(
"{\"type\":\"turn_context\",\"payload\":{\"model\":\"gpt-5.5\"}}\n",
"{\"type\":\"event_msg\",\"payload\":{\"type\":\"token_count\",\"info\":{\"total_token_usage\":{\"input_tokens\":11,\"output_tokens\":4,\"total_tokens\":15}}}}\n",
"{\"type\":\"response_item\",\"payload\":{\"type\":\"function_call\",\"name\":\"shell\"}}\n",
),
)
.expect("codex session");
let summary =
agentsight_stdout_with_env(&["report", "--local"], &[("HOME", temp.path().as_os_str())]);
assert!(
summary.contains("agent_native_session session"),
"{summary}"
);
assert!(summary.contains("gpt-5.5"), "{summary}");
assert!(summary.contains("15 tokens"), "{summary}");
assert!(summary.contains("shell(1)"), "{summary}");
}
#[test]
fn top_without_db_uses_live_process_view() {
let top = agentsight_stdout(&["top", "--once"]);
assert!(top.contains("AgentSight top -"), "{top}");
assert!(top.contains("live sessions"), "{top}");
assert!(top.contains("SESSION"), "{top}");
assert!(top.contains("AGENT"), "{top}");
assert!(top.contains("STATE"), "{top}");
assert!(top.contains("AGE"), "{top}");
assert!(top.contains("ACTIVITY"), "{top}");
assert!(top.contains("EVIDENCE"), "{top}");
}
#[test]
fn top_discovers_agent_native_sessions() {
let temp = tempfile::tempdir().expect("tempdir");
let session_dir = temp.path().join(".codex/sessions/2026/06/02");
std::fs::create_dir_all(&session_dir).expect("session dir");
std::fs::write(
session_dir.join("rollout-test.jsonl"),
concat!(
"{\"type\":\"turn_context\",\"payload\":{\"model\":\"gpt-5.5\"}}\n",
"{\"type\":\"event_msg\",\"payload\":{\"type\":\"token_count\",\"info\":{\"total_token_usage\":{\"input_tokens\":11,\"output_tokens\":4,\"total_tokens\":15}}}}\n",
"{\"type\":\"response_item\",\"payload\":{\"type\":\"function_call\",\"name\":\"shell\"}}\n",
"{\"type\":\"message\",\"content\":\"fix the test\"}\n",
),
)
.expect("codex session");
let top = agentsight_stdout_with_env(
&["top", "--once", "--limit", "20"],
&[("HOME", temp.path().as_os_str())],
);
assert!(top.contains("live sessions"), "{top}");
assert!(top.contains("codex:rollout-test"), "{top}");
assert!(top.contains("TOKENS"), "{top}");
assert!(top.contains("ACTIVITY"), "{top}");
assert!(top.contains("15"), "{top}");
assert!(top.contains("1 tool"), "{top}");
assert!(top.contains("fix the test"), "{top}");
}
#[test]
fn top_reads_active_claude_local_session_model_and_tokens() {
let temp = tempfile::tempdir().expect("tempdir");
let session_dir = temp.path().join(".claude/projects/-tmp-project");
std::fs::create_dir_all(&session_dir).expect("session dir");
std::fs::write(
session_dir.join("claude-active.jsonl"),
concat!(
"{\"type\":\"user\",\"sessionId\":\"claude-active\",\"message\":{\"content\":\"inspect the trace\"}}\n",
"{\"type\":\"assistant\",\"sessionId\":\"claude-active\",\"requestId\":\"req_1\",\"message\":{\"model\":\"claude-opus-4-6\",\"content\":[{\"type\":\"tool_use\",\"id\":\"toolu_1\",\"name\":\"Bash\",\"input\":{\"command\":\"true\"}}],\"usage\":{\"input_tokens\":3,\"cache_creation_input_tokens\":5,\"cache_read_input_tokens\":7,\"output_tokens\":11}}}\n",
"{\"type\":\"assistant\",\"sessionId\":\"claude-active\",\"requestId\":\"req_1\",\"message\":{\"model\":\"claude-opus-4-6\",\"content\":[{\"type\":\"text\",\"text\":\"done\"}],\"usage\":{\"input_tokens\":3,\"cache_creation_input_tokens\":5,\"cache_read_input_tokens\":7,\"output_tokens\":11}}}\n",
),
)
.expect("claude session");
let top = agentsight_stdout_with_env(
&["top", "--once", "--limit", "20"],
&[("HOME", temp.path().as_os_str())],
);
assert!(top.contains("claude:"), "{top}");
assert!(top.contains("inspect the trace"), "{top}");
assert!(top.contains("claude-opus-4-6"), "{top}");
assert!(top.contains("26"), "{top}");
assert!(top.contains("1 tool"), "{top}");
}