summaryrefslogtreecommitdiff
path: root/test/test_coordinator.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2019-12-29 16:06:24 -0800
committerGitHub <noreply@github.com>2019-12-29 16:06:24 -0800
commit2a86b23f477e5ed57aa987db97d11284a37d05a0 (patch)
tree391d4d2d99db38eb2f9869d76592851bbae57cf2 /test/test_coordinator.py
parent1a91a54688cb77fd77c342e719f24f346d5cee89 (diff)
downloadkafka-python-2a86b23f477e5ed57aa987db97d11284a37d05a0.tar.gz
Optionally return OffsetAndMetadata from consumer.committed(tp) (#1979)
Diffstat (limited to 'test/test_coordinator.py')
-rw-r--r--test/test_coordinator.py4
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