Add test for matching break types

mwilliamson · Jun 13, 2024 · 326d351 · 326d351
1 parent 9b1601c
commit 326d351
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/test/java/org/zwobble/mammoth/tests/conversion/DocumentToHtmlTests.java b/src/test/java/org/zwobble/mammoth/tests/conversion/DocumentToHtmlTests.java
@@ -250,10 +250,19 @@ public void breaksThatAreNotLineBreaksAreIgnored() {
     public void breaksCanBeMappedUsingStyleMappings() {
         assertThat(
             convertToHtml(
-                Break.PAGE_BREAK,
-                StyleMap.builder().mapBreak(BreakMatcher.PAGE_BREAK, HtmlPath.element("hr")).build()
+                run(withChildren(
+                    Break.PAGE_BREAK,
+                    Break.LINE_BREAK
+                )),
+                StyleMap.builder()
+                    .mapBreak(BreakMatcher.PAGE_BREAK, HtmlPath.element("hr"))
+                    .mapBreak(BreakMatcher.LINE_BREAK, HtmlPath.element("br", map("class", "line-break")))
+                    .build()
             ),
-            deepEquals(list(Html.element("hr")))
+            deepEquals(list(
+                Html.element("hr"),
+                Html.element("br", map("class", "line-break"))
+            ))
         );
     }