1 files changed, 43 insertions, 0 deletions
diff --git a/third_party/rust/tokio-tcp/tests/limit.rs b/third_party/rust/tokio-tcp/tests/limit.rs
new file mode 100644
index 0000000000..8714da9a51
--- /dev/null
+++ b/third_party/rust/tokio-tcp/tests/limit.rs
@@ -0,0 +1,43 @@
+extern crate futures;
+extern crate tokio_tcp;
+extern crate tokio_io;
+
+use std::net::TcpStream;
+use std::thread;
+use std::io::{Write, Read};
+
+use futures::Future;
+use futures::stream::Stream;
+use tokio_io::io::read_to_end;
+use tokio_tcp::TcpListener;
+
+macro_rules! t {
+    ($e:expr) => (match $e {
+        Ok(e) => e,
+        Err(e) => panic!("{} failed with {:?}", stringify!($e), e),
+    })
+}
+
+#[test]
+fn limit() {
+    let srv = t!(TcpListener::bind(&t!("127.0.0.1:0".parse())));
+    let addr = t!(srv.local_addr());
+
+    let t = thread::spawn(move || {
+        let mut s1 = TcpStream::connect(&addr).unwrap();
+        s1.write_all(b"foo bar baz").unwrap();
+    });
+
+    let clients = srv.incoming().take(1);
+    let copied = clients.collect().and_then(|clients| {
+        let mut clients = clients.into_iter();
+        let a = clients.next().unwrap();
+
+        read_to_end(a.take(4), Vec::new())
+    });
+
+    let (_, data) = t!(copied.wait());
+    t.join().unwrap();
+
+    assert_eq!(data, b"foo ");
+}