crux_http/testing/
response_builder.rs

1use http_types::headers::{HeaderName, ToHeaderValues};
2
3use crate::response::Response;
4
5/// Allows users to build an http response.
6///
7/// This is mostly expected to be useful in tests rather than application code.
8pub struct ResponseBuilder<Body> {
9    response: Response<Body>,
10}
11
12impl ResponseBuilder<Vec<u8>> {
13    /// Constructs a new ResponseBuilder with the 200 OK status code.
14    pub fn ok() -> ResponseBuilder<Vec<u8>> {
15        ResponseBuilder::with_status(http_types::StatusCode::Ok)
16    }
17
18    /// Constructs a new ResponseBuilder with the specified status code.
19    pub fn with_status(status: http_types::StatusCode) -> ResponseBuilder<Vec<u8>> {
20        let response = Response::new_with_status(status);
21
22        ResponseBuilder { response }
23    }
24}
25
26impl<Body> ResponseBuilder<Body> {
27    /// Sets the body of the Response
28    pub fn body<NewBody>(self, body: NewBody) -> ResponseBuilder<NewBody> {
29        let response = self.response.with_body(body);
30        ResponseBuilder { response }
31    }
32
33    /// Sets a header on the response.
34    pub fn header(mut self, key: impl Into<HeaderName>, value: impl ToHeaderValues) -> Self {
35        self.response.insert_header(key, value);
36        self
37    }
38
39    /// Builds the response
40    pub fn build(self) -> Response<Body> {
41        self.response
42    }
43}
crux_http/testing/response_builder.rs

crux_http/testing/
response_builder.rs
