diff options
Diffstat (limited to 'test/test_producer_legacy.py')
| -rw-r--r-- | test/test_producer_legacy.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/test/test_producer_legacy.py b/test/test_producer_legacy.py index 9b87c76..6d00116 100644 --- a/test/test_producer_legacy.py +++ b/test/test_producer_legacy.py @@ -73,7 +73,7 @@ class TestKafkaProducer(unittest.TestCase): @patch('kafka.producer.base._send_upstream') def test_producer_async_queue_overfilled(self, mock): queue_size = 2 - producer = Producer(MagicMock(), async=True, + producer = Producer(MagicMock(), async_send=True, async_queue_maxsize=queue_size) topic = b'test-topic' @@ -95,25 +95,25 @@ class TestKafkaProducer(unittest.TestCase): with patch.object(SimpleClient, '_send_broker_aware_request', return_value = [error]): client = SimpleClient(MagicMock()) - producer = SimpleProducer(client, async=False, sync_fail_on_error=False) + producer = SimpleProducer(client, async_send=False, sync_fail_on_error=False) # 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) + producer = SimpleProducer(client, async_send=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) + producer = Producer(MagicMock(), async_send=True) producer.stopped = True with patch.object(producer, 'stop') as mocked_stop: producer._cleanup_func(producer) self.assertEqual(mocked_stop.call_count, 0) def test_cleanup_is_called_on_running_producer(self): - producer = Producer(MagicMock(), async=True) + producer = Producer(MagicMock(), async_send=True) producer.stopped = False with patch.object(producer, 'stop') as mocked_stop: producer._cleanup_func(producer) |