From e28fbb181851ca16bc0ade9c371628f86c25adbc Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <david@luevano.xyz>
Date: Sun, 19 Feb 2023 22:59:32 -0600
Subject: refactor custom logger and add tests

---
 tests/conftest.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'tests/conftest.py')

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
-- 
cgit v1.2.3-54-g00ecf