diff options
| -rw-r--r-- | test/__init__.py | 3 | ||||
| -rw-r--r-- | test/test_coordinator.py | 2 |
2 files changed, 4 insertions, 1 deletions
diff --git a/test/__init__.py b/test/__init__.py index f91d0fa..0eb2edc 100644 --- a/test/__init__.py +++ b/test/__init__.py @@ -15,3 +15,6 @@ except ImportError: pass logging.getLogger(__name__).addHandler(NullHandler()) + +from kafka.future import Future +Future.error_on_callbacks = True # always fail during testing diff --git a/test/test_coordinator.py b/test/test_coordinator.py index 4b90f30..15b915d 100644 --- a/test/test_coordinator.py +++ b/test/test_coordinator.py @@ -548,7 +548,7 @@ def test_send_offset_fetch_request_success(patched_coord, partitions): patched_coord._client.send.return_value = _f future = patched_coord._send_offset_fetch_request(partitions) (node, request), _ = patched_coord._client.send.call_args - response = OffsetFetchResponse[0]([('foobar', [(0, 0), (1, 0)])]) + response = OffsetFetchResponse[0]([('foobar', [(0, 123, b'', 0), (1, 234, b'', 0)])]) _f.success(response) patched_coord._handle_offset_fetch_response.assert_called_with( future, response) |