relay-core-cli 0.2.0

Standalone CLI and TUI for relay-core: local proxy operation, rule management, traffic inspection
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

use predicates::prelude::*;
use std::fs;
use tempfile::tempdir;

#[test]
fn test_cli_help() {
    let mut cmd = assert_cmd::cargo::cargo_bin_cmd!("relay-core-cli");
    cmd.arg("--help")
        .assert()
        .success()
        .stdout(predicate::str::contains("Start the proxy server"));
}

#[test]
fn test_ca_init() {
    let dir = tempdir().unwrap();
    let cert_path = dir.path().join("test_ca.pem");
    let key_path = dir.path().join("test_key.pem");

    let mut cmd = assert_cmd::cargo::cargo_bin_cmd!("relay-core-cli");
    cmd.arg("ca")
        .arg("init")
        .arg("--cert")
        .arg(&cert_path)
        .arg("--key")
        .arg(&key_path)
        .assert()
        .success();

    assert!(cert_path.exists());
    assert!(key_path.exists());

    // Test CA status
    let mut cmd = assert_cmd::cargo::cargo_bin_cmd!("relay-core-cli");
    cmd.arg("ca")
        .arg("status")
        .arg("--cert")
        .arg(&cert_path)
        .assert()
        .success();
}

#[test]
fn test_rules_validate_success() {
    let dir = tempdir().unwrap();
    let rule_path = dir.path().join("valid_rule.yaml");
    
    // Create a simple valid rule (Must be a list)
    let content = r#"
- id: "test-rule-1"
  active: true
  url_pattern: "example.com"
  phase: "request"
"#;
    fs::write(&rule_path, content).unwrap();

    let mut cmd = assert_cmd::cargo::cargo_bin_cmd!("relay-core-cli");
    cmd.arg("rules")
        .arg("validate")
        .arg(&rule_path)
        .assert()
        .success();
}

#[test]
fn test_rules_validate_fail() {
    let dir = tempdir().unwrap();
    let rule_path = dir.path().join("invalid_rule.yaml");
    
    // Create an invalid rule (malformed YAML)
    let content = r#"
name: "Test Rule"
filters: [
  - type: host
    value: "example.com"
"#;
    fs::write(&rule_path, content).unwrap();

    let mut cmd = assert_cmd::cargo::cargo_bin_cmd!("relay-core-cli");
    cmd.arg("rules")
        .arg("validate")
        .arg(&rule_path)
        .assert()
        .failure();
}