summaryrefslogtreecommitdiff
path: root/test/test_consumer_integration.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2015-06-20 09:39:17 -0700
committerDana Powers <dana.powers@gmail.com>2015-06-20 09:39:17 -0700
commitadbd4ac052e4a5b40cfc2a3589b7adbcb656afe5 (patch)
treeee5dabf6b7383d1ebf94c682f2fa235bb3145a1e /test/test_consumer_integration.py
parent4b064340e8796cb85307e014853ddb0435809f2f (diff)
parent2028a232098abeb89a8125e26abc4f4a379ef1b9 (diff)
downloadkafka-python-adbd4ac052e4a5b40cfc2a3589b7adbcb656afe5.tar.gz
Merge pull request #412 from haosdent/seek_absolute_offset
fix #410 SimpleConsumer cannot seek to an absolute offset.
Diffstat (limited to 'test/test_consumer_integration.py')
-rw-r--r--test/test_consumer_integration.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py
index df2eeea..52b3e85 100644
--- a/test/test_consumer_integration.py
+++ b/test/test_consumer_integration.py
@@ -164,6 +164,20 @@ class TestConsumerIntegration(KafkaIntegrationTestCase):
consumer.seek(-13, 2)
self.assert_message_count([ message for message in consumer ], 13)
+ # Set absolute offset
+ consumer.seek(100)
+ self.assert_message_count([ message for message in consumer ], 0)
+ consumer.seek(100, partition=0)
+ self.assert_message_count([ message for message in consumer ], 0)
+ consumer.seek(101, partition=1)
+ self.assert_message_count([ message for message in consumer ], 0)
+ consumer.seek(90, partition=0)
+ self.assert_message_count([ message for message in consumer ], 10)
+ consumer.seek(20, partition=1)
+ self.assert_message_count([ message for message in consumer ], 80)
+ consumer.seek(0, partition=1)
+ self.assert_message_count([ message for message in consumer ], 100)
+
consumer.stop()
@kafka_versions("all")