#11 REFACTOR: Moved all tests into test classes

tests/unit/general/test_env.py

@@ -1,31 +1,31 @@
+from tests.base_test import TestBase
 from os import environ
 from warnings import warn
-from tests import old_env
-
-# Tests environment variables used within the projects domain are
-# set in the correct environment
-
-VARS = ['ENVIRON', 'APPSECRET', 'FILESTORE']
-
-ENV_STATES = ['test', 'prod']
 
 
-def test_env_vars():
-    """ Test that required environment variables are set
-        ahead of runtime
+class TestEnv(TestBase):
+    """ Tests environment variables used within the projects domain are
+        set in the correct environment
     """
-    for var in VARS:
-        env = environ.get(var)
+    VARS = ['ENVIRON', 'APPSECRET', 'FILESTORE']
 
-        # Check to see what variable we are comparing
-        if env is None:
-            warn(f"Variable {var} is not set!")
+    ENV_STATES = ['test', 'prod']
 
+    def test_env_vars(self):
+        """ Test that required environment variables are set
+            ahead of runtime
+        """
+        for var in self.VARS:
+            env = environ.get(var)
 
-def test_environment_var_state():
-    """ Tests that the 'ENVIRON' Environment variable
-        is in a correct state
-    """
-    var = old_env
-    assert var is not None
-    assert (var in ENV_STATES)
+            # Check to see what variable we are comparing
+            if env is None:
+                warn(f"Variable {var} is not set!")
+
+    def test_environment_var_state(self):
+        """ Tests that the 'ENVIRON' Environment variable
+            is in a correct state
+        """
+        var = self.old_env
+        assert var is not None
+        assert (var in self.ENV_STATES)