diff options
Diffstat (limited to 'test/test_producer_integration.py')
| -rw-r--r-- | test/test_producer_integration.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py index 7109886..e0939a6 100644 --- a/test/test_producer_integration.py +++ b/test/test_producer_integration.py @@ -15,8 +15,8 @@ from kafka.errors import UnknownTopicOrPartitionError, LeaderNotAvailableError from kafka.producer.base import Producer from kafka.structs import FetchRequestPayload, ProduceRequestPayload -from test.fixtures import ZookeeperFixture, KafkaFixture, version -from test.testutil import KafkaIntegrationTestCase, kafka_versions, current_offset +from test.fixtures import ZookeeperFixture, KafkaFixture +from test.testutil import KafkaIntegrationTestCase, env_kafka_version, current_offset # TODO: This duplicates a TestKafkaProducerIntegration method temporarily @@ -43,7 +43,7 @@ def assert_produce_response(resp, initial_offset): assert resp[0].offset == initial_offset -@pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set") +@pytest.mark.skipif(not env_kafka_version(), reason="No KAFKA_VERSION set") def test_produce_many_simple(simple_client, topic): """Test multiple produces using the SimpleClient """ @@ -353,7 +353,7 @@ class TestKafkaProducerIntegration(KafkaIntegrationTestCase): # KeyedProducer Tests # ############################ - @kafka_versions('>=0.8.1') + @pytest.mark.skipif(not env_kafka_version(), reason="No KAFKA_VERSION set") def test_keyedproducer_null_payload(self): partitions = self.client.get_partition_ids_for_topic(self.topic) start_offsets = [self.current_offset(self.topic, p) for p in partitions] |