summaryrefslogtreecommitdiff
path: root/test/conftest.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/conftest.py')
-rw-r--r--test/conftest.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/test/conftest.py b/test/conftest.py
index ffaae03..b6d3e3e 100644
--- a/test/conftest.py
+++ b/test/conftest.py
@@ -10,6 +10,7 @@ def version():
"""Return the Kafka version set in the OS environment"""
return kafka_version()
+
@pytest.fixture(scope="module")
def zookeeper():
"""Return a Zookeeper fixture"""
@@ -17,11 +18,13 @@ def zookeeper():
yield zk_instance
zk_instance.close()
+
@pytest.fixture(scope="module")
def kafka_broker(kafka_broker_factory):
"""Return a Kafka broker fixture"""
return kafka_broker_factory()[0]
+
@pytest.fixture(scope="module")
def kafka_broker_factory(version, zookeeper):
"""Return a Kafka broker fixture factory"""
@@ -42,6 +45,7 @@ def kafka_broker_factory(version, zookeeper):
for broker in _brokers:
broker.close()
+
@pytest.fixture
def simple_client(kafka_broker, request, topic):
"""Return a SimpleClient fixture"""
@@ -50,6 +54,7 @@ def simple_client(kafka_broker, request, topic):
yield client
client.close()
+
@pytest.fixture
def kafka_client(kafka_broker, request):
"""Return a KafkaClient fixture"""
@@ -57,11 +62,13 @@ def kafka_client(kafka_broker, request):
yield client
client.close()
+
@pytest.fixture
def kafka_consumer(kafka_consumer_factory):
"""Return a KafkaConsumer fixture"""
return kafka_consumer_factory()
+
@pytest.fixture
def kafka_consumer_factory(kafka_broker, topic, request):
"""Return a KafkaConsumer factory fixture"""
@@ -79,11 +86,13 @@ def kafka_consumer_factory(kafka_broker, topic, request):
if _consumer[0]:
_consumer[0].close()
+
@pytest.fixture
def kafka_producer(kafka_producer_factory):
"""Return a KafkaProducer fixture"""
yield kafka_producer_factory()
+
@pytest.fixture
def kafka_producer_factory(kafka_broker, request):
"""Return a KafkaProduce factory fixture"""
@@ -100,6 +109,7 @@ def kafka_producer_factory(kafka_broker, request):
if _producer[0]:
_producer[0].close()
+
@pytest.fixture
def topic(kafka_broker, request):
"""Return a topic fixture"""
@@ -107,6 +117,7 @@ def topic(kafka_broker, request):
kafka_broker.create_topics([topic_name])
return topic_name
+
@pytest.fixture
def conn(mocker):
"""Return a connection mocker fixture"""