diff options
Diffstat (limited to 'tokio-net/tests/tcp_peek.rs')
-rw-r--r-- | tokio-net/tests/tcp_peek.rs | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/tokio-net/tests/tcp_peek.rs b/tokio-net/tests/tcp_peek.rs new file mode 100644 index 00000000..60f0c76d --- /dev/null +++ b/tokio-net/tests/tcp_peek.rs @@ -0,0 +1,27 @@ +#![warn(rust_2018_idioms)] +#![feature(async_await)] + +use std::thread; +use std::{convert::TryInto, io::Write, net}; +use tokio::io::AsyncReadExt; +use tokio::net::TcpStream; +use tokio_test::assert_ok; + +#[tokio::test] +async fn peek() { + let listener = net::TcpListener::bind("127.0.0.1:0").unwrap(); + let addr = listener.local_addr().unwrap(); + let t = thread::spawn(move || assert_ok!(listener.accept()).0); + + let left = net::TcpStream::connect(&addr).unwrap(); + let mut right = t.join().unwrap(); + right.write(&[1, 2, 3, 4]).unwrap(); + + let mut left: TcpStream = left.try_into().unwrap(); + let mut buf = [0u8; 16]; + let n = assert_ok!(left.peek(&mut buf).await); + assert_eq!([1, 2, 3, 4], buf[..n]); + + let n = assert_ok!(left.read(&mut buf).await); + assert_eq!([1, 2, 3, 4], buf[..n]); +} |