Add missing document matcher parser tests

mwilliamson · Jun 13, 2024 · 43f5e18 · 43f5e18
1 parent a26168d
commit 43f5e18
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/src/test/java/org/zwobble/mammoth/tests/styles/parsing/DocumentMatcherParsingTests.java b/src/test/java/org/zwobble/mammoth/tests/styles/parsing/DocumentMatcherParsingTests.java
@@ -173,6 +173,50 @@ public void readsHighlightWithColor() {
         );
     }
 
+    @Test
+    public void readsCommentReference() {
+        assertThat(
+            parseDocumentMatcher("comment-reference"),
+            deepEquals(new StyleMapBuilder().commentReference(HTML_PATH).build())
+        );
+    }
+
+    @Test
+    public void readsLineBreaks() {
+        assertThat(
+            parseDocumentMatcher("br[type='line']"),
+            deepEquals(
+                new StyleMapBuilder()
+                    .mapBreak(BreakMatcher.LINE_BREAK, HTML_PATH)
+                    .build()
+            )
+        );
+    }
+
+    @Test
+    public void readsPageBreaks() {
+        assertThat(
+            parseDocumentMatcher("br[type='page']"),
+            deepEquals(
+                new StyleMapBuilder()
+                    .mapBreak(BreakMatcher.PAGE_BREAK, HTML_PATH)
+                    .build()
+            )
+        );
+    }
+
+    @Test
+    public void readsColumnBreaks() {
+        assertThat(
+            parseDocumentMatcher("br[type='column']"),
+            deepEquals(
+                new StyleMapBuilder()
+                    .mapBreak(BreakMatcher.COLUMN_BREAK, HTML_PATH)
+                    .build()
+            )
+        );
+    }
+
     private static final HtmlPath HTML_PATH = HtmlPath.element("placeholder");
 
     private StyleMap parseDocumentMatcher(String input) {