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
|
use async_codec_lite::{Bytes, Framed, LengthCodec};
use futures_lite::future::block_on;
use futures_util::{io::Cursor, sink::SinkExt, stream::StreamExt};
#[test]
fn same_msgs_are_received_as_were_sent() {
let cur = Cursor::new(vec![0; 256]);
let mut framed = Framed::new(cur, LengthCodec::<u64>::new());
let send_msgs = async {
framed.send(Bytes::from("msg1")).await.unwrap();
framed.send(Bytes::from("msg2")).await.unwrap();
framed.send(Bytes::from("msg3")).await.unwrap();
};
block_on(send_msgs);
let mut parts = framed.into_parts();
parts.io.set_position(0);
let framed = Framed::new(parts.io, LengthCodec::<u64>::new());
let recv_msgs = framed
.take(3)
.map(|res| res.unwrap())
.map(|buf| String::from_utf8(buf.to_vec()).unwrap())
.collect::<Vec<_>>();
let msgs: Vec<String> = block_on(recv_msgs);
assert!(msgs == vec!["msg1", "msg2", "msg3"]);
}
|
