astrid 0.10.1

Command-line interface for Astrid secure agent runtime
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
99
100
101

use clap::{CommandFactory, Parser};

use super::{Cli, Commands};

#[test]
fn grant_capsules_allows_an_operator_enforced_distro() {
    let cli = Cli::try_parse_from(["astrid", "init", "--grant-capsules"])
        .expect("the operator may supply the distro outside the CLI argument surface");

    assert!(matches!(
        cli.command,
        Some(Commands::Init {
            distro: None,
            grant_capsules: true,
            ..
        })
    ));
}

#[test]
fn distro_help_lists_only_supported_sources_and_the_launcher_exception() {
    let mut command = Cli::command();
    let init = command.find_subcommand_mut("init").expect("init command");
    let init_distro_help = init
        .get_arguments()
        .find(|argument| argument.get_id() == "distro")
        .and_then(|argument| argument.get_help())
        .map(ToString::to_string);

    let mut command = Cli::command();
    let apply = command
        .find_subcommand_mut("distro")
        .expect("distro command")
        .find_subcommand_mut("apply")
        .expect("distro apply command");
    let apply_name_help = apply
        .get_arguments()
        .find(|argument| argument.get_id() == "name")
        .and_then(|argument| argument.get_help())
        .map(ToString::to_string);

    assert_eq!(
        init_distro_help.as_deref(),
        Some(
            "Distro source to install. Required unless an embedding launcher sets \
             `ASTRID_ENFORCED_DISTRO` (`@owner/repo`, URL, local Distro.toml, or .shuttle)"
        )
    );
    assert_eq!(
        apply_name_help.as_deref(),
        Some("Distro source (`@owner/repo`, URL, local Distro.toml, or .shuttle)")
    );
}

#[test]
fn init_parses_target_separately_from_operator() {
    let cli = Cli::try_parse_from([
        "astrid",
        "--principal",
        "operator-1",
        "init",
        "--distro",
        "./Distro.toml",
        "--target-principal",
        "agent-1",
        "--grant-capsules",
    ])
    .expect("operator and target principal should parse independently");

    assert_eq!(cli.principal.as_deref(), Some("operator-1"));
    assert!(matches!(
        cli.command,
        Some(Commands::Init {
            target_principal: Some(ref target),
            grant_capsules: true,
            ..
        }) if target == "agent-1"
    ));
}

#[test]
fn global_options_before_init_preserve_the_requested_distro() {
    let cli = Cli::try_parse_from([
        "astrid",
        "--principal",
        "operator-1",
        "init",
        "--distro",
        "@example/other",
    ])
    .expect("global options before init should parse");

    assert_eq!(cli.principal.as_deref(), Some("operator-1"));
    assert!(matches!(
        cli.command,
        Some(Commands::Init {
            distro: Some(ref distro),
            ..
        }) if distro == "@example/other"
    ));
}