From 29e96f82ce0b028c21dc35fed2f4c26da4c8efef Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Thu, 18 Feb 2016 22:56:05 -0800 Subject: Mock client.ready() call to test coordinator offset fetch request --- kafka/coordinator/consumer.py | 3 ++- test/test_coordinator.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/kafka/coordinator/consumer.py b/kafka/coordinator/consumer.py index e7a687e..0e610c7 100644 --- a/kafka/coordinator/consumer.py +++ b/kafka/coordinator/consumer.py @@ -563,7 +563,8 @@ class ConsumerCoordinator(BaseCoordinator): # Verify node is ready if not self._client.ready(node_id): - log.debug("Node %s not ready -- failing offset fetch request") + log.debug("Node %s not ready -- failing offset fetch request", + node_id) return Future().failure(Errors.NodeNotReadyError) log.debug("Fetching committed offsets for partitions: %s", partitions) diff --git a/test/test_coordinator.py b/test/test_coordinator.py index e0906c7..94e0e66 100644 --- a/test/test_coordinator.py +++ b/test/test_coordinator.py @@ -385,6 +385,7 @@ def patched_coord(mocker, coordinator): coordinator.coordinator_id = 0 mocker.patch.object(coordinator._client, 'least_loaded_node', return_value=1) + mocker.patch.object(coordinator._client, 'ready', return_value=True) mocker.patch.object(coordinator._client, 'send') mocker.spy(coordinator, '_failed_request') mocker.spy(coordinator, '_handle_offset_commit_response') -- cgit v1.2.1