moved files

shared/utils/tests/unit/check_env_test.py

@@ -0,0 +1,53 @@
+"""Definition of tests for check_env function."""
+
+import os
+import unittest
+
+from shared.utils import check_env
+
+
+class TestFunctionCheckEnv(unittest.TestCase):
+    """Testing function check_env."""
+
+    def setUp(self):
+        """Prepare for tests."""
+        # define variables for test
+        self.not_set_env_var = 'NOT_SET_ENV_VAR'
+        self.set_env_var_map = {
+            'ENV_VAR_UPPER': 'ENV_VAR_UPPER',
+            'ENV_VAR_LOWER': 'env_var_lower',
+        }
+        # set env vars
+        for key, val in self.set_env_var_map.items():
+            os.environ[key] = val
+
+    def tearDown(self):
+        """Clean up after tests."""
+        # remove env vars
+        for key in self.set_env_var_map:
+            _ = os.environ.pop(key, default=None)
+
+    def test_input_type(self):
+        """Test that function fails on wrong input types."""
+        with self.assertRaises(AssertionError):
+            check_env('not_a_list')
+        with self.assertRaises(AssertionError):
+            check_env({1, 2, 3})
+
+    def test_env_var_not_set(self):
+        """Test that function raises AssertionError when checking environment
+        variable that is not set."""
+        var_list = {self.not_set_env_var}
+        msg = f'environment variable not set: {self.not_set_env_var}'
+        with self.assertRaises(AssertionError, msg=msg):
+            check_env(var_list)
+
+    def test_env_vars_set(self):
+        """Test that function does not raise any errors when environment
+        variables are set."""
+        var_list = set(self.set_env_var_map.keys())
+        check_env(var_list)
+
+
+if __name__ == '__main__':
+    unittest.main()

shared/utils/tests/unit/setup_logging_test.py

@@ -0,0 +1,38 @@
+"""Definition of tests for setup_logging function."""
+
+import logging
+import os
+import unittest
+
+from shared.utils import setup_logging
+
+
+class TestFunctionSetupLogging(unittest.TestCase):
+    """Testing function setup_logging."""
+
+    def setUp(self):
+        # set log level names to test
+        self.env_var_name = 'LOG_LEVEL'
+        self.log_level_list = [
+            'debug',
+            'info',
+            'error',
+        ]
+        self.wrong_log_level_name = 'weird_name'
+
+    def test_log_level_gets_set(self):
+        """Test that function updates the logging level of the root logger."""
+        for log_level in self.log_level_list:
+            # set env var
+            os.environ[self.env_var_name] = log_level
+            # execute function
+            setup_logging()
+            # get logger
+            logger = logging.getLogger()
+            # check log level
+            level_number = getattr(logging, log_level.upper())
+            self.assertEqual(logger.getEffectiveLevel(), level_number)
+
+
+if __name__ == '__main__':
+    unittest.main()