diff options
author | Alex Crichton <alex@alexcrichton.com> | 2016-07-30 17:53:12 -0700 |
---|---|---|
committer | Alex Crichton <alex@alexcrichton.com> | 2016-07-30 22:50:58 -0700 |
commit | bc64194be17bc9711b4a56364e677d823c7cc3d1 (patch) | |
tree | a0948ca757dcc938d863e847a458ded2d84da1f4 /tests/buffered.rs |
Let's rename everything!
Diffstat (limited to 'tests/buffered.rs')
-rw-r--r-- | tests/buffered.rs | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/tests/buffered.rs b/tests/buffered.rs new file mode 100644 index 00000000..3c30847e --- /dev/null +++ b/tests/buffered.rs @@ -0,0 +1,64 @@ +extern crate futures; +extern crate futures_io; +extern crate futures_mio; +extern crate env_logger; + +use std::net::TcpStream; +use std::thread; +use std::io::{Read, Write}; + +use futures::Future; +use futures::stream::Stream; +use futures_io::{BufReader, BufWriter, copy}; + +macro_rules! t { + ($e:expr) => (match $e { + Ok(e) => e, + Err(e) => panic!("{} failed with {:?}", stringify!($e), e), + }) +} + +#[test] +fn echo_server() { + const N: usize = 1024; + drop(env_logger::init()); + + let mut l = t!(futures_mio::Loop::new()); + let srv = l.handle().tcp_listen(&"127.0.0.1:0".parse().unwrap()); + let srv = t!(l.run(srv)); + let addr = t!(srv.local_addr()); + + let msg = "foo bar baz"; + let t = thread::spawn(move || { + let mut s = t!(TcpStream::connect(&addr)); + + let t2 = thread::spawn(move || { + let mut s = t!(TcpStream::connect(&addr)); + let mut b = vec![0; msg.len() * N]; + t!(s.read_exact(&mut b)); + b + }); + + let mut expected = Vec::<u8>::new(); + for _i in 0..N { + expected.extend(msg.as_bytes()); + assert_eq!(t!(s.write(msg.as_bytes())), msg.len()); + } + (expected, t2) + }); + + let clients = srv.incoming().take(2).map(|e| e.0).collect(); + let copied = clients.and_then(|clients| { + let mut clients = clients.into_iter(); + let a = BufReader::new(clients.next().unwrap()); + let b = BufWriter::new(clients.next().unwrap()); + copy(a, b) + }); + + let amt = t!(l.run(copied)); + let (expected, t2) = t.join().unwrap(); + let actual = t2.join().unwrap(); + + assert!(expected == actual); + assert_eq!(amt, msg.len() as u64 * 1024); +} |