From 392d674be6641078717a4d87e471916c9a4bbb22 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sun, 29 Sep 2019 17:04:17 -0700 Subject: Send socket data via non-blocking IO with send buffer (#1912) --- test/test_client_async.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'test') diff --git a/test/test_client_async.py b/test/test_client_async.py index 8bb2028..74da66a 100644 --- a/test/test_client_async.py +++ b/test/test_client_async.py @@ -25,6 +25,7 @@ from kafka.structs import BrokerMetadata @pytest.fixture def cli(mocker, conn): client = KafkaClient(api_version=(0, 9)) + mocker.patch.object(client, '_selector') client.poll(future=client.cluster.request_update()) return client @@ -32,6 +33,7 @@ def cli(mocker, conn): def test_bootstrap(mocker, conn): conn.state = ConnectionStates.CONNECTED cli = KafkaClient(api_version=(0, 9)) + mocker.patch.object(cli, '_selector') future = cli.cluster.request_update() cli.poll(future=future) @@ -86,7 +88,7 @@ def test_maybe_connect(cli, conn): def test_conn_state_change(mocker, cli, conn): - sel = mocker.patch.object(cli, '_selector') + sel = cli._selector node_id = 0 cli._conns[node_id] = conn -- cgit v1.2.1