cortex-code/src/cortex-cli/tests/verbose_output.rs at c77cba32f3b1aa7886d78df2197a0d57ff5dd14d · CortexLM/cortex-code · GitHub

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
102
use std::fs;
use std::process::Command;

use tempfile::tempdir;

fn combined_output(output: &std::process::Output) -> String {
    format!(
        "{}{}",
        String::from_utf8_lossy(&output.stdout),
        String::from_utf8_lossy(&output.stderr)
    )
}

#[test]
fn verbose_agent_list_adds_diagnostics() {
    let home_dir = tempdir().unwrap();

    let normal = Command::new(env!("CARGO_BIN_EXE_Cortex"))
        .args(["agent", "list"])
        .env("HOME", home_dir.path())
        .env_remove("CORTEX_HOME")
        .output()
        .unwrap();
    assert!(
        normal.status.success(),
        "normal agent list failed:\n{}",
        combined_output(&normal)
    );

    let verbose = Command::new(env!("CARGO_BIN_EXE_Cortex"))
        .args(["--verbose", "agent", "list"])
        .env("HOME", home_dir.path())
        .env_remove("CORTEX_HOME")
        .output()
        .unwrap();
    assert!(
        verbose.status.success(),
        "verbose agent list failed:\n{}",
        combined_output(&verbose)
    );

    let normal_output = combined_output(&normal);
    let verbose_output = combined_output(&verbose);
    assert_ne!(normal_output, verbose_output);
    assert!(verbose_output.contains("listing agents"));
}

#[test]
fn verbose_github_status_adds_diagnostics() {
    let repo = tempdir().unwrap();
    fs::create_dir(repo.path().join(".git")).unwrap();
    let workflows_dir = repo.path().join(".github").join("workflows");
    fs::create_dir_all(&workflows_dir).unwrap();
    fs::write(
        workflows_dir.join("Cortex.yml"),
        r#"# Cortex CI/CD Automation
# Generated by: cortex github install

name: Cortex

on:
  workflow_dispatch:

jobs:
  cortex:
    runs-on: ubuntu-latest
    steps: []
"#,
    )
    .unwrap();

    let normal = Command::new(env!("CARGO_BIN_EXE_Cortex"))
        .args(["github", "status", "--path", repo.path().to_str().unwrap()])
        .output()
        .unwrap();
    assert!(
        normal.status.success(),
        "normal github status failed:\n{}",
        combined_output(&normal)
    );

    let verbose = Command::new(env!("CARGO_BIN_EXE_Cortex"))
        .args([
            "--verbose",
            "github",
            "status",
            "--path",
            repo.path().to_str().unwrap(),
        ])
        .output()
        .unwrap();
    assert!(
        verbose.status.success(),
        "verbose github status failed:\n{}",
        combined_output(&verbose)
    );

    let normal_output = combined_output(&normal);
    let verbose_output = combined_output(&verbose);
    assert_ne!(normal_output, verbose_output);
    assert!(verbose_output.contains("checking GitHub Actions installation status"));
}