Merge pull request #126 from kpn/fix/end_offsets

fix: end_offsets off by 1
kpn · Aug 2, 2023 · 7c3d84c · 7c3d84c
2 parents c884319 + dd603f1
commit 7c3d84c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/kstreams/test_utils/test_clients.py b/kstreams/test_utils/test_clients.py
@@ -158,7 +158,7 @@ async def end_offsets(
     ) -> Dict[TopicPartition, int]:
         topic = TopicManager.get(partitions[0].topic)
         end_offsets = {
-            topic_partition: topic.offset(partition=topic_partition.partition)
+            topic_partition: topic.offset(partition=topic_partition.partition) + 1
             for topic_partition in partitions
         }
         return end_offsets

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -230,9 +230,9 @@ async def consume(stream):
 
         stream = stream_engine.get_stream("my-stream")
         assert (await stream.consumer.end_offsets(topic_partitions)) == {
-            TopicPartition(topic="local--kstreams", partition=0): 1,
-            TopicPartition(topic="local--kstreams", partition=2): 0,
-            TopicPartition(topic="local--kstreams", partition=10): 0,
+            TopicPartition(topic="local--kstreams", partition=0): 2,
+            TopicPartition(topic="local--kstreams", partition=2): 1,
+            TopicPartition(topic="local--kstreams", partition=10): 1,
         }