From 909f8f220cc4114465bd9d0d0204285530fcaa90 Mon Sep 17 00:00:00 2001
From: brian <schnitzelen@gmail.com>
Date: Sun, 20 Oct 2024 19:36:50 +0000
Subject: [PATCH] added tests for setup_logging function

---
 shared/utils/tests/setup_logging_test.py | 38 ++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 shared/utils/tests/setup_logging_test.py

diff --git a/shared/utils/tests/setup_logging_test.py b/shared/utils/tests/setup_logging_test.py
new file mode 100644
index 0000000..af05f32
--- /dev/null
+++ b/shared/utils/tests/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()