oops, fix tests

In some tests, we artificially call _event_connected() without first initiating a connection.
2016-06-03 23:30:31 -07:00 · 2016-06-03 23:30:31 -07:00 · 996c739b2a
commit 996c739b2a
parent 5b5b4b7134
1 changed files with 2 additions and 1 deletions
--- a/src/wormhole/wormhole.py
+++ b/src/wormhole/wormhole.py
@ -371,7 +371,8 @@ class _Wormhole:

    def _event_connected(self, ws):
        self._ws = ws
-        self._ws_t.finish()
+        if self._ws_t:
+            self._ws_t.finish()

    def _event_ws_opened(self, _):
        self._connection_state = OPEN