Поиск SUBBYTES переписан на алгоритм Кнута-Моррриса-Пратта, добавлены тесты поиска паттернов

2020-04-16 02:08:30 +03:00
parent 07affe3812
commit 9eff4ecfe1
4 changed files with 161 additions and 10 deletions
--- a/src/test/java/ru/serega6531/packmate/PatternMatcherTest.java
+++ b/src/test/java/ru/serega6531/packmate/PatternMatcherTest.java
@@ -0,0 +1,85 @@
+package ru.serega6531.packmate;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.util.Assert;
+import ru.serega6531.packmate.model.FoundPattern;
+import ru.serega6531.packmate.model.Pattern;
+import ru.serega6531.packmate.model.enums.PatternSearchType;
+import ru.serega6531.packmate.service.PatternMatcher;
+
+import java.util.List;
+import java.util.Set;
+
+public class PatternMatcherTest {
+
+    @Test
+    public void testRegex() {
+        String content = "ahkfkyafceffek";
+
+        Set<FoundPattern> correctMatches = Set.of(
+                FoundPattern.builder()
+                        .startPosition(6)
+                        .endPosition(8)
+                        .build(),
+                FoundPattern.builder()
+                        .startPosition(9)
+                        .endPosition(11)
+                        .build());
+
+        final Pattern pattern = new Pattern();
+        pattern.setValue("[a-f]{3}");
+        pattern.setSearchType(PatternSearchType.REGEX);
+
+        final PatternMatcher matcher = new PatternMatcher(content.getBytes(), List.of(pattern));
+        final Set<FoundPattern> matches = matcher.findMatches();
+
+        Assert.isTrue(matches.equals(correctMatches), "Incorrect search: " + matches.toString());
+    }
+
+    @Test
+    public void testSubstring() {
+        String content = "abaabbaaabaabbbbbbbaaabaaa";
+        Set<FoundPattern> correctMatches = Set.of(
+                FoundPattern.builder()
+                        .startPosition(12)
+                        .endPosition(14)
+                        .build(),
+                FoundPattern.builder()
+                        .startPosition(15)
+                        .endPosition(17)
+                        .build());
+
+        final Pattern pattern = new Pattern();
+        pattern.setValue("bbb");
+        pattern.setSearchType(PatternSearchType.SUBSTRING);
+
+        final PatternMatcher matcher = new PatternMatcher(content.getBytes(), List.of(pattern));
+        final Set<FoundPattern> matches = matcher.findMatches();
+
+        Assert.isTrue(matches.equals(correctMatches), "Incorrect search: " + matches.toString());
+    }
+
+    @Test
+    public void testSubbytes() {
+        byte[] content = new byte[]{0x11, (byte) 0xAA, (byte) 0xAA, (byte) 0xAA, (byte) 0xAA, 0x22};
+        Set<FoundPattern> correctMatches = Set.of(
+                FoundPattern.builder()
+                        .startPosition(1)
+                        .endPosition(2)
+                        .build(),
+                FoundPattern.builder()
+                        .startPosition(3)
+                        .endPosition(4)
+                        .build());
+
+        final Pattern pattern = new Pattern();
+        pattern.setValue("AAaa");
+        pattern.setSearchType(PatternSearchType.SUBBYTES);
+
+        final PatternMatcher matcher = new PatternMatcher(content, List.of(pattern));
+        final Set<FoundPattern> matches = matcher.findMatches();
+
+        Assert.isTrue(matches.equals(correctMatches), "Incorrect search: " + matches.toString());
+    }
+
+}