From 86552bf3a53730566844f4b44684373dec57b9ce Mon Sep 17 00:00:00 2001
From: Titouan Rigoudy <titouan.rigoudy@gmail.com>
Date: Sat, 23 Jan 2021 15:27:22 +0100
Subject: [PATCH] Check all responses in stream_closed test.

---
 src/proto/server/client.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/proto/server/client.rs b/src/proto/server/client.rs
index 8d296eb..acaea6d 100644
--- a/src/proto/server/client.rs
+++ b/src/proto/server/client.rs
@@ -292,13 +292,14 @@ mod tests {
     // chance to send all of `outbound`.
     handle.shutdown();
 
-    // Check that the client returns the correct error.
+    // Check that the client returns the correct error, then stops running.
     assert!(inbound
       .next()
       .await
       .unwrap()
       .unwrap_err()
       .is_stream_closed());
+    assert!(inbound.next().await.is_none());
 
     server_task.await.unwrap().unwrap();
   }