lmn-core 0.3.0

Core engine for the lmn HTTP load testing CLI
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

use std::collections::HashMap;

// ── StatusCodeHistogram ───────────────────────────────────────────────────────

/// Tracks the frequency of HTTP status codes and connection errors across requests.
pub struct StatusCodeHistogram {
    counts: HashMap<u16, u64>,
    error_count: u64,
}

impl StatusCodeHistogram {
    /// Creates a new empty histogram.
    pub fn new() -> Self {
        Self {
            counts: HashMap::new(),
            error_count: 0,
        }
    }

    /// Records a single request outcome.
    ///
    /// `None` status code represents a connection error (no HTTP response received).
    pub fn record(&mut self, status_code: Option<u16>) {
        match status_code {
            Some(code) => *self.counts.entry(code).or_insert(0) += 1,
            None => self.error_count += 1,
        }
    }

    /// Returns the map of HTTP status code to count.
    pub fn counts(&self) -> &HashMap<u16, u64> {
        &self.counts
    }

    /// Returns the number of connection errors (requests with no HTTP response).
    pub fn error_count(&self) -> u64 {
        self.error_count
    }

    /// Returns the total number of recorded requests (status codes + errors).
    pub fn total(&self) -> u64 {
        self.counts.values().sum::<u64>() + self.error_count
    }
}

impl Default for StatusCodeHistogram {
    fn default() -> Self {
        Self::new()
    }
}

// ── Tests ─────────────────────────────────────────────────────────────────────

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn record_200_increments_count() {
        let mut h = StatusCodeHistogram::new();
        h.record(Some(200));
        assert_eq!(h.counts()[&200], 1);
    }

    #[test]
    fn record_none_increments_error_count() {
        let mut h = StatusCodeHistogram::new();
        h.record(None);
        h.record(None);
        assert_eq!(h.error_count(), 2);
    }

    #[test]
    fn total_sums_all_codes_and_errors() {
        let mut h = StatusCodeHistogram::new();
        h.record(Some(200));
        h.record(Some(200));
        h.record(Some(404));
        h.record(None);
        assert_eq!(h.total(), 4);
    }

    #[test]
    fn multiple_codes_tracked_independently() {
        let mut h = StatusCodeHistogram::new();
        h.record(Some(200));
        h.record(Some(201));
        h.record(Some(404));
        h.record(Some(503));
        assert_eq!(h.counts()[&200], 1);
        assert_eq!(h.counts()[&201], 1);
        assert_eq!(h.counts()[&404], 1);
        assert_eq!(h.counts()[&503], 1);
    }
}