ai-dispatch 8.100.6

Multi-AI CLI team orchestrator
// Tests for dispatch preparation result-file defaults.
// Exports: none.
// Deps: super::prepare_dispatch, crate::store, RunArgs.

use super::*;
use std::sync::Arc;

#[test]
fn report_mode_dirty_skip_uses_narrow_predicate() {
    use crate::agent::classifier::TaskCategory;

    assert!(crate::cmd::report_mode::skips_dirty_enforcement(
        "Cross-audit the nonce fix",
        false,
        TaskCategory::Research,
    ));
    assert!(!crate::cmd::report_mode::skips_dirty_enforcement(
        "review and fix the parser bug",
        false,
        TaskCategory::ComplexImpl,
    ));
    assert!(crate::cmd::report_mode::skips_dirty_enforcement(
        "anything",
        true,
        TaskCategory::ComplexImpl,
    ));
    assert!(crate::cmd::report_mode::skips_dirty_enforcement(
        "code review of module X",
        false,
        TaskCategory::Research,
    ));
}

#[test]
fn prepare_dispatch_updates_log_path_when_id_is_auto_suffixed() {
    let temp = tempfile::tempdir().unwrap();
    let _guard = crate::paths::AidHomeGuard::set(temp.path());
    let store = Arc::new(Store::open_memory().unwrap());
    let mut first = RunArgs {
        agent_name: "codex".to_string(),
        existing_task_id: Some(TaskId("t-ebcf".to_string())),
        prompt: "Investigate a concrete task routing bug.".to_string(),
        ..Default::default()
    };
    prepare_dispatch(&store, &mut first).unwrap();

    let mut second = RunArgs {
        agent_name: "codex".to_string(),
        existing_task_id: Some(TaskId("t-ebcf".to_string())),
        prompt: "Investigate a concrete task routing bug again.".to_string(),
        ..Default::default()
    };
    let prepared = prepare_dispatch(&store, &mut second).unwrap();

    let expected = crate::paths::log_path("t-ebcf-2");
    let saved = store.get_task("t-ebcf-2").unwrap().unwrap();
    assert_eq!(prepared.task_id.as_str(), "t-ebcf-2");
    assert_eq!(prepared.log_path, expected);
    assert_eq!(saved.log_path.as_deref(), Some(expected.to_str().unwrap()));
}

#[test]
fn prepare_dispatch_uses_task_specific_audit_result_file() {
    let store = Arc::new(Store::open_memory().unwrap());
    let mut args = RunArgs {
        agent_name: "codex".to_string(),
        prompt: "Review the implementation and list findings.".to_string(),
        read_only: true,
        ..Default::default()
    };

    let prepared = prepare_dispatch(&store, &mut args).unwrap();

    assert_eq!(
        args.result_file.as_deref(),
        Some(crate::cmd::report_mode::task_result_file(prepared.task_id.as_str()).as_str())
    );
}

#[test]
fn prepare_dispatch_skips_auto_result_file_when_output_is_set() {
    let store = Arc::new(Store::open_memory().unwrap());
    let mut args = RunArgs {
        agent_name: "codex".to_string(),
        prompt: "Review the implementation and list findings.".to_string(),
        read_only: true,
        output: Some("audit.md".to_string()),
        ..Default::default()
    };

    prepare_dispatch(&store, &mut args).unwrap();

    assert_eq!(args.result_file, None);
}

#[test]
fn prepare_dispatch_keeps_dirty_enforcement_for_write_intent_result_file() {
    let store = Arc::new(Store::open_memory().unwrap());
    let mut args = RunArgs {
        agent_name: "codex".to_string(),
        prompt: "review and fix the parser bug".to_string(),
        result_file: Some("out.md".to_string()),
        ..Default::default()
    };

    prepare_dispatch(&store, &mut args).unwrap();

    assert!(!args.audit_report_mode);
    assert_eq!(args.result_file.as_deref(), Some("out.md"));
}