diff --git a/tests/unit/test_logging_securityfilter.py b/tests/unit/test_logging_securityfilter.py index d73e89bf..7b216680 100644 --- a/tests/unit/test_logging_securityfilter.py +++ b/tests/unit/test_logging_securityfilter.py @@ -6,6 +6,7 @@ sensitive information from log messages without causing false positives. """ import logging + import pytest from routstr.core.logging import SecurityFilter @@ -20,6 +21,7 @@ def security_filter() -> SecurityFilter: @pytest.fixture def filter_message(security_filter: SecurityFilter): """A helper fixture to apply the filter to a message string.""" + def _filter(msg: str) -> str: record = logging.LogRecord( name="test_logger", @@ -32,6 +34,7 @@ def filter_message(security_filter: SecurityFilter): ) security_filter.filter(record) return record.getMessage() + return _filter @@ -45,7 +48,7 @@ def test_redacts_unquoted_key_value_pairs(filter_message) -> None: def test_redacts_quoted_key_value_pairs(filter_message) -> None: """Test that a quoted token is correctly redacted.""" original = 'User authenticated with token="cashuA123abc"' - expected = 'User authenticated with token: [REDACTED]' + expected = "User authenticated with token: [REDACTED]" assert filter_message(original) == expected @@ -109,4 +112,4 @@ def test_is_case_insensitive_for_standalone(filter_message) -> None: """Test that standalone matching is case-insensitive.""" original = "Using NSEC1a8d9f8s7d9f8a7s6d5f4a3s2d1f9a8s7d6f5a4s3d2f1a9s8d7f6a5s4d3f and CaShuA123abc" expected = "Using [REDACTED] and [REDACTED]" - assert filter_message(original) == expected \ No newline at end of file + assert filter_message(original) == expected