@@ -117,8 +117,8 @@ void customPattern_employeeId_detected() {
117117
118118 RedactionResult result = spg.redact("Assigned to EMP-042731 for review.");
119119
120- assertTrue(result.getRedactedText().contains("[GENERIC_PII_1 ]"),
121- "Employee ID should be tokenised to [GENERIC_PII_1 ]");
120+ assertTrue(result.getRedactedText().contains("[PII_1 ]"),
121+ "Employee ID should be tokenised to [PII_1 ]");
122122 assertFalse(result.getRedactedText().contains("EMP-042731"),
123123 "Original employee ID should be removed");
124124 assertEquals(1, result.getMatchCount());
@@ -135,9 +135,9 @@ void customPattern_multipleMatches_numberedSequentially() {
135135 "Reviewers: EMP-000001, EMP-000002, EMP-000003.");
136136
137137 String redacted = result.getRedactedText();
138- assertTrue(redacted.contains("[GENERIC_PII_1 ]"), "First employee ID missing");
139- assertTrue(redacted.contains("[GENERIC_PII_2 ]"), "Second employee ID missing");
140- assertTrue(redacted.contains("[GENERIC_PII_3 ]"), "Third employee ID missing");
138+ assertTrue(redacted.contains("[PII_1 ]"), "First employee ID missing");
139+ assertTrue(redacted.contains("[PII_2 ]"), "Second employee ID missing");
140+ assertTrue(redacted.contains("[PII_3 ]"), "Third employee ID missing");
141141 assertEquals(3, result.getMatchCount());
142142 }
143143
0 commit comments