From 9990e273f619b09743ef4298677f37fa12bcd867 Mon Sep 17 00:00:00 2001
From: Sean McArthur <sean@seanmonstar.com>
Date: Thu, 1 Mar 2018 14:13:02 -0800
Subject: [PATCH] test(server): fix streaming_body incorrect test

---
 tests/server.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/server.rs b/tests/server.rs
index 12e8a9bb..50e1c172 100644
--- a/tests/server.rs
+++ b/tests/server.rs
@@ -1029,6 +1029,7 @@ fn streaming_body() {
     let listener = TcpListener::bind(&"127.0.0.1:0".parse().unwrap(), &core.handle()).unwrap();
     let addr = listener.local_addr().unwrap();
 
+    let (tx, rx) = oneshot::channel();
     thread::spawn(move || {
         let mut tcp = connect(&addr);
         tcp.write_all(b"GET / HTTP/1.1\r\n\r\n").unwrap();
@@ -1045,9 +1046,12 @@ fn streaming_body() {
                 break;
             }
         }
-        assert_eq!(sum, 1_007_089);
+        assert_eq!(sum, 100_789);
+        let _ = tx.send(());
     });
 
+    let rx = rx.map_err(|_| panic!("thread panicked"));
+
     let fut = listener.incoming()
         .into_future()
         .map_err(|_| unreachable!())
@@ -1064,7 +1068,7 @@ fn streaming_body() {
                 .map(|_| ())
         });
 
-    core.run(fut).unwrap();
+    core.run(fut.join(rx)).unwrap();
 }
 
 #[test]