test: 全面迁移 pytest 并加入格式门禁

tests/test_logger.py

@@ -3,23 +3,28 @@
 import gzip
 import logging
 import tempfile
-import unittest
 from datetime import datetime, timedelta
 from pathlib import Path
 
-from utils.logger import _SensitiveDataFilter, _SizeAndDayRotatingFileHandler, _parse_size
+from utils.logger import (
+    _SensitiveDataFilter,
+    _SizeAndDayRotatingFileHandler,
+    _parse_size,
+)
 
 
-class LoggerTests(unittest.TestCase):
+class TestLogger:
     def test_size_parser_accepts_human_readable_values(self):
-        self.assertEqual(_parse_size("2K"), 2 * 1024)
-        self.assertEqual(_parse_size("3MiB"), 3 * 1024 * 1024)
-        self.assertEqual(_parse_size("invalid", default=123), 123)
+        assert _parse_size("2K") == 2 * 1024
+        assert _parse_size("3MiB") == 3 * 1024 * 1024
+        assert _parse_size("invalid", default=123) == 123
 
     def test_file_log_redacts_credentials_and_compresses_rotation(self):
         with tempfile.TemporaryDirectory() as tmpdir:
             log_path = Path(tmpdir) / "app-2026-08-28.log"
-            handler = _SizeAndDayRotatingFileHandler(log_path, max_bytes=1, retention_days=14)
+            handler = _SizeAndDayRotatingFileHandler(
+                log_path, max_bytes=1, retention_days=14
+            )
             handler.setFormatter(logging.Formatter("%(message)s"))
             handler.addFilter(_SensitiveDataFilter())
             test_logger = logging.getLogger("tests.logger.redaction")
@@ -34,20 +39,24 @@ class LoggerTests(unittest.TestCase):
                 test_logger.handlers = []
 
             archives = list(Path(tmpdir).glob("app-2026-08-28.log.*.gz"))
-            self.assertEqual(len(archives), 1)
+            assert len(archives) == 1
             archived_content = gzip.open(archives[0], "rt", encoding="utf-8").read()
             current_content = log_path.read_text(encoding="utf-8")
             combined = archived_content + current_content
-            self.assertNotIn("super-secret", combined)
-            self.assertNotIn("top-secret-token", combined)
-            self.assertIn("Cookie: [REDACTED]", combined)
-            self.assertIn("Authorization: [REDACTED]", combined)
+            assert "super-secret" not in combined
+            assert "top-secret-token" not in combined
+            assert "Cookie: [REDACTED]" in combined
+            assert "Authorization: [REDACTED]" in combined
 
     def test_daily_log_switches_to_a_new_dated_file(self):
         with tempfile.TemporaryDirectory() as tmpdir:
             today = datetime.now().date()
-            old_path = Path(tmpdir) / f"app-{(today - timedelta(days=1)).isoformat()}.log"
-            handler = _SizeAndDayRotatingFileHandler(old_path, max_bytes=1024, retention_days=14)
+            old_path = (
+                Path(tmpdir) / f"app-{(today - timedelta(days=1)).isoformat()}.log"
+            )
+            handler = _SizeAndDayRotatingFileHandler(
+                old_path, max_bytes=1024, retention_days=14
+            )
             handler.setFormatter(logging.Formatter("%(message)s"))
             handler._active_day = today - timedelta(days=1)
             test_logger = logging.getLogger("tests.logger.daily")
@@ -61,9 +70,5 @@ class LoggerTests(unittest.TestCase):
                 test_logger.handlers = []
 
             current_path = Path(tmpdir) / f"app-{today.isoformat()}.log"
-            self.assertTrue(current_path.exists())
-            self.assertEqual(current_path.read_text(encoding="utf-8").strip(), "today")
-
-
-if __name__ == "__main__":
-    unittest.main()
+            assert current_path.exists()
+            assert current_path.read_text(encoding="utf-8").strip() == "today"