asserhttp 0.7.1

Fluent http response assertions
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

use super::accessor::{BodyAccessor, HeaderAccessor, StatusAccessor};
use crate::header::key::HeaderKey;
use crate::{AsserhttpError, AsserhttpResult};

type RocketResponse<'a> = rocket::local::blocking::LocalResponse<'a>;
type AsyncRocketResponse<'a> = rocket::local::asynchronous::LocalResponse<'a>;

impl<'a> StatusAccessor for RocketResponse<'a> {
    fn get_status(&self) -> u16 {
        self.status().code
    }
}

impl<'a> StatusAccessor for AsyncRocketResponse<'a> {
    fn get_status(&self) -> u16 {
        self.status().code
    }
}

impl<'a> HeaderAccessor for RocketResponse<'a> {
    fn get_keys(&self) -> Vec<HeaderKey> {
        self.headers().iter().map(|k| k.name.as_str().into()).collect::<Vec<_>>()
    }

    fn get_raw_values(&self, key: &HeaderKey) -> Vec<String> {
        self.headers().get(key.as_ref()).map(str::to_string).collect::<Vec<_>>()
    }
}

impl<'a> HeaderAccessor for AsyncRocketResponse<'a> {
    fn get_keys(&self) -> Vec<HeaderKey> {
        self.headers().iter().map(|k| k.name.as_str().into()).collect::<Vec<_>>()
    }

    fn get_raw_values(&self, key: &HeaderKey) -> Vec<String> {
        self.headers().get(key.as_ref()).map(str::to_string).collect::<Vec<_>>()
    }
}

impl<'a> BodyAccessor for RocketResponse<'a> {
    fn get_bytes(&mut self) -> AsserhttpResult<Vec<u8>> {
        let mut buf: Vec<u8> = vec![];
        use std::io::Read as _;
        self.read_to_end(&mut buf).map_err(AsserhttpError::from)?;
        if buf.is_empty() {
            return Err(AsserhttpError::BodyAbsent);
        }
        Ok(buf)
    }
}

impl<'a> BodyAccessor for AsyncRocketResponse<'a> {
    fn get_bytes(&mut self) -> AsserhttpResult<Vec<u8>> {
        let mut buf: Vec<u8> = vec![];

        use rocket::tokio::io::AsyncReadExt as _;
        futures_lite::future::block_on(self.read_to_end(&mut buf)).map_err(AsserhttpError::from)?;
        if buf.is_empty() {
            return Err(AsserhttpError::BodyAbsent);
        }
        Ok(buf)
    }
}