diff options
Diffstat (limited to 'test/test_conn.py')
-rw-r--r-- | test/test_conn.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/test_conn.py b/test/test_conn.py index 248ab88..2c418d4 100644 --- a/test/test_conn.py +++ b/test/test_conn.py @@ -11,6 +11,7 @@ import pytest from kafka.conn import BrokerConnection, ConnectionStates, collect_hosts from kafka.protocol.api import RequestHeader from kafka.protocol.metadata import MetadataRequest +from kafka.protocol.produce import ProduceRequest import kafka.common as Errors @@ -112,7 +113,7 @@ def test_send_max_ifr(conn): def test_send_no_response(_socket, conn): conn.connect() assert conn.state is ConnectionStates.CONNECTED - req = MetadataRequest[0]([]) + req = ProduceRequest[0](required_acks=0, timeout=0, topics=[]) header = RequestHeader(req, client_id=conn.config['client_id']) payload_bytes = len(header.encode()) + len(req.encode()) third = payload_bytes // 3 @@ -120,7 +121,7 @@ def test_send_no_response(_socket, conn): _socket.send.side_effect = [4, third, third, third, remainder] assert len(conn.in_flight_requests) == 0 - f = conn.send(req, expect_response=False) + f = conn.send(req) assert f.succeeded() is True assert f.value is None assert len(conn.in_flight_requests) == 0 |