use super::{
Mode, agent_uses_claude_hooks, apply_branch, but_available, install_claude_hooks, is_active,
on_done_command, task_worktree_integration_plan,
};
use serde_json::{Value, json};
use std::fs;
use std::sync::{Mutex, MutexGuard, OnceLock};
fn env_lock() -> MutexGuard<'static, ()> {
static LOCK: OnceLock<Mutex<()>> = OnceLock::new();
match LOCK.get_or_init(|| Mutex::new(())).lock() {
Ok(guard) => guard,
Err(poison) => poison.into_inner(),
}
}
fn gitbutler_test_present() -> bool {
std::env::var("AID_GITBUTLER_TEST_PRESENT")
.map(|value| matches!(value.trim().to_ascii_lowercase().as_str(), "1" | "true" | "yes"))
.unwrap_or(false)
}
#[test]
fn gitbutler_mode_parse_round_trip() {
for expected in [Mode::Off, Mode::Auto, Mode::Always] {
let parsed = Mode::from_str(expected.as_str()).unwrap();
assert_eq!(parsed, expected);
}
}
#[test]
fn gitbutler_mode_rejects_unknown_value() {
assert!(Mode::from_str("sometimes").is_err());
}
#[test]
fn gitbutler_is_active_is_false_for_off() {
assert!(!is_active(Mode::Off));
}
#[test]
fn gitbutler_is_active_is_true_for_always() {
assert!(is_active(Mode::Always));
}
#[test]
#[ignore = "process-wide cache; enable when explicitly validating detection"]
fn gitbutler_but_available_respects_test_override() {
unsafe {
std::env::set_var("AID_GITBUTLER_TEST_PRESENT", "1");
}
assert!(but_available());
}
#[test]
fn apply_branch_errors_when_test_but_detection_is_disabled() {
if gitbutler_test_present() {
return;
}
let temp = tempfile::tempdir().unwrap();
let error = apply_branch(temp.path(), "lane-branch").unwrap_err().to_string();
assert_eq!(error, "GitButler CLI not found. Install: https://gitbutler.com");
}
#[test]
fn apply_branch_real_execution_requires_test_override() {
if !gitbutler_test_present() {
return;
}
let temp = tempfile::tempdir().unwrap();
let result = apply_branch(temp.path(), "lane-branch");
assert!(result.is_err());
}
#[test]
fn agent_uses_claude_hooks_matches_known_agents() {
assert!(agent_uses_claude_hooks("claude"));
assert!(agent_uses_claude_hooks("claude-code"));
assert!(!agent_uses_claude_hooks("codex"));
assert!(!agent_uses_claude_hooks("cursor"));
assert!(!agent_uses_claude_hooks("opencode"));
assert!(!agent_uses_claude_hooks("gemini"));
}
#[test]
fn install_claude_hooks_writes_expected_settings_json() {
let temp = tempfile::tempdir().unwrap();
install_claude_hooks(temp.path()).unwrap();
let value: Value =
serde_json::from_slice(&fs::read(temp.path().join(".claude/settings.local.json")).unwrap()).unwrap();
assert_eq!(value["hooks"]["PreToolUse"][0]["matcher"].as_str(), Some("Edit|MultiEdit|Write"));
assert_eq!(value["hooks"]["PreToolUse"][0]["hooks"][0]["command"].as_str(), Some("but claude pre-tool"));
assert_eq!(value["hooks"]["PostToolUse"][0]["matcher"].as_str(), Some("Edit|MultiEdit|Write"));
assert_eq!(value["hooks"]["PostToolUse"][0]["hooks"][0]["command"].as_str(), Some("but claude post-tool"));
assert_eq!(value["hooks"]["Stop"][0]["hooks"][0]["command"].as_str(), Some("but claude stop"));
}
#[test]
fn install_claude_hooks_preserves_existing_settings_keys() {
let temp = tempfile::tempdir().unwrap();
let settings_dir = temp.path().join(".claude");
fs::create_dir_all(&settings_dir).unwrap();
fs::write(
settings_dir.join("settings.local.json"),
serde_json::to_vec_pretty(&json!({
"theme": "dark",
"hooks": {
"Notification": [{
"hooks": [{"type": "command", "command": "echo notify"}]
}]
}
}))
.unwrap(),
)
.unwrap();
install_claude_hooks(temp.path()).unwrap();
let value: Value =
serde_json::from_slice(&fs::read(temp.path().join(".claude/settings.local.json")).unwrap()).unwrap();
assert_eq!(value["theme"].as_str(), Some("dark"));
assert_eq!(value["hooks"]["Notification"][0]["hooks"][0]["command"].as_str(), Some("echo notify"));
assert_eq!(value["hooks"]["Stop"][0]["hooks"][0]["command"].as_str(), Some("but claude stop"));
}
#[test]
fn install_claude_hooks_is_idempotent_across_reruns() {
let temp = tempfile::tempdir().unwrap();
install_claude_hooks(temp.path()).unwrap();
install_claude_hooks(temp.path()).unwrap();
let value: Value =
serde_json::from_slice(&fs::read(temp.path().join(".claude/settings.local.json")).unwrap()).unwrap();
assert_eq!(value["hooks"]["PreToolUse"].as_array().map(Vec::len), Some(1));
assert_eq!(value["hooks"]["PostToolUse"].as_array().map(Vec::len), Some(1));
assert_eq!(value["hooks"]["Stop"].as_array().map(Vec::len), Some(1));
}
#[test]
fn on_done_command_contains_gitbutler_commit_shell_command() {
let temp = tempfile::tempdir().unwrap();
let command = on_done_command(temp.path());
let worktree = temp.path().to_string_lossy();
assert!(command.contains("but -C"));
assert!(command.contains(worktree.as_ref()));
assert!(command.contains("commit -i"));
assert!(command.contains("|| true"));
}
#[test]
fn task_worktree_integration_plan_skips_hooks_and_emits_hint_without_main_repo_project() {
let _guard = env_lock();
let temp = tempfile::tempdir().unwrap();
unsafe {
std::env::set_var("AID_GITBUTLER_TEST_PRESENT", "1");
std::env::set_var("AID_GITBUTLER_TEST_PROJECT_PRESENT", "0");
}
let first = task_worktree_integration_plan(temp.path(), temp.path(), Mode::Auto, "codex");
let second = task_worktree_integration_plan(temp.path(), temp.path(), Mode::Auto, "claude");
unsafe {
std::env::remove_var("AID_GITBUTLER_TEST_PRESENT");
std::env::remove_var("AID_GITBUTLER_TEST_PROJECT_PRESENT");
}
assert_eq!(first.on_done_command, None);
assert!(!first.install_claude_hooks);
assert!(first.emit_setup_hint);
assert_eq!(second.on_done_command, None);
assert!(!second.install_claude_hooks);
assert!(!second.emit_setup_hint);
}
#[test]
fn task_worktree_integration_plan_preserves_hook_modes_when_main_repo_has_project() {
let _guard = env_lock();
let temp = tempfile::tempdir().unwrap();
unsafe {
std::env::set_var("AID_GITBUTLER_TEST_PRESENT", "1");
std::env::set_var("AID_GITBUTLER_TEST_PROJECT_PRESENT", "1");
}
let claude = task_worktree_integration_plan(temp.path(), temp.path(), Mode::Auto, "claude");
let codex = task_worktree_integration_plan(temp.path(), temp.path(), Mode::Auto, "codex");
unsafe {
std::env::remove_var("AID_GITBUTLER_TEST_PRESENT");
std::env::remove_var("AID_GITBUTLER_TEST_PROJECT_PRESENT");
}
assert!(claude.install_claude_hooks);
assert_eq!(claude.on_done_command, None);
assert!(!claude.emit_setup_hint);
assert!(!codex.install_claude_hooks);
assert!(codex.on_done_command.as_deref().is_some_and(|value| value.contains("but -C")));
assert!(!codex.emit_setup_hint);
}