ai-dispatch 10.18.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

// Custom-agent rate-limit marker isolation tests.
// Verifies two customs do not share a hold, and built-in paths stay unchanged.

use super::*;
use crate::paths::{self, AidHomeGuard};
use crate::types::AgentKind;

#[test]
fn two_custom_agents_do_not_share_a_hold() {
    let temp = tempfile::tempdir().unwrap();
    let _guard = AidHomeGuard::set(temp.path());
    std::fs::create_dir_all(paths::aid_dir()).unwrap();

    mark_rate_limited(
        &AgentKind::Custom,
        Some("auditor"),
        "try again at Mar 21st, 2099 2:27 PM.",
    );

    assert!(is_rate_limited(&AgentKind::Custom, Some("auditor")));
    assert!(!is_rate_limited(&AgentKind::Custom, Some("reviewer")));
    assert!(!is_rate_limited(&AgentKind::Custom, None));
    assert!(paths::aid_dir().join("rate-limit-auditor").exists());
    assert!(!paths::aid_dir().join("rate-limit-custom").exists());
    assert!(!paths::aid_dir().join("rate-limit-reviewer").exists());

    assert!(dispatch_blocking_hold(&AgentKind::Custom, Some("auditor")).is_some());
    assert!(dispatch_blocking_hold(&AgentKind::Custom, Some("reviewer")).is_none());

    assert!(clear_rate_limit(&AgentKind::Custom, Some("auditor")));
    assert!(!is_rate_limited(&AgentKind::Custom, Some("auditor")));
}

#[test]
fn builtin_marker_slug_unchanged() {
    assert_eq!(marker_slug(&AgentKind::Codex, None), "codex");
    assert_eq!(marker_slug(&AgentKind::Codex, Some("ignored")), "codex");
    assert_eq!(marker_slug(&AgentKind::Custom, Some("glm5")), "glm5");
    assert_eq!(marker_slug(&AgentKind::Custom, None), "custom");
}

#[test]
fn resolve_agent_maps_custom_names() {
    assert_eq!(resolve_agent("codex"), (AgentKind::Codex, None));
    assert_eq!(resolve_agent("glm5"), (AgentKind::Custom, Some("glm5")));
}