iron_test/
response.rs

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
use iron::prelude::*;

/// Extracts a utf8 response body to a String.
pub fn extract_body_to_string(response: Response) -> String {
    let result = extract_body_to_bytes(response);
    String::from_utf8(result).unwrap()
}

/// Extracts a response body to a Vector of bytes.
pub fn extract_body_to_bytes(response: Response) -> Vec<u8> {
    let mut result = Vec::new();

    if let Some(mut body) = response.body {
        body.write_body(&mut result).ok();
    }

    result
}

#[cfg(test)]
mod test {
    use iron::headers::Headers;
    use iron::prelude::*;
    use iron::{Handler, status};

    use request;

    use super::*;

    struct HelloWorldHandler;

    impl Handler for HelloWorldHandler {
        fn handle(&self, _: &mut Request) -> IronResult<Response> {
            Ok(Response::with((status::Ok, "Hello, world!")))
        }
    }

    #[test]
    fn test_extract_body_to_string() {
        let response = request::get("http://localhost:3000",
                           Headers::new(),
                           &HelloWorldHandler);
        let result = extract_body_to_string(response.unwrap());

        assert_eq!(result, "Hello, world!");
    }

    #[test]
    fn test_extract_body_to_bytes() {
        let response = request::get("http://localhost:3000",
                           Headers::new(),
                           &HelloWorldHandler);
        let result = extract_body_to_bytes(response.unwrap());

        assert_eq!(result, b"Hello, world!");
    }
}