Add basic tests for DocumentConverter

2 files changed, 49 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/DocumentConverterTests.java b/src/test/java/org/zwobble/mammoth/tests/DocumentConverterTests.java
new file mode 100644
index 0000000..6152822
--- /dev/null
+++ b/src/test/java/org/zwobble/mammoth/tests/DocumentConverterTests.java
@@ -0,0 +1,42 @@
+package org.zwobble.mammoth.tests;
+
+import org.junit.Test;
+import org.zwobble.mammoth.DocumentConverter;
+import org.zwobble.mammoth.documents.Document;
+import org.zwobble.mammoth.documents.DocumentElement;
+import org.zwobble.mammoth.html.Html;
+import org.zwobble.mammoth.html.HtmlNode;
+
+import java.util.List;
+
+import static com.natpryce.makeiteasy.MakeItEasy.*;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.zwobble.mammoth.tests.DeepReflectionMatcher.deepEquals;
+import static org.zwobble.mammoth.tests.documents.DocumentElementMakers.*;
+import static org.zwobble.mammoth.util.MammothLists.list;
+
+public class DocumentConverterTests {
+    @Test
+    public void plainParagraphIsConvertedToPlainParagraph() {
+        assertThat(
+            convertToHtml(make(a(PARAGRAPH, with(CHILDREN, list(runWithText("Hello")))))),
+            deepEquals(list(Html.element("p", list(Html.text("Hello"))))));
+    }
+
+    @Test
+    public void multipleParagraphsInDocumentAreConvertedToMultipleParagraphs() {
+        assertThat(
+            DocumentConverter.convertToHtml(new Document(list(
+                make(a(PARAGRAPH, with(CHILDREN, list(runWithText("Hello"))))),
+                make(a(PARAGRAPH, with(CHILDREN, list(runWithText("there")))))))),
+
+            deepEquals(list(
+                Html.element("p", list(Html.text("Hello"))),
+                Html.element("p", list(Html.text("there"))))));
+    }
+
+
+    private List<HtmlNode> convertToHtml(DocumentElement element) {
+        return DocumentConverter.convertToHtml(element);
+    }
+}
diff --git a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
index 890a68e..396a95a 100644
--- a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
+++ b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
@@ -7,6 +7,9 @@ import org.zwobble.mammoth.documents.*;
 import java.util.List;
 import java.util.Optional;
 
+import static com.natpryce.makeiteasy.MakeItEasy.a;
+import static com.natpryce.makeiteasy.MakeItEasy.make;
+import static com.natpryce.makeiteasy.MakeItEasy.with;
 import static com.natpryce.makeiteasy.Property.newProperty;
 import static org.zwobble.mammoth.util.MammothLists.list;
 
@@ -28,4 +31,8 @@ public class DocumentElementMakers {
             VerticalAlignment.BASELINE,
             Optional.empty(),
             propertyLookup.valueOf(CHILDREN, list()));
+
+    public static Run runWithText(String text) {
+        return make(a(RUN, with(CHILDREN, list(new Text(text)))));
+    }
 }