openstack/deb-python-kafka
Code Issues Proposed changes

Update conn fixture to check for HANDSHAKE state

Browse Source
This commit is contained in:
Dana Powers
2016-04-04 18:57:32 -07:00
parent 5b3042cb6d
commit 688b7755a9
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/conftest.py
View File

@@ -51,7 +51,8 @@ def conn(mocker):
return state return state
conn._set_conn_state = _set_conn_state conn._set_conn_state = _set_conn_state
conn.connect.side_effect = lambda: 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.connected = lambda: conn.state is ConnectionStates.CONNECTED
conn.disconnected = lambda: conn.state is ConnectionStates.DISCONNECTED conn.disconnected = lambda: conn.state is ConnectionStates.DISCONNECTED
return conn return conn