diff options
| author | Dana Powers <dana.powers@gmail.com> | 2016-04-07 21:30:35 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2016-04-07 22:54:08 -0700 |
| commit | d05f6aa1132a89a35e788655a881bff697170999 (patch) | |
| tree | 45622db1b0400d64c315d706a23adc3df5dca57a | |
| parent | e78959a8ca682da7f7e7887aa441fde7d5f605e9 (diff) | |
| download | kafka-python-d05f6aa1132a89a35e788655a881bff697170999.tar.gz | |
Reduce batch time in producer integration test (speedup test)
| -rw-r--r-- | test/test_producer_integration.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index 176c99e..a304e83 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -398,14 +398,17 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): partition = self.client.get_partition_ids_for_topic(self.topic)[0] start_offset = self.current_offset(self.topic, partition) - producer = KeyedProducer(self.client, partitioner = RoundRobinPartitioner, async=True) + producer = KeyedProducer(self.client, + partitioner=RoundRobinPartitioner, + async=True, + batch_send_every_t=1) resp = producer.send_messages(self.topic, self.key("key1"), self.msg("one")) self.assertEqual(len(resp), 0) # wait for the server to report a new highwatermark while self.current_offset(self.topic, partition) == start_offset: - time.sleep(0.1) + time.sleep(0.1) self.assert_fetch_offset(partition, start_offset, [ self.msg("one") ]) |