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
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

use crate::wiremock::ResponseTemplate;
use crate::StubrResult;
use serde_json::{Map, Value};

use super::{
    template::{data::HandlebarsData, HandlebarTemplatable},
    ResponseAppender,
};

#[derive(serde::Serialize, serde::Deserialize, Debug, Default, Clone, Eq, PartialEq)]
pub struct HttpRespHeadersStub {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub headers: Option<Map<String, Value>>,
}

impl HttpRespHeadersStub {
    fn _render_response_template(&self, mut resp: ResponseTemplate, data: &HandlebarsData) -> StubrResult<ResponseTemplate> {
        if let Some(headers) = self.headers.as_ref() {
            for (k, v) in headers {
                if let Some(v) = v.as_str() {
                    let rendered = self.render(v, data).unwrap_or_default();
                    resp = resp.insert_header(k.as_str(), rendered.as_str())
                }
            }
        }
        Ok(resp)
    }
}

impl ResponseAppender for HttpRespHeadersStub {
    fn add(&self, mut resp: ResponseTemplate) -> ResponseTemplate {
        if let Some(headers) = self.headers.as_ref() {
            for (k, v) in headers {
                if let Some(v) = v.as_str() {
                    resp = resp.insert_header(k.as_str(), v)
                }
            }
        }
        resp
    }
}

impl HandlebarTemplatable for HttpRespHeadersStub {
    fn register_template(&self) {
        if let Some(headers) = self.headers.as_ref() {
            for (_, v) in headers {
                if let Some(v) = v.as_str() {
                    self.register(v, v)
                }
            }
        }
    }

    #[cfg(not(feature = "grpc"))]
    fn render_response_template(&self, resp: ResponseTemplate, data: &HandlebarsData) -> StubrResult<ResponseTemplate> {
        self._render_response_template(resp, data)
    }

    #[cfg(feature = "grpc")]
    fn render_response_template(
        &self, resp: ResponseTemplate, data: &HandlebarsData, _md: Option<&protobuf::reflect::MessageDescriptor>,
    ) -> StubrResult<ResponseTemplate> {
        self._render_response_template(resp, data)
    }
}