diff options
| author | Dana Powers <dana.powers@gmail.com> | 2016-04-04 18:57:32 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2016-04-09 09:24:39 -0700 |
| commit | 688b7755a9e364d72c9622fed226c849e41d9b05 (patch) | |
| tree | e0a367bf83e34d40cdf8111c668b59c9b0d46a3f | |
| parent | 5b3042cb6d4fe3bb70e30e2ce7e776d8f124a27e (diff) | |
| download | kafka-python-688b7755a9e364d72c9622fed226c849e41d9b05.tar.gz | |
Update conn fixture to check for HANDSHAKE state
| -rw-r--r-- | test/conftest.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/conftest.py b/test/conftest.py index 1f37960..c2ef1dd 100644 --- 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 |