stubr 0.6.2

Wiremock implemented in Rust
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

use std::time::Duration;

use crate::wiremock::ResponseTemplate;

use crate::Config;

use super::ResponseAppender;

/// See [https://wiremock.org/docs/simulating-faults/#per-stub-fixed-delays]
pub struct Delay<'a>(pub Option<u64>, pub &'a Option<RandomDelay>, pub &'a Config);

impl ResponseAppender for Delay<'_> {
    fn add(&self, mut resp: ResponseTemplate) -> ResponseTemplate {
        if let Some(global_delay) = self.2.global_delay {
            resp = resp.set_delay(Duration::from_millis(global_delay))
        } else if let Some(latency) = self.2.latency {
            if let Some(delay) = self.0 {
                resp = resp.set_delay(Duration::from_millis(latency + delay))
            } else {
                resp = resp.set_delay(Duration::from_millis(latency))
            }
        } else if let Some(delay) = self.0 {
            resp = resp.set_delay(Duration::from_millis(delay))
        } else if let Some(RandomDelay::Lognormal { median, sigma }) = self.1 {
            resp = resp.set_lognormal_delay(*median, *sigma)
        }
        resp
    }
}

/// See [https://wiremock.org/docs/simulating-faults/#per-stub-random-delays]
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
#[serde(tag = "type")]
pub enum RandomDelay {
    #[serde(rename = "lognormal")]
    Lognormal { median: u64, sigma: f64 },
}