mocktail 0.3.2

HTTP & gRPC server mocking for 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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

//! Mock body
use std::{collections::vec_deque, convert::Infallible, pin::Pin, task::Poll};

use bytes::{Buf, Bytes};
use futures::Stream;
use http_body::Frame;

use crate::ext::MessageExt;

mod buf_list;
use buf_list::BufList;

/// The body of a mock request or response.
#[derive(Default, Debug, Clone)]
pub struct Body {
    bufs: BufList,
}

impl Body {
    /// Creates an empty body.
    pub fn empty() -> Self {
        Self::default()
    }

    /// Creates a raw bytes body.
    pub fn bytes(body: impl Into<Bytes>) -> Self {
        let bytes: Bytes = body.into();
        Self { bufs: bytes.into() }
    }

    /// Creates a raw bytes streaming body.
    pub fn bytes_stream(messages: impl IntoIterator<Item = impl Into<Bytes>>) -> Self {
        let bufs = messages.into_iter().map(|msg| msg.into()).collect();
        Self { bufs }
    }

    /// Creates a JSON body.
    pub fn json(body: impl serde::Serialize) -> Self {
        let bytes = serde_json::to_vec(&body).unwrap();
        Self { bufs: bytes.into() }
    }

    /// Creates a newline delimited JSON streaming body.
    pub fn json_lines_stream(messages: impl IntoIterator<Item = impl serde::Serialize>) -> Self {
        let bufs = messages
            .into_iter()
            .map(|msg| {
                let mut bytes = serde_json::to_vec(&msg).unwrap();
                bytes.push(b'\n');
                bytes.into()
            })
            .collect();
        Self { bufs }
    }

    /// Creates a protobuf body.
    pub fn pb(body: impl prost::Message) -> Self {
        let bytes = body.to_bytes();
        Self { bufs: bytes.into() }
    }

    /// Creates a protobuf streaming body.
    pub fn pb_stream(messages: impl IntoIterator<Item = impl prost::Message>) -> Self {
        let bufs = messages.into_iter().map(|msg| msg.to_bytes()).collect();
        Self { bufs }
    }

    /// Returns true if empty.
    pub fn is_empty(&self) -> bool {
        self.len() == 0
    }

    /// Returns the byte length of the body.
    pub fn len(&self) -> usize {
        self.bufs.remaining()
    }

    pub fn as_bytes(&mut self) -> Bytes {
        self.bufs.as_bytes()
    }

    pub fn iter(&self) -> vec_deque::Iter<'_, Bytes> {
        self.bufs.iter()
    }
}

impl PartialEq for Body {
    fn eq(&self, other: &Self) -> bool {
        // We want to compare the merged bytes from all bufs
        // as the request body will be buffered chunks.
        // TODO: figure out a better approach with less overhead?
        self.bufs.clone().as_bytes() == other.bufs.clone().as_bytes()
    }
}

impl Eq for Body {}

impl PartialOrd for Body {
    fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
        Some(self.cmp(other))
    }
}

impl Ord for Body {
    fn cmp(&self, other: &Self) -> std::cmp::Ordering {
        self.bufs.cmp(&other.bufs)
    }
}

impl Stream for Body {
    type Item = Bytes;

    fn poll_next(
        mut self: Pin<&mut Self>,
        _cx: &mut std::task::Context<'_>,
    ) -> Poll<Option<Self::Item>> {
        if let Some(data) = self.bufs.pop() {
            Poll::Ready(Some(data))
        } else {
            Poll::Ready(None)
        }
    }
}

impl http_body::Body for Body {
    type Data = Bytes;
    type Error = Infallible;

    fn poll_frame(
        mut self: Pin<&mut Self>,
        _cx: &mut std::task::Context<'_>,
    ) -> Poll<Option<Result<Frame<Self::Data>, Self::Error>>> {
        if let Some(data) = self.bufs.pop() {
            let frame = Frame::data(data);
            Poll::Ready(Some(Ok(frame)))
        } else {
            Poll::Ready(None)
        }
    }
}

impl From<Bytes> for Body {
    fn from(value: Bytes) -> Self {
        Self::bytes(value)
    }
}