tokio/tokio-io/tests/take.rs
2019-08-20 20:09:07 -07:00

17 lines
347 B
Rust

#![warn(rust_2018_idioms)]
#![feature(async_await)]
use tokio_io::AsyncReadExt;
use tokio_test::assert_ok;
#[tokio::test]
async fn take() {
let mut buf = [0; 6];
let rd: &[u8] = b"hello world";
let mut rd = rd.take(4);
let n = assert_ok!(rd.read(&mut buf).await);
assert_eq!(n, 4);
assert_eq!(&buf, &b"hell\0\0"[..]);
}