balls 0.4.1

//! Unit tests for the leaf helpers in `review_safety`. The end-to-end
//! transactional behavior of `commit_squash_and_flip` is exercised by
//! `tests/review.rs`; this file targets the small leaf helpers and
//! their failure paths so coverage stays at 100%.

use super::*;
use crate::store::Store;
use std::path::Path;
use std::process::Command;
use tempfile::tempdir;

fn raw_git(path: &Path, args: &[&str]) -> std::process::Output {
    let mut cmd = Command::new("git");
    cmd.current_dir(path).args(args);
    for var in crate::git::GIT_ENV_VARS {
        cmd.env_remove(var);
    }
    cmd.output().expect("spawn git")
}

fn init_repo(path: &Path) {
    let run = |args: &[&str]| {
        assert!(raw_git(path, args).status.success());
    };
    run(&["init", "-q", "-b", "main"]);
    run(&["config", "user.email", "test@example.com"]);
    run(&["config", "user.name", "test"]);
    run(&["config", "commit.gpgsign", "false"]);
    run(&["commit", "--allow-empty", "-m", "init"]);
}

#[test]
fn add_user_changes_stages_normal_files() {
    let td = tempdir().unwrap();
    init_repo(td.path());
    std::fs::write(td.path().join("a.txt"), "hi").unwrap();
    add_user_changes(td.path()).unwrap();
    let staged =
        String::from_utf8(raw_git(td.path(), &["diff", "--cached", "--name-only"]).stdout).unwrap();
    assert!(staged.contains("a.txt"), "got: {staged}");
}

#[test]
fn add_user_changes_excludes_balls_runtime_paths() {
    let td = tempdir().unwrap();
    init_repo(td.path());
    std::fs::create_dir_all(td.path().join(".balls/local")).unwrap();
    std::fs::create_dir_all(td.path().join(".balls/tasks")).unwrap();
    std::fs::create_dir_all(td.path().join(".balls/worktree")).unwrap();
    std::fs::write(td.path().join(".balls/local/lock"), "x").unwrap();
    std::fs::write(td.path().join(".balls/tasks/t.json"), "x").unwrap();
    std::fs::write(td.path().join(".balls/worktree/x"), "x").unwrap();
    std::fs::write(td.path().join("user.txt"), "ok").unwrap();
    add_user_changes(td.path()).unwrap();
    let staged =
        String::from_utf8(raw_git(td.path(), &["diff", "--cached", "--name-only"]).stdout).unwrap();
    assert!(staged.contains("user.txt"), "got: {staged}");
    for p in RUNTIME_PATHS {
        assert!(
            !staged.lines().any(|l| l.starts_with(p)),
            "runtime path {p} leaked into staging: {staged}"
        );
    }
}

#[test]
fn commit_touches_runtime_flags_runtime_paths() {
    let td = tempdir().unwrap();
    init_repo(td.path());
    std::fs::create_dir_all(td.path().join(".balls/local")).unwrap();
    std::fs::write(td.path().join(".balls/local/lock"), "x").unwrap();
    assert!(raw_git(td.path(), &["add", "-A"]).status.success());
    assert!(raw_git(td.path(), &["commit", "-m", "bad"])
        .status
        .success());
    let sha = String::from_utf8(raw_git(td.path(), &["rev-parse", "HEAD"]).stdout)
        .unwrap()
        .trim()
        .to_string();
    let hits = commit_touches_runtime(td.path(), &sha).unwrap();
    assert_eq!(hits, vec![".balls/local/lock".to_string()]);
}

#[test]
fn commit_touches_runtime_empty_for_clean_commit() {
    let td = tempdir().unwrap();
    init_repo(td.path());
    std::fs::write(td.path().join("ok.txt"), "x").unwrap();
    assert!(raw_git(td.path(), &["add", "-A"]).status.success());
    assert!(raw_git(td.path(), &["commit", "-m", "ok"]).status.success());
    let sha = String::from_utf8(raw_git(td.path(), &["rev-parse", "HEAD"]).stdout)
        .unwrap()
        .trim()
        .to_string();
    let hits = commit_touches_runtime(td.path(), &sha).unwrap();
    assert!(hits.is_empty(), "expected empty, got {hits:?}");
}


