From ebaa77940eb7a263779dcac80d38a4cecb42e97b Mon Sep 17 00:00:00 2001
From: jbannon <jbann1994@gmail.com>
Date: Tue, 7 Jun 2022 07:33:48 +0000
Subject: [PATCH] more unit tests

---
 tests/unit/utils/test_logger.py | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/tests/unit/utils/test_logger.py b/tests/unit/utils/test_logger.py
index 4bdde4cc..e7a5fe51 100644
--- a/tests/unit/utils/test_logger.py
+++ b/tests/unit/utils/test_logger.py
@@ -8,12 +8,29 @@ from ytdl_sub.utils.logger import LoggerLevels
 
 @pytest.fixture(autouse=True)
 def cleanup_debug_file():
+    Logger.set_log_level(log_level_name=LoggerLevels.DEBUG.name)
     yield
-
     Logger.cleanup(delete_debug_file=True)
 
 
 class TestLogger:
+    @pytest.mark.parametrize(
+        "log_level",
+        [
+            LoggerLevels.QUIET,
+            LoggerLevels.INFO,
+            LoggerLevels.VERBOSE,
+            LoggerLevels.DEBUG,
+        ],
+    )
+    def test_logger_level_set(self, log_level):
+        Logger.set_log_level(log_level_name=log_level.name)
+        assert Logger._LOGGER_LEVEL == log_level
+
+    def test_logger_level_invalid_name(self):
+        with pytest.raises(ValueError):
+            Logger.set_log_level("nope")
+
     @pytest.mark.parametrize(
         "log_level, outputs_to_stdout",
         [