1 #![warn(rust_2018_idioms)]
2 #![cfg(all(feature = "full", not(target_os = "wasi"), not(miri)))] // Wasi doesn't support bind
3 // No `socket` on miri.
4
5 use std::io::Result;
6 use std::io::{Read, Write};
7 use std::{net, thread};
8
9 use tokio::io::{AsyncReadExt, AsyncWriteExt};
10 use tokio::net::TcpStream;
11
12 #[tokio::test]
split() -> Result<()>13 async fn split() -> Result<()> {
14 const MSG: &[u8] = b"split";
15
16 let listener = net::TcpListener::bind("127.0.0.1:0")?;
17 let addr = listener.local_addr()?;
18
19 let handle = thread::spawn(move || {
20 let (mut stream, _) = listener.accept().unwrap();
21 stream.write_all(MSG).unwrap();
22
23 let mut read_buf = [0u8; 32];
24 let read_len = stream.read(&mut read_buf).unwrap();
25 assert_eq!(&read_buf[..read_len], MSG);
26 });
27
28 let mut stream = TcpStream::connect(&addr).await?;
29 let (mut read_half, mut write_half) = stream.split();
30
31 let mut read_buf = [0u8; 32];
32 let peek_len1 = read_half.peek(&mut read_buf[..]).await?;
33 let peek_len2 = read_half.peek(&mut read_buf[..]).await?;
34 assert_eq!(peek_len1, peek_len2);
35
36 let read_len = read_half.read(&mut read_buf[..]).await?;
37 assert_eq!(peek_len1, read_len);
38 assert_eq!(&read_buf[..read_len], MSG);
39
40 assert_eq!(write_half.write(MSG).await?, MSG.len());
41 handle.join().unwrap();
42 Ok(())
43 }
44