@@ -14,23 +14,29 @@ def test_email_redaction(self):
1414 """Test that email addresses are redacted from log messages."""
1515 record = {"message" : "User email is test@example.com" , "exception" : None }
1616 scrub_sensitive_data (record )
17- assert "test@example.com" not in record ["message" ]
18- assert "{{EMAIL}}" in record ["message" ]
17+ message = record ["message" ]
18+ assert isinstance (message , str )
19+ assert "test@example.com" not in message
20+ assert "{{EMAIL}}" in message
1921
2022 def test_phone_redaction (self ):
2123 """Test that phone numbers are redacted (new capability via scrubadub)."""
2224 record = {"message" : "Call me at 1-800-555-0199" , "exception" : None }
2325 scrub_sensitive_data (record )
24- assert "1-800-555-0199" not in record ["message" ]
25- assert "{{PHONE}}" in record ["message" ]
26+ message = record ["message" ]
27+ assert isinstance (message , str )
28+ assert "1-800-555-0199" not in message
29+ assert "{{PHONE}}" in message
2630
2731 def test_api_key_redaction (self ):
2832 """Test that OpenAI API keys are redacted from log messages."""
2933 api_key = "sk-abc123def456ghi789jkl012mno345pqr678stu901"
3034 record = {"message" : f"Using key: { api_key } " , "exception" : None }
3135 scrub_sensitive_data (record )
32- assert api_key not in record ["message" ]
33- assert "[REDACTED_API_KEY]" in record ["message" ]
36+ message = record ["message" ]
37+ assert isinstance (message , str )
38+ assert api_key not in message
39+ assert "[REDACTED_API_KEY]" in message
3440
3541 def test_multiple_redactions (self ):
3642 """Test redacting multiple sensitive items in a single message."""
@@ -39,10 +45,12 @@ def test_multiple_redactions(self):
3945 "exception" : None ,
4046 }
4147 scrub_sensitive_data (record )
42- assert "{{EMAIL}}" in record ["message" ]
43- assert "[REDACTED_API_KEY]" in record ["message" ]
44- assert "test@example.com" not in record ["message" ]
45- assert "sk-123456789012345678901234" not in record ["message" ]
48+ message = record ["message" ]
49+ assert isinstance (message , str )
50+ assert "{{EMAIL}}" in message
51+ assert "[REDACTED_API_KEY]" in message
52+ assert "test@example.com" not in message
53+ assert "sk-123456789012345678901234" not in message
4654
4755 def test_exception_message_redaction (self ):
4856 """Test that PII is redacted from exception messages."""
@@ -87,8 +95,10 @@ def test_anthropic_api_key_redaction(self):
8795 api_key = "sk-ant-api03-abc123def456ghi789jkl012mno345pqr678"
8896 record = {"message" : f"Using Anthropic key: { api_key } " , "exception" : None }
8997 scrub_sensitive_data (record )
90- assert api_key not in record ["message" ]
91- assert "[REDACTED_API_KEY]" in record ["message" ]
98+ message = record ["message" ]
99+ assert isinstance (message , str )
100+ assert api_key not in message
101+ assert "[REDACTED_API_KEY]" in message
92102
93103 def test_stripe_api_key_redaction (self ):
94104 """Test that Stripe API keys are redacted."""
@@ -100,16 +110,20 @@ def test_stripe_api_key_redaction(self):
100110 key = prefix + suffix
101111 record = {"message" : f"Stripe key: { key } " , "exception" : None }
102112 scrub_sensitive_data (record )
103- assert key not in record ["message" ]
104- assert "[REDACTED_API_KEY]" in record ["message" ]
113+ message = record ["message" ]
114+ assert isinstance (message , str )
115+ assert key not in message
116+ assert "[REDACTED_API_KEY]" in message
105117
106118 def test_bearer_token_redaction (self ):
107119 """Test that Authorization Bearer tokens are redacted."""
108120 token = "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkw"
109121 record = {"message" : f"Authorization: { token } " , "exception" : None }
110122 scrub_sensitive_data (record )
111- assert "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9" not in record ["message" ]
112- assert "[REDACTED_BEARER_TOKEN]" in record ["message" ]
123+ message = record ["message" ]
124+ assert isinstance (message , str )
125+ assert "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9" not in message
126+ assert "[REDACTED_BEARER_TOKEN]" in message
113127
114128 def test_generic_api_key_redaction (self ):
115129 """Test that generic api_key patterns are redacted."""
@@ -123,5 +137,7 @@ def test_generic_api_key_redaction(self):
123137 for pattern in patterns :
124138 record = {"message" : f"Config: { pattern } " , "exception" : None }
125139 scrub_sensitive_data (record )
126- assert "abc123def456ghi789jkl012" not in record ["message" ]
127- assert "[REDACTED_KEY]" in record ["message" ]
140+ message = record ["message" ]
141+ assert isinstance (message , str )
142+ assert "abc123def456ghi789jkl012" not in message
143+ assert "[REDACTED_KEY]" in message
0 commit comments