From e3b1ad24b80dd60e3159566740f40fc6f5811070 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sun, 15 Oct 2017 21:52:58 -0700 Subject: Move callback processing from BrokerConnection to KafkaClient --- test/test_client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'test') diff --git a/test/test_client.py b/test/test_client.py index 42a1623..d02c621 100644 --- a/test/test_client.py +++ b/test/test_client.py @@ -28,6 +28,7 @@ def mock_conn(conn, success=True): else: mocked.send.return_value = Future().failure(Exception()) conn.return_value = mocked + conn.recv.return_value = [] class TestSimpleClient(unittest.TestCase): @@ -94,7 +95,7 @@ class TestSimpleClient(unittest.TestCase): mock_conn(mocked_conns[('kafka03', 9092)], success=False) future = Future() mocked_conns[('kafka02', 9092)].send.return_value = future - mocked_conns[('kafka02', 9092)].recv.side_effect = lambda: future.success('valid response') + mocked_conns[('kafka02', 9092)].recv.return_value = [('valid response', future)] def mock_get_conn(host, port, afi): return mocked_conns[(host, port)] -- cgit v1.2.1