diff options
| -rw-r--r-- | test/test_consumer_integration.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py index 218ed2c..2169145 100644 --- a/test/test_consumer_integration.py +++ b/test/test_consumer_integration.py @@ -11,7 +11,9 @@ from kafka import ( create_gzip_message, KafkaProducer ) from kafka.consumer.base import MAX_FETCH_BUFFER_SIZE_BYTES -from kafka.errors import ConsumerFetchSizeTooSmall, OffsetOutOfRangeError +from kafka.errors import ( + ConsumerFetchSizeTooSmall, OffsetOutOfRangeError, UnsupportedVersionError +) from kafka.structs import ProduceRequestPayload, TopicPartition from test.fixtures import ZookeeperFixture, KafkaFixture @@ -666,5 +668,5 @@ class TestConsumerIntegration(KafkaIntegrationTestCase): consumer = self.kafka_consumer() tp = TopicPartition(self.topic, 0) - with self.assertRaises(): + with self.assertRaises(UnsupportedVersionError): consumer.offsets_for_times({tp: int(time.time())}) |