diff options
| author | Dana Powers <dana.powers@gmail.com> | 2015-12-17 23:28:35 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2015-12-17 23:28:35 -0800 |
| commit | 04a3a32eb724fee8c6c21195ea005fe8f35e9fb7 (patch) | |
| tree | 9526d8cb2571ffe97dac71935eb74d3e281e4da7 /test/test_producer.py | |
| parent | ba4ec47af00d4e45f9a48d36be16b20aece6556e (diff) | |
| parent | f89b9da1f3dc15cc728be22d532e4927d33c47f7 (diff) | |
| download | kafka-python-04a3a32eb724fee8c6c21195ea005fe8f35e9fb7.tar.gz | |
Merge pull request #489 from dpkp/producer_new_topic
Handle new topic auto-creation in SimpleProducer.send_messages
Diffstat (limited to 'test/test_producer.py')
| -rw-r--r-- | test/test_producer.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index 31282bf..cc65a0a 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -97,19 +97,20 @@ class TestKafkaProducer(unittest.TestCase): def test_producer_sync_fail_on_error(self): error = FailedPayloadsError('failure') with patch.object(KafkaClient, 'load_metadata_for_topics'): - with patch.object(KafkaClient, 'get_partition_ids_for_topic', return_value=[0, 1]): - with patch.object(KafkaClient, '_send_broker_aware_request', return_value = [error]): + with patch.object(KafkaClient, 'ensure_topic_exists'): + with patch.object(KafkaClient, 'get_partition_ids_for_topic', return_value=[0, 1]): + with patch.object(KafkaClient, '_send_broker_aware_request', return_value = [error]): - client = KafkaClient(MagicMock()) - producer = SimpleProducer(client, async=False, sync_fail_on_error=False) + client = KafkaClient(MagicMock()) + producer = SimpleProducer(client, async=False, sync_fail_on_error=False) - # This should not raise - (response,) = producer.send_messages('foobar', b'test message') - self.assertEqual(response, error) + # This should not raise + (response,) = producer.send_messages('foobar', b'test message') + self.assertEqual(response, error) - producer = SimpleProducer(client, async=False, sync_fail_on_error=True) - with self.assertRaises(FailedPayloadsError): - producer.send_messages('foobar', b'test message') + producer = SimpleProducer(client, async=False, sync_fail_on_error=True) + with self.assertRaises(FailedPayloadsError): + producer.send_messages('foobar', b'test message') def test_cleanup_is_not_called_on_stopped_producer(self): producer = Producer(MagicMock(), async=True) |