summaryrefslogtreecommitdiff
path: root/test/test_client_async.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2019-09-28 19:30:09 -0700
committerGitHub <noreply@github.com>2019-09-28 19:30:09 -0700
commit89bf6a6ee51e8a54f909eae4785d04e485b91198 (patch)
treedcecac3f96738699441226c4f4a38f2cedc3632e /test/test_client_async.py
parent5d1d42429e07f4aa2959b488ea76efb6d0bafc79 (diff)
downloadkafka-python-89bf6a6ee51e8a54f909eae4785d04e485b91198.tar.gz
Rely on socket selector to detect completed connection attempts (#1909)
Diffstat (limited to 'test/test_client_async.py')
-rw-r--r--test/test_client_async.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/test/test_client_async.py b/test/test_client_async.py
index 82d1467..8bb2028 100644
--- a/test/test_client_async.py
+++ b/test/test_client_async.py
@@ -94,7 +94,7 @@ def test_conn_state_change(mocker, cli, conn):
sock = conn._sock
cli._conn_state_change(node_id, sock, conn)
assert node_id in cli._connecting
- sel.register.assert_called_with(sock, selectors.EVENT_WRITE)
+ sel.register.assert_called_with(sock, selectors.EVENT_WRITE, conn)
conn.state = ConnectionStates.CONNECTED
cli._conn_state_change(node_id, sock, conn)