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-08 09:23:57 -0700 |
| commit | 9c8cb7dea19b9e791aed7e7ec27ea854119d6ac3 (patch) | |
| tree | 9b3288901536396e5c18128fb5ea7b7cb032d3d7 /test/test_client_async.py | |
| parent | 2527ff30ce84452106a7a5cf7be0cff06bc921a0 (diff) | |
| download | kafka-python-9c8cb7dea19b9e791aed7e7ec27ea854119d6ac3.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() |