ai-dispatch 9.1.0

Multi-AI CLI team orchestrator
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

// Tests for disabled-agent filtering in auto-selection and fallback chains.
// Exports: test cases only.
// Deps: selection helpers, agent_config, Store, AidHomeGuard.

use super::{coding_fallback_for, select_agent_from};
use crate::agent::RunOpts;
use crate::agent_config;
use crate::paths::AidHomeGuard;
use crate::store::Store;
use crate::types::AgentKind;

fn opts() -> RunOpts {
    RunOpts {
        dir: None,
        output: None,
        model: None,
        result_file: None,
        budget: false,
        read_only: false,
        sandbox: false,
        context_files: vec![],
        session_id: None,
        env: None,
        env_forward: None,
    }
}

#[test]
fn selection_skips_disabled_available_candidate() {
    let dir = tempfile::tempdir().expect("tempdir");
    let _guard = AidHomeGuard::set(dir.path());
    agent_config::save_agent_disabled("gemini", true).expect("disable agent");
    let store = Store::open_memory().expect("store");

    let (selected, reason) = select_agent_from(
        "Explain the authentication flow and compare the docs?",
        &opts(),
        &[AgentKind::Gemini, AgentKind::Qwen],
        &store,
        None,
    );

    assert_eq!(selected, AgentKind::Qwen.as_str());
    assert!(reason.contains("qwen"));
}

#[test]
fn fallback_chain_skips_disabled_agent() {
    let dir = tempfile::tempdir().expect("tempdir");
    let _guard = AidHomeGuard::set(dir.path());
    let _agents = crate::agent::DetectAgentsGuard::set(vec![
        AgentKind::Gemini,
        AgentKind::Qwen,
        AgentKind::Codex,
    ]);
    agent_config::save_agent_disabled("qwen", true).expect("disable agent");

    let result = coding_fallback_for(&AgentKind::Gemini);

    assert_eq!(result, Some(AgentKind::Codex));
}