diff options
Diffstat (limited to 'test/test_producer_integration.py')
| -rw-r--r-- | test/test_producer_integration.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index c99ed63..ee0b2fd 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -11,7 +11,7 @@ from kafka import ( ) from kafka.codec import has_snappy from kafka.common import ( - FetchRequest, ProduceRequest, + FetchRequestPayload, ProduceRequestPayload, UnknownTopicOrPartitionError, LeaderNotAvailableError ) from kafka.producer.base import Producer @@ -488,7 +488,7 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): def assert_produce_request(self, messages, initial_offset, message_ct, partition=0): - produce = ProduceRequest(self.bytes_topic, partition, messages=messages) + produce = ProduceRequestPayload(self.bytes_topic, partition, messages=messages) # There should only be one response message from the server. # This will throw an exception if there's more than one. @@ -506,7 +506,7 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): # There should only be one response message from the server. # This will throw an exception if there's more than one. - resp, = self.client.send_fetch_request([ FetchRequest(self.bytes_topic, partition, start_offset, 1024) ]) + resp, = self.client.send_fetch_request([FetchRequestPayload(self.bytes_topic, partition, start_offset, 1024)]) self.assertEqual(resp.error, 0) self.assertEqual(resp.partition, partition) |