Add test for matching break types

1 files changed, 12 insertions, 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
index ea60595..03e5a52 100644
--- 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 class DocumentToHtmlTests {
     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"))
+            ))
         );
     }