diff options
Diffstat (limited to 'test/test_producer_integration.py')
| -rw-r--r-- | test/test_producer_integration.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index ca0da6a..6cd3d13 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -2,6 +2,7 @@ import os import time import uuid +import pytest from six.moves import range from kafka import ( @@ -14,9 +15,11 @@ from kafka.errors import UnknownTopicOrPartitionError, LeaderNotAvailableError from kafka.producer.base import Producer from kafka.structs import FetchRequestPayload, ProduceRequestPayload +from test.conftest import version from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import KafkaIntegrationTestCase, kafka_versions, current_offset + # TODO: This duplicates a TestKafkaProducerIntegration method temporarily # while the migration to pytest is in progress def assert_produce_request(client, topic, messages, initial_offset, message_ct, @@ -32,6 +35,7 @@ def assert_produce_request(client, topic, messages, initial_offset, message_ct, assert current_offset(client, topic, partition) == initial_offset + message_ct + def assert_produce_response(resp, initial_offset): """Verify that a produce response is well-formed """ @@ -39,6 +43,8 @@ def assert_produce_response(resp, initial_offset): assert resp[0].error == 0 assert resp[0].offset == initial_offset + +@pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set") def test_produce_many_simple(simple_client, topic): """Test multiple produces using the SimpleClient """ |