From ee5e1389542a4306b380a32ce58b0c88d64b98d0 Mon Sep 17 00:00:00 2001
From: "M. Eren Akbiyik" <erenakbiyik@gmail.com>
Date: Mon, 28 Aug 2023 23:04:33 +0200
Subject: [PATCH] test: logging

---
 tests/test_logging.py | 48 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 tests/test_logging.py

diff --git a/tests/test_logging.py b/tests/test_logging.py
new file mode 100644
index 0000000..dc6968c
--- /dev/null
+++ b/tests/test_logging.py
@@ -0,0 +1,48 @@
+"""Tests for the logging module."""
+import logging
+import os
+import sys
+
+from torchcache.logging import set_logger_config
+
+
+def test_set_logger_config() -> None:
+    """Test the set_logger_config function."""
+    set_logger_config()
+    assert logging.getLogger().level == logging.WARN
+    assert len(logging.getLogger().handlers) == 1
+    assert isinstance(logging.getLogger().handlers[0], logging.StreamHandler)
+    assert logging.getLogger().handlers[0].stream == sys.stdout
+    assert (
+        logging.getLogger().handlers[0].formatter._fmt
+        == "[torchcache] - %(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    )
+    assert logging.getLogger().handlers[0].formatter.datefmt == "%Y-%m-%d %H:%M:%S"
+    # Also test various environment variable configurations
+    os.environ["TORCHCACHE_LOG_LEVEL"] = "DEBUG"
+    os.environ[
+        "TORCHCACHE_LOG_FMT"
+    ] = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    os.environ["TORCHCACHE_LOG_DATEFMT"] = "%Y-%m-%d %H:%M:%S"
+    os.environ["TORCHCACHE_LOG_FILE"] = "test.log"
+    set_logger_config()
+    assert logging.getLogger().level == logging.DEBUG
+    assert len(logging.getLogger().handlers) == 2
+    assert isinstance(logging.getLogger().handlers[0], logging.StreamHandler)
+    assert logging.getLogger().handlers[0].stream == sys.stdout
+    assert (
+        logging.getLogger().handlers[0].formatter._fmt
+        == "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    )
+    assert logging.getLogger().handlers[0].formatter.datefmt == "%Y-%m-%d %H:%M:%S"
+    assert isinstance(logging.getLogger().handlers[1], logging.FileHandler)
+    assert (
+        logging.getLogger().handlers[1].formatter._fmt
+        == "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    )
+    assert logging.getLogger().handlers[1].formatter.datefmt == "%Y-%m-%d %H:%M:%S"
+    assert logging.getLogger().handlers[1].baseFilename.endswith("test.log")
+    # Test invalid log level
+    os.environ["TORCHCACHE_LOG_LEVEL"] = "INVALID"
+    set_logger_config()
+    assert logging.getLogger().level == logging.WARN