diff options
| author | Dana Powers <dana.powers@gmail.com> | 2017-10-09 21:48:00 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2017-10-09 21:48:00 -0700 |
| commit | 671524e5ce95b891b8c14c7a79eeeeaf66407a76 (patch) | |
| tree | 9a4a79d7cf722ef0d7408f085aee63b96f4b8716 /test | |
| parent | 7305f03ff0758dad811d51f5e21006f273bb4dc2 (diff) | |
| download | kafka-python-KAFKA_4034_coordinator_lookup.tar.gz | |
KAFKA-4034: Avoid unnecessary consumer coordinator lookupKAFKA_4034_coordinator_lookup
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_coordinator.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/test/test_coordinator.py b/test/test_coordinator.py index 4115c03..aea2662 100644 --- a/test/test_coordinator.py +++ b/test/test_coordinator.py @@ -234,7 +234,7 @@ def test_fetch_committed_offsets(mocker, coordinator): assert coordinator._client.poll.call_count == 0 # general case -- send offset fetch request, get successful future - mocker.patch.object(coordinator, 'ensure_coordinator_known') + mocker.patch.object(coordinator, 'ensure_coordinator_ready') mocker.patch.object(coordinator, '_send_offset_fetch_request', return_value=Future().success('foobar')) partitions = [TopicPartition('foobar', 0)] @@ -295,16 +295,15 @@ def offsets(): def test_commit_offsets_async(mocker, coordinator, offsets): mocker.patch.object(coordinator._client, 'poll') - mocker.patch.object(coordinator, 'ensure_coordinator_known') + mocker.patch.object(coordinator, 'coordinator_unknown', return_value=False) mocker.patch.object(coordinator, '_send_offset_commit_request', return_value=Future().success('fizzbuzz')) - ret = coordinator.commit_offsets_async(offsets) - assert isinstance(ret, Future) + coordinator.commit_offsets_async(offsets) assert coordinator._send_offset_commit_request.call_count == 1 def test_commit_offsets_sync(mocker, coordinator, offsets): - mocker.patch.object(coordinator, 'ensure_coordinator_known') + mocker.patch.object(coordinator, 'ensure_coordinator_ready') mocker.patch.object(coordinator, '_send_offset_commit_request', return_value=Future().success('fizzbuzz')) cli = coordinator._client |