PD Guard: модуль безопасности персональных данных

2026-09-22 17:43:33 +03:00
commit 41052fac6a
80 changed files with 91230 additions and 0 deletions
@@ -0,0 +1,86 @@
+package ru.pdguard;
+
+import org.junit.jupiter.api.Test;
+import ru.pdguard.config.SystemPolicy;
+import ru.pdguard.core.PayloadStore;
+import ru.pdguard.core.Pipeline;
+import ru.pdguard.detect.RuleRegistry;
+import ru.pdguard.detect.Validators;
+import ru.pdguard.mask.MaskMode;
+import ru.pdguard.mask.Masker;
+
+import java.util.Set;
+import java.util.UUID;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+/** Виды замены: звёздочки, токены, правдоподобные значения. */
+class MaskModeTest {
+
+    private final Pipeline pipeline =
+            new Pipeline(new RuleRegistry(), new Masker(), new PayloadStore(1_000_000L, 30));
+
+    private SystemPolicy policy(MaskMode mode) {
+        return new SystemPolicy(SystemPolicy.DEFAULT_NAME, true, true, mode,
+                Set.of(SystemPolicy.ALL), SystemPolicy.DEFAULT.requireCompanion(), null);
+    }
+
+    private String mask(MaskMode mode, String text) {
+        return pipeline.process(text, UUID.randomUUID().toString(), policy(mode));
+    }
+
+    @Test
+    void tokenModeNumbersEachType() {
+        String masked = mask(MaskMode.TOKEN, "Клиент Иванов Иван Иванович, почта ivan@mail.ru");
+        assertTrue(masked.contains("[FIO_1]"), masked);
+        assertTrue(masked.contains("[EMAIL_1]"), masked);
+    }
+
+    @Test
+    void sameValueGetsSameTokenWithinRequest() {
+        String masked = mask(MaskMode.TOKEN, "ivan@mail.ru и ещё раз ivan@mail.ru, а также petr@mail.ru");
+        assertEquals(2, count(masked, "[EMAIL_1]"), masked);
+        assertEquals(1, count(masked, "[EMAIL_2]"), masked);
+    }
+
+    @Test
+    void syntheticModeProducesPlausibleValues() {
+        String masked = mask(MaskMode.SYNTHETIC, "Карта 4111 1111 1111 1111 клиента Иванова Ивана");
+
+        assertFalse(masked.contains("4111 1111 1111 1111"), masked);
+        assertFalse(masked.contains("*"), "подстановка должна выглядеть настоящей: " + masked);
+
+        Matcher card = Pattern.compile("\\d{4} \\d{4} \\d{4} \\d{4}").matcher(masked);
+        assertTrue(card.find(), masked);
+        assertTrue(Validators.luhn(card.group()), "подставленный номер карты обязан проходить проверку Луна");
+    }
+
+    @Test
+    void syntheticValuesAreStable() {
+        String text = "Почта ivan@mail.ru, паспорт 4509 123456";
+        assertEquals(mask(MaskMode.SYNTHETIC, text), mask(MaskMode.SYNTHETIC, text));
+    }
+
+    @Test
+    void unmaskingWorksInEveryMode() {
+        String original = "Клиент Иванов Иван Иванович, карта 4111 1111 1111 1111, почта ivan@mail.ru";
+        for (MaskMode mode : MaskMode.values()) {
+            String id = "mode-" + mode;
+            String masked = pipeline.process(original, id, policy(mode));
+            assertFalse(masked.contains("Иванов Иван Иванович"), mode + ": " + masked);
+            assertEquals(original, pipeline.process(masked, id, policy(mode)), mode.name());
+        }
+    }
+
+    private static int count(String text, String fragment) {
+        int n = 0;
+        for (int i = text.indexOf(fragment); i >= 0; i = text.indexOf(fragment, i + fragment.length())) {
+            n++;
+        }
+        return n;
+    }
+}