diff options
| author | Dana Powers <dana.powers@gmail.com> | 2019-09-28 15:58:17 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2019-09-28 15:58:17 -0700 |
| commit | 2e6a43839bde28c3eaeeb010bcdfb936ae54730f (patch) | |
| tree | ce7248ca21b3e167b022e1d70e76a21066a0a87e /test | |
| parent | 61fa0b27685c2d4e67d1b6575ca6797f36eb1bfa (diff) | |
| download | kafka-python-connection_delay_inf.tar.gz | |
Rely on selector to detect completed connection attemptsconnection_delay_inf
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_client_async.py | 2 | ||||
| -rw-r--r-- | test/test_conn.py | 2 |
2 files changed, 2 insertions, 2 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) diff --git a/test/test_conn.py b/test/test_conn.py index 6412cb6..966f7b3 100644 --- a/test/test_conn.py +++ b/test/test_conn.py @@ -85,7 +85,7 @@ def test_connection_delay(conn): conn.last_attempt = 1000 assert conn.connection_delay() == conn.config['reconnect_backoff_ms'] conn.state = ConnectionStates.CONNECTING - assert conn.connection_delay() == 0 + assert conn.connection_delay() == float('inf') conn.state = ConnectionStates.CONNECTED assert conn.connection_delay() == float('inf') |