openstack
/
deb-python-kafka
Code Issues Proposed changes

Do not test fetched_size with fetch_max_bytes since 0.11 brokers will return more data than 0.10

This commit is contained in:
Dana Powers 2017-06-17 08:47:44 -07:00
parent c5f1c6901f
commit 26a810220a
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_consumer_integration.py
View File

@ -588,20 +588,17 @@ class TestConsumerIntegration(KafkaIntegrationTestCase):
# Start a consumer
consumer = self.kafka_consumer(
auto_offset_reset='earliest', fetch_max_bytes=300)
fetched_size = 0
seen_partitions = set([])
for i in range(10):
poll_res = consumer.poll(timeout_ms=100)
for partition, msgs in six.iteritems(poll_res):
for msg in msgs:
fetched_size += len(msg.value)
seen_partitions.add(partition)
# Check that we fetched at least 1 message from both partitions
self.assertEqual(
seen_partitions, set([
TopicPartition(self.topic, 0), TopicPartition(self.topic, 1)]))
self.assertLess(fetched_size, 3000)
@kafka_versions('>=0.10.1')
def test_kafka_consumer_max_bytes_one_msg(self):