summaryrefslogtreecommitdiff
path: root/test/test_client_async.py
diff options
context:
space:
mode:
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)