//! The daemon binary as a run's entry wrapper (`run_cgroup::RunCgroup::entry`
//! sets these variables; its unit test pins the names).

use std::process::Command;

#[test]
fn the_entry_enters_the_cgroup_raises_the_score_and_execs_the_program() {
    let cgroup = std::env::temp_dir().join(format!("hive-run-entry-{}", std::process::id()));
    std::fs::create_dir_all(&cgroup).expect("fake cgroup");
    let out = Command::new(env!("CARGO_BIN_EXE_hive-subagent-daemon"))
        // Stands in for `claude`: prints what it was started with.
        .env("HIVE_RUN_ENTRY_PROGRAM", "/bin/sh")
        .env("HIVE_RUN_ENTRY_CGROUP", &cgroup)
        .env("HIVE_RUN_ENTRY_OOM_SCORE_ADJ", "1000")
        .args([
            "-c",
            r#"printf '%s|%s|%s' "$(cat /proc/self/oom_score_adj)" "${HIVE_RUN_ENTRY_CGROUP-unset}" "$1""#,
            "sh",
            "an-arg",
        ])
        .output()
        .expect("the binary runs");
    let procs = std::fs::read_to_string(cgroup.join("cgroup.procs"));
    std::fs::remove_dir_all(&cgroup).ok();
    assert!(out.status.success(), "{out:?}");
    assert_eq!(String::from_utf8_lossy(&out.stdout), "1000|unset|an-arg");
    assert_eq!(procs.expect("cgroup.procs written"), "0");
}

#[test]
fn the_entry_without_a_cgroup_still_raises_the_score() {
    let out = Command::new(env!("CARGO_BIN_EXE_hive-subagent-daemon"))
        .env("HIVE_RUN_ENTRY_PROGRAM", "/bin/sh")
        .env_remove("HIVE_RUN_ENTRY_CGROUP")
        .env("HIVE_RUN_ENTRY_OOM_SCORE_ADJ", "1000")
        .args(["-c", "cat /proc/self/oom_score_adj"])
        .output()
        .expect("the binary runs");
    assert!(out.status.success(), "{out:?}");
    assert_eq!(String::from_utf8_lossy(&out.stdout).trim(), "1000");
}