diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_producer_integration.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index 7d3a180..b4e076a 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -148,7 +148,8 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): producer = SimpleProducer(self.client) # At first it doesn't exist - with self.assertRaises(UnknownTopicOrPartitionError): + with self.assertRaises((UnknownTopicOrPartitionError, + LeaderNotAvailableError)): resp = producer.send_messages(new_topic, self.msg("one")) @kafka_versions("all") |