1616
1717import logging
1818
19- from met_api .utils .logging_masker import (
20- SensitiveDataFilter ,
21- mask_dict ,
22- mask_url ,
23- )
19+ from met_api .utils .logging_masker import SensitiveDataFilter , mask_dict , mask_url
2420
2521
2622def test_logging_masker ():
@@ -32,8 +28,7 @@ def test_logging_masker():
3228 assert 'user' not in masked_url
3329 # Count occurrences of redaction to ensure we are masking both username and password
3430 assert masked_url .count ('***REDACTED***' ) == 2
35-
36-
31+
3732 # Test dictionary masking with various sensitive keys
3833 data = {
3934 'username' : 'admin' ,
@@ -52,10 +47,10 @@ def test_logging_masker():
5247 assert masked_data ['config' ]['secret' ] == '***REDACTED***'
5348 assert masked_data ['username' ] == 'admin'
5449 assert masked_data ['email' ] == 'test@example.com'
55-
50+
5651 # Test SensitiveDataFilter with various patterns
5752 filter_obj = SensitiveDataFilter ()
58-
53+
5954 # Test SQLAlchemy URL
6055 log_record = logging .LogRecord (
6156 name = 'test' , level = logging .INFO , pathname = '' , lineno = 0 ,
@@ -65,7 +60,7 @@ def test_logging_masker():
6560 filter_obj .filter (log_record )
6661 assert 'complexP@ss123' not in str (log_record .msg )
6762 assert '***REDACTED***' in str (log_record .msg )
68-
63+
6964 # Test JWT token
7065 jwt_token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0In0.abc123'
7166 log_record = logging .LogRecord (
@@ -76,7 +71,7 @@ def test_logging_masker():
7671 filter_obj .filter (log_record )
7772 assert jwt_token not in str (log_record .msg )
7873 assert '***REDACTED_JWT***' in str (log_record .msg )
79-
74+
8075 # Test normal content preservation
8176 normal_message = 'Processing user request for engagement ID 123'
8277 log_record = logging .LogRecord (
0 commit comments