Need magic PII-free debugging method for API

tests/app/test_utils.py

@@ -13,7 +13,6 @@ from app.utils import (
     get_reference_from_personalisation,
     get_uuid_string_or_none,
     midnight_n_days_ago,
-    scrub,
 )
 
 
@@ -95,13 +94,6 @@ def test_get_public_notify_type_text():
     )
 
 
-def test_scrub():
-    result = scrub(
-        "This is a message with 17775554324, and also 18884449323 and also 17775554324"
-    )
-    assert result == "This is a message with 1XXXXX54324, and also 1XXXXX49323 and also 1XXXXX54324"
-
-
 # This method is used for simulating bulk sends.  We use localstack and run on a developer's machine to do the
 # simulation.  Please see docs->bulk_testing.md for instructions.
 # def test_generate_csv_for_bulk_testing():

tests/notifications_utils/test_logging.py

@@ -49,3 +49,20 @@ def test_base_json_formatter_contains_service_id():
         == "message to log"
     )
     assert service_id_filter.filter(record).service_id == "notify-admin"
+
+
+def test_scrub():
+    record = builtin_logging.LogRecord(
+        name="log thing",
+        level="info",
+        pathname="path",
+        lineno=123,
+        msg="phone1: 1555555555, phone2: 1555555554, email1: fake@fake.gov, email2: fake@fake2.fake.gov",
+        exc_info=None,
+        args=None,
+    )
+
+    assert (
+        json.loads(logging.PIIFormatter().format(record))["message"]
+        == "phone1: 1XXXXX55555, phone2: 1XXXXX55554, email1: XXXXXe@fake.gov, email2: XXXXX2.fake.gov"
+    )