diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-12-10 12:21:37 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-12-10 13:31:59 -0800 |
| commit | 4bd20aad2b7a0710458545009328f8729c89fee1 (patch) | |
| tree | eab1c86b1e63418b8d7f7fe13e66d354c3178082 /test/test_failover_integration.py | |
| parent | 1856063f4e0c36a8ec6266358d82432adf879170 (diff) | |
| download | kafka-python-kafka_version_tests.tar.gz | |
Refactor kafka_versions to support arbitrary operators (> >= < <= ! =)kafka_version_tests
Diffstat (limited to 'test/test_failover_integration.py')
| -rw-r--r-- | test/test_failover_integration.py | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index 91779d7..28d671a 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -8,9 +8,7 @@ from kafka.producer.base import Producer from kafka.util import kafka_bytestring from test.fixtures import ZookeeperFixture, KafkaFixture -from test.testutil import ( - KafkaIntegrationTestCase, kafka_versions, random_string -) +from test.testutil import KafkaIntegrationTestCase, random_string log = logging.getLogger(__name__) @@ -21,7 +19,7 @@ class TestFailover(KafkaIntegrationTestCase): def setUp(self): if not os.environ.get('KAFKA_VERSION'): - return + self.skipTest('integration test requires KAFKA_VERSION') zk_chroot = random_string(10) replicas = 3 @@ -46,7 +44,6 @@ class TestFailover(KafkaIntegrationTestCase): broker.close() self.zk.close() - @kafka_versions("all") def test_switch_leader(self): topic = self.topic partition = 0 @@ -94,7 +91,6 @@ class TestFailover(KafkaIntegrationTestCase): self.assert_message_count(topic, 201, partitions=(partition,), at_least=True) - @kafka_versions("all") def test_switch_leader_async(self): topic = self.topic partition = 0 @@ -142,7 +138,6 @@ class TestFailover(KafkaIntegrationTestCase): self.assert_message_count(topic, 21, partitions=(partition + 1,), at_least=True) - @kafka_versions("all") def test_switch_leader_keyed_producer(self): topic = self.topic @@ -180,7 +175,6 @@ class TestFailover(KafkaIntegrationTestCase): msg = random_string(10).encode('utf-8') producer.send_messages(topic, key, msg) - @kafka_versions("all") def test_switch_leader_simple_consumer(self): producer = Producer(self.client, async=False) consumer = SimpleConsumer(self.client, None, self.topic, partitions=None, auto_commit=False, iter_timeout=10) |