diff options
| author | Dana Powers <dana.powers@gmail.com> | 2019-03-31 10:11:21 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2019-03-31 16:42:41 -0700 |
| commit | 9638d1cfaa42d0902e39db1f6c3db103cd7a0cc9 (patch) | |
| tree | 9542d82a7ed8609da0a389420278b7cebe6289c2 /test/test_client_async.py | |
| parent | b1effa24aca3a6bcf2268354caae12ee82d6b36d (diff) | |
| download | kafka-python-popped_conn_close.tar.gz | |
Dont treat popped conn.close() as failure in state change callbackpopped_conn_close
Diffstat (limited to 'test/test_client_async.py')
| -rw-r--r-- | test/test_client_async.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/test_client_async.py b/test/test_client_async.py index 246e36c..0951cb4 100644 --- a/test/test_client_async.py +++ b/test/test_client_async.py @@ -93,6 +93,7 @@ def test_conn_state_change(mocker, cli, conn): sel = mocker.patch.object(cli, '_selector') node_id = 0 + cli._conns[node_id] = conn conn.state = ConnectionStates.CONNECTING cli._conn_state_change(node_id, conn) assert node_id in cli._connecting @@ -180,8 +181,8 @@ def test_close(mocker, cli, conn): # All node close cli._maybe_connect(1) cli.close() - # +3 close: node 0, node 1, node bootstrap - call_count += 3 + # +2 close: node 1, node bootstrap (node 0 already closed) + call_count += 2 assert conn.close.call_count == call_count |