diff options
Diffstat (limited to 'test/test_producer_integration.py')
| -rw-r--r-- | test/test_producer_integration.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index e0939a6..8f32cf8 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -13,6 +13,7 @@ from kafka import ( from kafka.codec import has_snappy from kafka.errors import UnknownTopicOrPartitionError, LeaderNotAvailableError from kafka.producer.base import Producer +from kafka.protocol.message import PartialMessage from kafka.structs import FetchRequestPayload, ProduceRequestPayload from test.fixtures import ZookeeperFixture, KafkaFixture @@ -521,7 +522,8 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): self.assertEqual(resp.error, 0) self.assertEqual(resp.partition, partition) - messages = [ x.message.value for x in resp.messages ] + messages = [ x.message.value for x in resp.messages + if not isinstance(x.message, PartialMessage) ] self.assertEqual(messages, expected_messages) self.assertEqual(resp.highwaterMark, start_offset+len(expected_messages)) |