summaryrefslogtreecommitdiff
path: root/test/test_conn.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_conn.py')
-rw-r--r--test/test_conn.py5
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