From c04e4394c499a1fac76d92d5a791e687e4cf532f Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@gmail.com>
Date: Thu, 22 Mar 2018 14:10:53 -0700
Subject: [PATCH] Fix skipped integration tests if KAFKA_VERSION unset

---
 test/test_consumer_integration.py | 3 +++
 test/test_producer_integration.py | 6 ++++++
 2 files changed, 9 insertions(+)

diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py
index 78a8a3c1e..cc036cc4b 100644
--- a/test/test_consumer_integration.py
+++ b/test/test_consumer_integration.py
@@ -2,6 +2,7 @@
 import os
 import time
 
+import pytest
 from six.moves import xrange
 import six
 
@@ -19,12 +20,14 @@
     ProduceRequestPayload, TopicPartition, OffsetAndTimestamp
 )
 
+from test.conftest import version
 from test.fixtures import ZookeeperFixture, KafkaFixture
 from test.testutil import (
     KafkaIntegrationTestCase, kafka_versions, random_string, Timer,
     send_messages
 )
 
+@pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set")
 def test_kafka_consumer(simple_client, topic, kafka_consumer_factory):
     """Test KafkaConsumer
     """
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py
index ca0da6abd..6cd3d13ad 100644
--- a/test/test_producer_integration.py
+++ b/test/test_producer_integration.py
@@ -2,6 +2,7 @@
 import time
 import uuid
 
+import pytest
 from six.moves import range
 
 from kafka import (
@@ -14,9 +15,11 @@
 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
     """