diff options
| author | Dana Powers <dana.powers@gmail.com> | 2015-06-06 20:10:20 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2015-06-06 20:10:20 -0700 |
| commit | f32009cef42e8ca706c661da209107e4513899da (patch) | |
| tree | c2eb33cb630d0c698796f3f5e404cb0ec7d3b40d /test | |
| parent | 87bea900df341ddc2431569bdc329d069b6c5c61 (diff) | |
| parent | 1d5f4b1f889737ef3ba04d8303a02a4957a2d183 (diff) | |
| download | kafka-python-f32009cef42e8ca706c661da209107e4513899da.tar.gz | |
Merge pull request #388 from dpkp/331_fixups
async producer fixups
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_producer.py | 17 | ||||
| -rw-r--r-- | test/test_producer_integration.py | 4 |
2 files changed, 2 insertions, 19 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index 85a5a2e..c12af02 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -57,23 +57,6 @@ class TestKafkaProducer(unittest.TestCase): assert client.send_produce_request.called @patch('kafka.producer.base._send_upstream') - def test_producer_async_queue_overfilled_batch_send(self, mock): - queue_size = 2 - producer = Producer(MagicMock(), batch_send=True, - async_queue_maxsize=queue_size) - - topic = b'test-topic' - partition = 0 - message = b'test-message' - - with self.assertRaises(AsyncProducerQueueFull): - message_list = [message] * (queue_size + 1) - producer.send_messages(topic, partition, *message_list) - self.assertEqual(producer.queue.qsize(), queue_size) - for _ in xrange(producer.queue.qsize()): - producer.queue.get() - - @patch('kafka.producer.base._send_upstream') def test_producer_async_queue_overfilled(self, mock): queue_size = 2 producer = Producer(MagicMock(), async=True, diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index 099b975..3c414e1 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -221,7 +221,7 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): batch_interval = 5 producer = SimpleProducer( self.client, - batch_send=True, + async=True, batch_send_every_n=batch_messages, batch_send_every_t=batch_interval, random_start=False) @@ -287,7 +287,7 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): batch_interval = 5 producer = SimpleProducer( self.client, - batch_send=True, + async=True, batch_send_every_n=100, batch_send_every_t=batch_interval, random_start=False) |