Skip to content

Commit ad49166

Browse files
committed
Corrected the token format.
1 parent 1ca6b30 commit ad49166

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/test/java/com/semanticprivacyguard/CustomPatternTest.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)