#[test]
fn commit_squash_and_flip_rewinds_main_when_squash_carries_runtime() {
    // Defense-in-depth path: if a work branch tip somehow carries a
    // runtime file all the way past the wip step (here, by skipping
    // `add_user_changes` entirely and committing the runtime path
    // directly on the work branch off a base that doesn't have it),
    // the post-squash check must reject the review and rewind main.
    let td = tempdir().unwrap();
    init_repo(td.path());
    let store = Store::init(td.path(), false, None).unwrap();
    let pre_main = git::git_resolve_sha(td.path(), "HEAD").unwrap();

    // Build a work branch off main with a single commit that tracks
    // a runtime path. Bypasses `add_user_changes`, which is exactly
    // the scenario the post-squash check is the backstop for.
    let run = |args: &[&str]| {
        assert!(raw_git(td.path(), args).status.success(), "git {args:?}");
    };
    run(&["checkout", "-q", "-b", "work/bl-test"]);
    std::fs::create_dir_all(td.path().join(".balls/local")).unwrap();
    std::fs::write(td.path().join(".balls/local/x"), "x").unwrap();
    run(&["add", "-f", ".balls/local/x"]);
    run(&["commit", "-m", "runtime"]);
    run(&["checkout", "-q", "main"]);

    let err = commit_squash_and_flip(
        &store,
        "bl-test",
        "work/bl-test",
        "msg [bl-test]",
        None,
        "test",
        &pre_main,
        "main",
    )
    .unwrap_err();
    assert!(
        matches!(&err, BallError::Other(s) if s.contains(".balls/local")),
        "{err:?}"
    );
    let post_main = git::git_resolve_sha(td.path(), "HEAD").unwrap();
    assert_eq!(
        pre_main, post_main,
        "main should be rewound after rejection"
    );
}

#[test]
fn rewind_main_uses_update_ref_on_bare_layout() {
    // Bare-repo path of rewind_main: a working-tree `reset --hard`
    // would fail because there is no working tree, so the helper
    // routes through `update-ref` instead. Build a Store, advance
    // main one commit, flip the repo to bare, then rewind and
    // verify main is back at the pre-advance tip.
    let td = tempdir().unwrap();
    init_repo(td.path());
    let store = Store::init(td.path(), false, None).unwrap();
    let pre = git::git_resolve_sha(td.path(), "HEAD").unwrap();

    let run = |args: &[&str]| {
        assert!(raw_git(td.path(), args).status.success(), "git {args:?}");
    };
    run(&["commit", "--allow-empty", "-m", "advance"]);
    let advanced = git::git_resolve_sha(td.path(), "HEAD").unwrap();
    assert_ne!(pre, advanced, "sanity: HEAD advanced");

    run(&["config", "core.bare", "true"]);
    rewind_main(&store, "main", &pre).unwrap();
    let after = git::git_resolve_sha(td.path(), "refs/heads/main").unwrap();
    assert_eq!(pre, after, "bare rewind must move main back via update-ref");
}

#[test]
fn runtime_in_squash_error_pluralizes_correctly() {
    let one = runtime_in_squash_error("bl-aaaa", &[".balls/local".into()]);
    assert!(
        matches!(&one, BallError::Other(s) if s.contains("path .balls/local")),
        "{one:?}"
    );
    let many = runtime_in_squash_error("bl-bbbb", &[".balls/local".into(), ".balls/tasks".into()]);
    assert!(
        matches!(&many, BallError::Other(s) if s.contains("paths .balls/local, .balls/tasks")),
        "{many:?}"
    );
}