Adds state to integration tests

kaufland-ecommerce · Sep 19, 2019 · 15e7d64 · 15e7d64
1 parent 6ac8a71
commit 15e7d64
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -12,6 +12,7 @@
 from confluent_kafka.cimpl import Producer, TopicPartition
 from pykafka.exceptions import NoBrokersAvailableError
 
+from esque.cli.options import State
 from esque.cluster import Cluster
 from esque.config import sample_config_path, Config
 from esque.errors import raise_for_kafka_error
@@ -211,3 +212,8 @@ def cluster(test_config):
         raise ex
 
     yield cluster
+
+
+@pytest.fixture()
+def state(test_config):
+    yield State()
diff --git a/tests/integration/test_topic_controller.py b/tests/integration/test_topic_controller.py
@@ -3,17 +3,11 @@
 import confluent_kafka
 import pytest
 
-from esque.cli.options import State
 from esque.controller.topic_controller import TopicController
 from esque.errors import KafkaException
 from esque.resources.topic import Topic, TopicDiff
 
 
-@pytest.fixture()
-def state(test_config):
-    yield State()
-
-
 @pytest.fixture()
 def topic_controller(cluster):
     yield cluster.topic_controller