open.source/magic-wormhole
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test_server: error properly when EventSource fails

Browse Source
This commit is contained in:
Brian Warner 2015-11-23 16:49:46 -08:00
parent c482c248ff
commit 2318c94169
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/wormhole/test/test_server.py
View File

@ -328,6 +328,7 @@ class OneEventAtATime:
def __init__(self, url, parser=lambda e: e): def __init__(self, url, parser=lambda e: e):
self.parser = parser self.parser = parser
self.d = None self.d = None
self._connected = False
self.connected_d = defer.Deferred() self.connected_d = defer.Deferred()
self.disconnected_d = defer.Deferred() self.disconnected_d = defer.Deferred()
self.events = [] self.events = []
@ -358,11 +359,14 @@ class OneEventAtATime:
def wait_for_connection(self): def wait_for_connection(self):
return self.connected_d return self.connected_d
def connected(self): def connected(self):
self._connected = True
self.connected_d.callback(None) self.connected_d.callback(None)
def wait_for_disconnection(self): def wait_for_disconnection(self):
return self.disconnected_d return self.disconnected_d
def disconnected(self, why): def disconnected(self, why):
if not self._connected:
self.connected_d.errback(why)
self.disconnected_d.callback((why,)) self.disconnected_d.callback((why,))
class Summary(unittest.TestCase): class Summary(unittest.TestCase):