1 files changed, 38 insertions, 0 deletions
diff --git a/src/test/java/org/zwobble/mammoth/tests/styles/HighlightMatcherTests.java b/src/test/java/org/zwobble/mammoth/tests/styles/HighlightMatcherTests.java
new file mode 100644
index 0000000..4714f9a
--- /dev/null
+++ b/src/test/java/org/zwobble/mammoth/tests/styles/HighlightMatcherTests.java
@@ -0,0 +1,38 @@
+package org.zwobble.mammoth.tests.styles;
+
+import org.junit.jupiter.api.Test;
+import org.zwobble.mammoth.internal.styles.HighlightMatcher;
+
+import java.util.Optional;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.equalTo;
+
+public class HighlightMatcherTests {
+    @Test
+    public void highlightMatcherWithoutColorMatchesAllHighlightElements() {
+        HighlightMatcher matcher = new HighlightMatcher(Optional.empty());
+
+        boolean result = matcher.matches("yellow");
+
+        assertThat(result, equalTo(true));
+    }
+
+    @Test
+    public void highlightMatcherWithColorMatchesHighlightWithThatColor() {
+        HighlightMatcher matcher = new HighlightMatcher(Optional.of("yellow"));
+
+        boolean result = matcher.matches("yellow");
+
+        assertThat(result, equalTo(true));
+    }
+
+    @Test
+    public void highlightMatcherWithColorDoesNotMatchHighlightsWithOtherColors() {
+        HighlightMatcher matcher = new HighlightMatcher(Optional.of("yellow"));
+
+        boolean result = matcher.matches("red");
+
+        assertThat(result, equalTo(false));
+    }
+}