agent-code 0.21.0

An AI-powered coding agent for the terminal, written in pure Rust
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

//! End-to-end test for output-style propagation across the
//! Agent-tool subprocess boundary.
//!
//! Regression for the codex finding that `applies_to: [subagent]`
//! styles never reached the spawned child: the parent set
//! `AGENT_CODE_SUBAGENT=1` but did not propagate the active
//! disk-loaded style. The fix introduces
//! `AGENT_CODE_DISK_OUTPUT_STYLE=<name>`; this test boots the binary
//! with that env var and `--dump-system-prompt` and asserts the
//! prompt contains the disk style's body.

use assert_cmd::Command;
use std::fs;

const STYLE_BODY: &str =
    "Be a marker phrase that should not occur in the default prompt: subagent_propagation_works.";

fn agent() -> Command {
    Command::cargo_bin("agent").expect("binary should exist")
}

#[test]
fn subagent_inherits_disk_output_style_via_env_var() {
    let project = tempfile::tempdir().expect("tempdir");
    let styles_dir = project.path().join(".agent").join("output-styles");
    fs::create_dir_all(&styles_dir).unwrap();
    let style_path = styles_dir.join("subagent-only.md");
    fs::write(
        &style_path,
        format!(
            "---\n\
             name: subagent-only\n\
             description: subagent-only style fixture\n\
             applies_to:\n  - subagent\n\
             ---\n\
             {STYLE_BODY}",
        ),
    )
    .unwrap();

    // Pin the user-level dir to an empty tempdir so this test cannot
    // pick up whatever may live in `~/.config/agent-code/output-styles/`
    // on the developer's machine.
    let user_dir = tempfile::tempdir().expect("user tempdir");

    let output = agent()
        .arg("--dump-system-prompt")
        .arg("--cwd")
        .arg(project.path())
        .env("AGENT_CODE_SUBAGENT", "1")
        .env("AGENT_CODE_DISK_OUTPUT_STYLE", "subagent-only")
        .env("AGENT_CODE_USER_OUTPUT_STYLES_DIR", user_dir.path())
        // No API key needed for --dump-system-prompt, but scrub anyway
        // so a developer's environment can't influence the prompt.
        .env_remove("AGENT_CODE_API_KEY")
        .env_remove("ANTHROPIC_API_KEY")
        .output()
        .expect("failed to invoke binary");

    let stdout = String::from_utf8_lossy(&output.stdout);
    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(
        output.status.success(),
        "binary should exit cleanly under --dump-system-prompt; stderr: {stderr}"
    );
    assert!(
        stdout.contains(STYLE_BODY),
        "subagent-applied disk style body must appear in the dumped system prompt; \
         stdout was:\n{stdout}",
    );
}

#[test]
fn missing_disk_output_style_logs_warning_and_continues() {
    // The CLI must NOT crash when AGENT_CODE_DISK_OUTPUT_STYLE points
    // at a style that doesn't exist (e.g. parent had a project file
    // but the subagent runs in a worktree without it).
    let project = tempfile::tempdir().expect("tempdir");
    let user_dir = tempfile::tempdir().expect("user tempdir");

    let output = agent()
        .arg("--dump-system-prompt")
        .arg("--cwd")
        .arg(project.path())
        .env("AGENT_CODE_SUBAGENT", "1")
        .env("AGENT_CODE_DISK_OUTPUT_STYLE", "no-such-style")
        .env("AGENT_CODE_USER_OUTPUT_STYLES_DIR", user_dir.path())
        .env_remove("AGENT_CODE_API_KEY")
        .env_remove("ANTHROPIC_API_KEY")
        .output()
        .expect("failed to invoke binary");

    let stderr = String::from_utf8_lossy(&output.stderr);
    assert!(
        output.status.success(),
        "binary must not crash on a missing style id; stderr: {stderr}"
    );
}