summaryrefslogtreecommitdiff
path: root/test/test_fetcher.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2019-12-29 12:02:10 -0800
committerDana Powers <dana.powers@gmail.com>2019-12-29 12:02:10 -0800
commit8f8deea7731e26b4cdee3ce5ae8bfdd8e3f57d3a (patch)
tree40bdde2881753565601a8754870e26f71a164af1 /test/test_fetcher.py
parentcf28da8420b007d836b0db9d865d74fee89cdbcd (diff)
downloadkafka-python-issue-1082-offset-and-metadata.tar.gz
Optionally return OffsetAndMetadata from consumer.committed(tp)issue-1082-offset-and-metadata
Diffstat (limited to 'test/test_fetcher.py')
-rw-r--r--test/test_fetcher.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_fetcher.py b/test/test_fetcher.py
index b61a0f0..697f8be 100644
--- a/test/test_fetcher.py
+++ b/test/test_fetcher.py
@@ -21,7 +21,7 @@ from kafka.errors import (
UnknownTopicOrPartitionError, OffsetOutOfRangeError
)
from kafka.record.memory_records import MemoryRecordsBuilder, MemoryRecords
-from kafka.structs import TopicPartition
+from kafka.structs import OffsetAndMetadata, TopicPartition
@pytest.fixture
@@ -124,7 +124,7 @@ def test_update_fetch_positions(fetcher, topic, mocker):
fetcher._reset_offset.reset_mock()
fetcher._subscriptions.need_offset_reset(partition)
fetcher._subscriptions.assignment[partition].awaiting_reset = False
- fetcher._subscriptions.assignment[partition].committed = 123
+ fetcher._subscriptions.assignment[partition].committed = OffsetAndMetadata(123, b'')
mocker.patch.object(fetcher._subscriptions, 'seek')
fetcher.update_fetch_positions([partition])
assert fetcher._reset_offset.call_count == 0