rhei-cli 0.2.0

Command-line driver for the Rhei agent runtime.
Documentation 
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

//! §FS-rhei-run-json.3: the durable log and its follower.

use super::*;
use crate::rhei_tui::event::{AgentStream, MessageLevel};

fn message(text: &str) -> RunEvent {
    RunEvent::Message { level: MessageLevel::Info, text: text.to_string() }
}

#[test]
fn records_are_sequenced_from_one_and_flushed_per_line() {
    let tmp = tempfile::tempdir().unwrap();
    let sink = EventLogSink::create(tmp.path()).unwrap();
    sink.emit(message("first"));
    // Read *while the sink is still open*: a follower must not have to wait
    // for the run to end.
    let contents = fs::read_to_string(sink.path()).unwrap();
    assert_eq!(contents.lines().count(), 1);
    assert!(contents.contains(r#""seq":1"#));

    sink.emit(message("second"));
    let contents = fs::read_to_string(sink.path()).unwrap();
    let seqs: Vec<Option<u64>> =
        contents.lines().filter_map(event_json::decode).map(|r| r.seq).collect();
    assert_eq!(seqs, vec![Some(1), Some(2)]);
    assert_eq!(sink.last_seq(), 2);
}

#[test]
fn agent_output_stays_out_of_the_durable_log() {
    let tmp = tempfile::tempdir().unwrap();
    let sink = EventLogSink::create(tmp.path()).unwrap();
    sink.emit(RunEvent::AgentOutput {
        slot: 0,
        task: "auth.1".to_string(),
        stream: AgentStream::Stdout,
        line: "noise".to_string(),
        wall_clock: SystemTime::now(),
    });
    sink.emit(message("kept"));
    let contents = fs::read_to_string(sink.path()).unwrap();
    assert_eq!(contents.lines().count(), 1, "only the structural event belongs here");
    // The one record present is still seq 1: a skipped event must not burn a
    // sequence number, or a consumer sees a gap it cannot explain.
    assert!(contents.contains(r#""seq":1"#));
}

#[test]
fn a_new_run_truncates_the_previous_runs_log() {
    let tmp = tempfile::tempdir().unwrap();
    {
        let sink = EventLogSink::create(tmp.path()).unwrap();
        sink.emit(message("old run"));
    }
    let sink = EventLogSink::create(tmp.path()).unwrap();
    sink.emit(message("new run"));
    let contents = fs::read_to_string(sink.path()).unwrap();
    assert_eq!(contents.lines().count(), 1);
    assert!(contents.contains("new run"));
}

#[test]
fn the_reader_follows_appends_and_returns_only_new_records() {
    let tmp = tempfile::tempdir().unwrap();
    let sink = EventLogSink::create(tmp.path()).unwrap();
    let mut reader = EventLogReader::open(sink.path());

    sink.emit(message("one"));
    let first = reader.poll();
    assert_eq!(first.len(), 1);
    assert_eq!(first[0].seq, Some(1), "records carry the sequence they were written under");
    assert_eq!(reader.last_seq(), 1);

    // Nothing new: a poll on a quiet run must return nothing, not a replay.
    assert!(reader.poll().is_empty());

    sink.emit(message("two"));
    sink.emit(message("three"));
    let rest = reader.poll();
    assert_eq!(rest.len(), 2);
    assert_eq!(reader.last_seq(), 3);
}

#[test]
fn a_torn_final_line_is_re_read_whole_on_the_next_poll() {
    let tmp = tempfile::tempdir().unwrap();
    let path = event_log_path(tmp.path());
    fs::create_dir_all(path.parent().unwrap()).unwrap();
    let whole =
        format!("{}\n", event_json::encode(Some(1), &message("complete"), SystemTime::now(), None));
    let fragment =
        event_json::encode(Some(2), &message("half written"), SystemTime::now(), None).to_string();
    let torn = &fragment[..fragment.len() / 2];
    fs::write(&path, format!("{whole}{torn}")).unwrap();

    let mut reader = EventLogReader::open(&path);
    assert_eq!(reader.poll().len(), 1, "the torn line must not be decoded");
    assert_eq!(reader.last_seq(), 1);

    // The run finishes writing it.
    fs::write(&path, format!("{whole}{fragment}\n")).unwrap();
    let events = reader.poll();
    assert_eq!(events.len(), 1, "the completed line arrives whole");
    assert_eq!(reader.last_seq(), 2);
}

#[test]
fn a_missing_log_polls_empty_rather_than_failing() {
    let tmp = tempfile::tempdir().unwrap();
    let mut reader = EventLogReader::open(tmp.path().join("runtime/events.jsonl"));
    assert!(reader.poll().is_empty());
}

#[test]
fn a_truncating_new_run_resets_a_live_reader() {
    let tmp = tempfile::tempdir().unwrap();
    let sink = EventLogSink::create(tmp.path()).unwrap();
    let mut reader = EventLogReader::open(sink.path());
    sink.emit(message("one"));
    sink.emit(message("two"));
    assert_eq!(reader.poll().len(), 2);

    drop(sink);
    let sink = EventLogSink::create(tmp.path()).unwrap();
    sink.emit(message("fresh"));
    let events = reader.poll();
    assert_eq!(events.len(), 1, "the reader must restart, not read past the new end");
    assert_eq!(reader.last_seq(), 1);
}