diff options
| author | Dana Powers <dana.powers@gmail.com> | 2016-04-07 16:11:25 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2016-04-07 16:53:29 -0700 |
| commit | 3b26839e90c84f19abee68d8e7cc933f8f391859 (patch) | |
| tree | 10c4080921a74b392aca192badcf4fee3c084509 /test/test_client_async.py | |
| parent | 8b592cda918b2890768386f40f4f27ab82d46f9a (diff) | |
| download | kafka-python-3b26839e90c84f19abee68d8e7cc933f8f391859.tar.gz | |
Add BrokerConnection.disconnected() method; update tests
Diffstat (limited to 'test/test_client_async.py')
| -rw-r--r-- | test/test_client_async.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/test/test_client_async.py b/test/test_client_async.py index c326d55..88f0fc7 100644 --- a/test/test_client_async.py +++ b/test/test_client_async.py @@ -85,21 +85,20 @@ def test_maybe_connect(conn): assert 0 not in cli._conns conn.state = ConnectionStates.DISCONNECTED - conn.connect.side_effect = lambda: ConnectionStates.CONNECTING + conn.connect.side_effect = lambda: conn._set_conn_state(ConnectionStates.CONNECTING) assert cli._maybe_connect(0) is False assert cli._conns[0] is conn assert 0 in cli._connecting - conn.state = ConnectionStates.CONNECTING - conn.connect.side_effect = lambda: ConnectionStates.CONNECTED + conn.connect.side_effect = lambda: conn._set_conn_state(ConnectionStates.CONNECTED) assert cli._maybe_connect(0) is True assert 0 not in cli._connecting # Failure to connect should trigger metadata update assert cli.cluster._need_update is False - cli._connecting.add(0) conn.state = ConnectionStates.CONNECTING - conn.connect.side_effect = lambda: ConnectionStates.DISCONNECTED + cli._connecting.add(0) + conn.connect.side_effect = lambda: conn._set_conn_state(ConnectionStates.DISCONNECTED) assert cli._maybe_connect(0) is False assert 0 not in cli._connecting assert cli.cluster._need_update is True @@ -155,7 +154,7 @@ def test_ready(conn): # connecting node connects cli._connecting.add(0) conn.state = ConnectionStates.CONNECTING - conn.connect.side_effect = lambda: ConnectionStates.CONNECTED + conn.connect.side_effect = lambda: conn._set_conn_state(ConnectionStates.CONNECTED) cli.ready(0) assert 0 not in cli._connecting assert cli._conns[0].connect.called_with() |