2 files changed, 6 insertions, 4 deletions
diff --git a/test/test_client_async.py b/test/test_client_async.py
index 8874c67..97be827 100644
--- a/test/test_client_async.py
+++ b/test/test_client_async.py
@@ -236,13 +236,14 @@ def test_send(cli, conn):
     cli._maybe_connect(0)
     # ProduceRequest w/ 0 required_acks -> no response
     request = ProduceRequest[0](0, 0, [])
+    assert request.expect_response() is False
     ret = cli.send(0, request)
-    assert conn.send.called_with(request, expect_response=False)
+    assert conn.send.called_with(request)
     assert isinstance(ret, Future)
 
     request = MetadataRequest[0]([])
     cli.send(0, request)
-    assert conn.send.called_with(request, expect_response=True)
+    assert conn.send.called_with(request)
 
 
 def test_poll(mocker):
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