aa-cli 0.0.1-alpha.9

aasm — command-line tool for Agent Assembly
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

//! Integration tests for `aasm policy apply` via REST API.

use std::io::Write;
use std::path::PathBuf;
use std::process::ExitCode;

use wiremock::matchers::{body_partial_json, method, path};
use wiremock::{Mock, MockServer, ResponseTemplate};

fn make_context(api_url: &str) -> aa_cli::config::ResolvedContext {
    aa_cli::config::ResolvedContext {
        name: None,
        api_url: api_url.to_string(),
        api_key: None,
    }
}

#[tokio::test]
async fn apply_sends_post_to_api_and_prints_response() {
    let server = MockServer::start().await;

    Mock::given(method("POST"))
        .and(path("/api/v1/policies"))
        .and(body_partial_json(serde_json::json!({
            "policy_yaml": "apiVersion: agent-assembly.dev/v1alpha1\nkind: GovernancePolicy\nmetadata:\n  name: test-policy\n  version: \"1.0.0\"\nspec:\n  rules: []\n"
        })))
        .respond_with(ResponseTemplate::new(201).set_body_json(serde_json::json!({
            "name": "abc123def456",
            "version": "2026-04-30T12:00:00Z",
            "active": true,
            "rule_count": 0
        })))
        .expect(1)
        .mount(&server)
        .await;

    let mut tmp = tempfile::NamedTempFile::new().unwrap();
    tmp.write_all(b"apiVersion: agent-assembly.dev/v1alpha1\nkind: GovernancePolicy\nmetadata:\n  name: test-policy\n  version: \"1.0.0\"\nspec:\n  rules: []\n").unwrap();

    let file_path = tmp.path().to_path_buf();
    let uri = server.uri();

    // run_apply creates its own tokio runtime via block_on, so run it
    // on a separate thread to avoid nesting inside the #[tokio::test] runtime.
    let result = std::thread::spawn(move || {
        let args = aa_cli::commands::policy::history::ApplyArgs {
            file: file_path,
            applied_by: None,
        };
        let ctx = make_context(&uri);
        aa_cli::commands::policy::history::run_apply(args, &ctx)
    })
    .join()
    .unwrap();

    assert_eq!(result, ExitCode::SUCCESS);
}

#[tokio::test]
async fn apply_fails_fast_on_invalid_yaml_without_api_call() {
    let server = MockServer::start().await;

    // No mock mounted — any API call would cause an unmatched request error.

    let mut tmp = tempfile::NamedTempFile::new().unwrap();
    tmp.write_all(b"this is not valid yaml: [").unwrap();

    let file_path = tmp.path().to_path_buf();
    let uri = server.uri();

    let result = std::thread::spawn(move || {
        let args = aa_cli::commands::policy::history::ApplyArgs {
            file: file_path,
            applied_by: None,
        };
        let ctx = make_context(&uri);
        aa_cli::commands::policy::history::run_apply(args, &ctx)
    })
    .join()
    .unwrap();

    assert_eq!(result, ExitCode::FAILURE);
}

#[test]
fn apply_fails_on_missing_file() {
    let args = aa_cli::commands::policy::history::ApplyArgs {
        file: PathBuf::from("/tmp/nonexistent-policy-file-aaasm179.yaml"),
        applied_by: None,
    };
    let ctx = make_context("http://localhost:9999");

    let result = aa_cli::commands::policy::history::run_apply(args, &ctx);
    assert_eq!(result, ExitCode::FAILURE);
}