From 2e6a43839bde28c3eaeeb010bcdfb936ae54730f Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sat, 28 Sep 2019 15:58:17 -0700 Subject: Rely on selector to detect completed connection attempts --- test/test_client_async.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'test/test_client_async.py') 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) -- cgit v1.2.1