ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter3/log-sanitization/tests/test_null_pii_values.py

@@ -0,0 +1,33 @@
+"""detect_pii must treat JSON null pii_values like an empty list."""
+import json
+from unittest.mock import patch
+
+from agent import LogSanitizationAgent
+
+
+def test_null_pii_values_returns_empty_list():
+    agent = object.__new__(LogSanitizationAgent)
+    agent.count_tokens = lambda text: len(text) // 4
+    agent._chat_stream = lambda messages: iter(
+        [json.dumps({"pii_values": None})]
+    )
+
+    with patch("builtins.print"):
+        pii_values, metrics = agent.detect_pii("Alice phone 555-0100")
+
+    assert pii_values == []
+    assert metrics["pii_items_found"] == 0
+
+
+def test_list_pii_values_still_cleaned():
+    agent = object.__new__(LogSanitizationAgent)
+    agent.count_tokens = lambda text: len(text) // 4
+    agent._chat_stream = lambda messages: iter(
+        [json.dumps({"pii_values": ["  Alice  ", "-", "Bob"]})]
+    )
+
+    with patch("builtins.print"):
+        pii_values, metrics = agent.detect_pii("text")
+
+    assert pii_values == ["Alice", "Bob"]
+    assert metrics["pii_items_found"] == 2