Added generic fixtures for testing

Author: Relrin

tests/amqp/test_extension.py

@@ -1,22 +1,10 @@
 import asyncio
-import os
 
 import pytest
 from sage_utils.amqp.extension import AmqpExtension
 from sage_utils.amqp.base import AmqpWorker
 
-
-class FakeConfig(object):
-
-    def get(self, key, default=None):
-        return os.environ.get(key, default)
-
-
-class Application(object):
-
-    def __init__(self, config=None):
-        super(Application, self).__init__()
-        self.config = config
+from tests.fixtures import Application, FakeConfig
 
 
 class CustomWorker(AmqpWorker):

tests/fixtures.py

@@ -0,0 +1,14 @@
+import os
+
+
+class FakeConfig(object):
+
+    def get(self, key, default=None):
+        return os.environ.get(key, default)
+
+
+class Application(object):
+
+    def __init__(self, config=None):
+        super(Application, self).__init__()
+        self.config = config

tests/test_extension.py

@@ -2,7 +2,10 @@
 from sage_utils.extension import BaseExtension
 
 
-class FakeConfig(object):
+from tests.fixtures import Application, FakeConfig as BaseFakeConfig
+
+
+class FakeConfig(BaseFakeConfig):
     key = 'value'
 
 
@@ -11,13 +14,6 @@ class CustomExtension(BaseExtension):
     app_attribute = 'ext'
 
 
-class Application(object):
-
-    def __init__(self, config=None):
-        super(Application, self).__init__()
-        self.config = config
-
-
 def test_constructor_with_init():
     app = Application()
     extension = CustomExtension(app)