diff options
Diffstat (limited to 'test/conftest.py')
| -rw-r--r-- | test/conftest.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/test/conftest.py b/test/conftest.py index a389480..1f37960 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -46,7 +46,12 @@ def conn(mocker): [(0, 'foo', 12), (1, 'bar', 34)], # brokers [])) # topics conn.blacked_out.return_value = False + def _set_conn_state(state): + conn.state = state + return state + conn._set_conn_state = _set_conn_state conn.connect.side_effect = lambda: conn.state - conn.connecting = lambda: conn.connect() is ConnectionStates.CONNECTING - conn.connected = lambda: conn.connect() is ConnectionStates.CONNECTED + conn.connecting = lambda: conn.state is ConnectionStates.CONNECTING + conn.connected = lambda: conn.state is ConnectionStates.CONNECTED + conn.disconnected = lambda: conn.state is ConnectionStates.DISCONNECTED return conn |