Uses assert_called_with instead of called_with (#2375)

dpkp · Nov 3, 2023 · 4861bee · 4861bee
1 parent 779a23c
commit 4861bee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_client_async.py b/test/test_client_async.py
@@ -220,12 +220,12 @@ def test_send(cli, conn):
     request = ProduceRequest[0](0, 0, [])
     assert request.expect_response() is False
     ret = cli.send(0, request)
-    assert conn.send.called_with(request)
+    conn.send.assert_called_with(request, blocking=False)
     assert isinstance(ret, Future)
 
     request = MetadataRequest[0]([])
     cli.send(0, request)
-    assert conn.send.called_with(request)
+    conn.send.assert_called_with(request, blocking=False)
 
 
 def test_poll(mocker):