Update conn fixture to check for HANDSHAKE state

2016-04-04 18:57:32 -07:00
parent 5b3042cb6d
commit 688b7755a9
1 changed files with 2 additions and 1 deletions
--- a/test/conftest.py
+++ b/test/conftest.py
@@ -51,7 +51,8 @@ def conn(mocker):
        return state
    conn._set_conn_state = _set_conn_state
    conn.connect.side_effect = lambda: conn.state
-    conn.connecting = lambda: conn.state is ConnectionStates.CONNECTING
+    conn.connecting = lambda: conn.state in (ConnectionStates.CONNECTING,
+                                             ConnectionStates.HANDSHAKE)
    conn.connected = lambda: conn.state is ConnectionStates.CONNECTED
    conn.disconnected = lambda: conn.state is ConnectionStates.DISCONNECTED
    return conn