1 #![warn(rust_2018_idioms)] 2 #![cfg(feature = "full")] 3 4 use tokio::net::TcpListener; 5 use tokio::prelude::*; 6 use tokio_test::assert_ok; 7 8 use std::io::prelude::*; 9 use std::net::TcpStream; 10 use std::thread; 11 12 #[tokio::test] echo_server()13async fn echo_server() { 14 const N: usize = 1024; 15 16 let mut srv = assert_ok!(TcpListener::bind("127.0.0.1:0").await); 17 let addr = assert_ok!(srv.local_addr()); 18 19 let msg = "foo bar baz"; 20 21 let t = thread::spawn(move || { 22 let mut s = assert_ok!(TcpStream::connect(&addr)); 23 24 let t2 = thread::spawn(move || { 25 let mut s = assert_ok!(TcpStream::connect(&addr)); 26 let mut b = vec![0; msg.len() * N]; 27 assert_ok!(s.read_exact(&mut b)); 28 b 29 }); 30 31 let mut expected = Vec::<u8>::new(); 32 for _i in 0..N { 33 expected.extend(msg.as_bytes()); 34 let res = assert_ok!(s.write(msg.as_bytes())); 35 assert_eq!(res, msg.len()); 36 } 37 38 (expected, t2) 39 }); 40 41 let (mut a, _) = assert_ok!(srv.accept().await); 42 let (mut b, _) = assert_ok!(srv.accept().await); 43 44 let n = assert_ok!(io::copy(&mut a, &mut b).await); 45 46 let (expected, t2) = t.join().unwrap(); 47 let actual = t2.join().unwrap(); 48 49 assert!(expected == actual); 50 assert_eq!(n, msg.len() as u64 * 1024); 51 } 52