35 lines
937 B
Rust
35 lines
937 B
Rust
#![warn(rust_2018_idioms)]
|
|
|
|
use bytes::Bytes;
|
|
use tokio::io::AsyncReadExt;
|
|
use tokio_stream::iter;
|
|
use tokio_util::io::StreamReader;
|
|
|
|
#[tokio::test]
|
|
async fn test_stream_reader() -> std::io::Result<()> {
|
|
let stream = iter(vec![
|
|
std::io::Result::Ok(Bytes::from_static(&[])),
|
|
Ok(Bytes::from_static(&[0, 1, 2, 3])),
|
|
Ok(Bytes::from_static(&[])),
|
|
Ok(Bytes::from_static(&[4, 5, 6, 7])),
|
|
Ok(Bytes::from_static(&[])),
|
|
Ok(Bytes::from_static(&[8, 9, 10, 11])),
|
|
Ok(Bytes::from_static(&[])),
|
|
]);
|
|
|
|
let mut read = StreamReader::new(stream);
|
|
|
|
let mut buf = [0; 5];
|
|
read.read_exact(&mut buf).await?;
|
|
assert_eq!(buf, [0, 1, 2, 3, 4]);
|
|
|
|
assert_eq!(read.read(&mut buf).await?, 3);
|
|
assert_eq!(&buf[..3], [5, 6, 7]);
|
|
|
|
assert_eq!(read.read(&mut buf).await?, 4);
|
|
assert_eq!(&buf[..4], [8, 9, 10, 11]);
|
|
|
|
assert_eq!(read.read(&mut buf).await?, 0);
|
|
|
|
Ok(())
|
|
}
|