Allow HTML elements to have a choice of tag names

1 files changed, 17 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/html/HtmlCollapseTests.java b/src/test/java/org/zwobble/mammoth/tests/html/HtmlCollapseTests.java
index 8a25385..dfe4f78 100644
--- a/src/test/java/org/zwobble/mammoth/tests/html/HtmlCollapseTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/html/HtmlCollapseTests.java
@@ -78,4 +78,21 @@ public class HtmlCollapseTests {
             Html.collapse(list(Html.element("p"), Html.collapsibleElement("p"))),
             deepEquals(list(Html.element("p"))));
     }
+
+    @Test
+    public void elementWithChoiceOfTagNamesCanCollapseIntoPreviousElementIfItHasOneOfThoseTagNamesAsItsMainTagName() {
+        assertThat(
+            Html.collapse(list(
+                Html.collapsibleElement("ol"),
+                Html.collapsibleElement(list("ul", "ol")))),
+            deepEquals(list(Html.collapsibleElement("ol"))));
+
+        assertThat(
+            Html.collapse(list(
+                Html.collapsibleElement(list("ul", "ol")),
+                Html.collapsibleElement("ol"))),
+            deepEquals(list(
+                Html.collapsibleElement(list("ul", "ol")),
+                Html.collapsibleElement("ol"))));
+    }
 }