refactor custom logger and add tests

4 files changed, 43 insertions, 1 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
index fcf4189..58416ea 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,7 +1,30 @@
+import sys
+from typing import Callable
 import pytest
+from logging import getLogger, DEBUG
+
 from pyssg.arg_parser import get_parser
+from pyssg.custom_logger import setup_logger
 
 
 @pytest.fixture(scope='session')
 def arg_parser():
     return get_parser()
+
+
+@pytest.fixture(scope='session')
+def logger():
+    setup_logger(__name__, DEBUG)
+    return getLogger(__name__)
+
+
+@pytest.fixture
+def capture_stdout(monkeypatch: Callable) -> dict[str, str | int]:
+    buffer: dict[str, str | int] = {'stdout': '', 'write_calls': 0}
+
+    def fake_writer(s):
+        buffer['stdout'] += s
+        buffer['write_calls'] += 1  # type: ignore
+
+    monkeypatch.setattr(sys.stdout, 'write', fake_writer)
+    return buffer
diff --git a/tests/io_files/__init__.py b/tests/io_files/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/io_files/__init__.py
diff --git a/tests/test_arg_parser.py b/tests/test_arg_parser.py
index 1a5b48c..e233e93 100644
--- a/tests/test_arg_parser.py
+++ b/tests/test_arg_parser.py
@@ -31,7 +31,6 @@ def test_valid_args(args: list[str],
 @pytest.mark.parametrize('args', [
     (['--something-random']),
     (['-z']),
-    (['hello']),
     (['help']),
     (['h'])
 ])
diff --git a/tests/test_custom_logger.py b/tests/test_custom_logger.py
new file mode 100644
index 0000000..1062e41
--- /dev/null
+++ b/tests/test_custom_logger.py
@@ -0,0 +1,20 @@
+import pytest
+from logging import DEBUG, INFO, WARNING, ERROR, CRITICAL
+from typing import Callable
+
+
+@pytest.mark.parametrize('log_level, starts_with, message', [
+    (DEBUG, '[DEBUG]', 'first message'),
+    (INFO, 'second message', 'second message'),
+    (WARNING, '\x1b[33m[WARNING]', 'third message'),
+    (ERROR, '\x1b[31m[ERROR]', 'fourth message'),
+    (CRITICAL, '\x1b[31;1m[CRITICAL]', 'fifth message'),
+])
+def test_log_levels(log_level: int,
+                    starts_with: str,
+                    message: str,
+                    logger: Callable,
+                    capture_stdout: dict[str, str | int]) -> None:
+    logger.log(log_level, message)
+    assert str(capture_stdout['stdout']).startswith(starts_with)
+    assert message in str(capture_stdout['stdout'])