[telepathy-gabble/master] BytestreamS5BRelay.wait_bytestream_closed: wait for events if needed

[Guillaume Desmottes]

---
 tests/twisted/bytestream.py |    7 +++++--
 1 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/twisted/bytestream.py b/tests/twisted/bytestream.py
index 02587cf..f1772fb 100644
--- a/tests/twisted/bytestream.py
+++ b/tests/twisted/bytestream.py
@@ -533,8 +533,11 @@ class BytestreamS5BRelay(BytestreamS5B):
         # connects, auth and requests connection
         return True
 
-    def wait_bytestream_closed(self):
-        pass
+    def wait_bytestream_closed(self, expected=[]):
+        if expected == []:
+            return []
+
+        return self.q.expect_many(*expected)
 
 
 class BytestreamS5BRelayBugged(BytestreamS5BRelay):
-- 
1.5.6.5