From 8aa719c53205d3f168c601bad39eb8f7c7a8eb1f Mon Sep 17 00:00:00 2001
From: Toralf Wittner <tw@dtex.org>
Date: Thu, 21 Jun 2018 11:57:15 +0200
Subject: [PATCH] Return `WouldBlock` if flushing did not complete. (#265)

---
 rw-stream-sink/src/lib.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/rw-stream-sink/src/lib.rs b/rw-stream-sink/src/lib.rs
index fe6e9eeb..c59e2551 100644
--- a/rw-stream-sink/src/lib.rs
+++ b/rw-stream-sink/src/lib.rs
@@ -145,8 +145,10 @@ where
 
     #[inline]
     fn flush(&mut self) -> Result<(), IoError> {
-        self.inner.poll_complete()?;
-        Ok(())
+        match self.inner.poll_complete()? {
+            Async::Ready(()) => Ok(()),
+            Async::NotReady => Err(IoError::new(IoErrorKind::WouldBlock, "not ready"))
+        }
     }
 }