diff options
Diffstat (limited to 'test/test_coordinator.py')
| -rw-r--r-- | test/test_coordinator.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_coordinator.py b/test/test_coordinator.py index 4afdcd9..88ca4c1 100644 --- a/test/test_coordinator.py +++ b/test/test_coordinator.py @@ -20,7 +20,7 @@ from kafka.protocol.commit import ( OffsetCommitRequest, OffsetCommitResponse, OffsetFetchRequest, OffsetFetchResponse) from kafka.protocol.metadata import MetadataResponse -from kafka.structs import TopicPartition, OffsetAndMetadata +from kafka.structs import OffsetAndMetadata, TopicPartition from kafka.util import WeakMethod @@ -211,7 +211,7 @@ def test_refresh_committed_offsets_if_needed(mocker, coordinator): assert coordinator._subscription.needs_fetch_committed_offsets is True coordinator.refresh_committed_offsets_if_needed() assignment = coordinator._subscription.assignment - assert assignment[TopicPartition('foobar', 0)].committed == 123 + assert assignment[TopicPartition('foobar', 0)].committed == OffsetAndMetadata(123, b'') assert TopicPartition('foobar', 1) not in assignment assert coordinator._subscription.needs_fetch_committed_offsets is False |