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

use http::headers::{HeaderName, ToHeaderValues};

use crate::http;

use crate::response::Response;

/// Allows users to build an http response.
///
/// This is mostly expected to be useful in tests rather than application code.
pub struct ResponseBuilder<Body> {
    response: Response<Body>,
}

impl ResponseBuilder<Vec<u8>> {
    /// Constructs a new ResponseBuilder with the 200 OK status code.
    pub fn ok() -> ResponseBuilder<Vec<u8>> {
        ResponseBuilder::with_status(http::StatusCode::Ok)
    }

    /// Constructs a new ResponseBuilder with the specified status code.
    pub fn with_status(status: http::StatusCode) -> ResponseBuilder<Vec<u8>> {
        let response = Response::new_with_status(status);

        ResponseBuilder { response }
    }
}

impl<Body> ResponseBuilder<Body> {
    /// Sets the body of the Response
    pub fn body<NewBody>(self, body: NewBody) -> ResponseBuilder<NewBody> {
        let response = self.response.with_body(body);
        ResponseBuilder { response }
    }

    /// Sets a header on the response.
    pub fn header(mut self, key: impl Into<HeaderName>, value: impl ToHeaderValues) -> Self {
        self.response.insert_header(key, value);
        self
    }

    /// Builds the response
    pub fn build(self) -> Response<Body> {
        self.response
    }
}