Read defaults from content types XML

1 files changed, 27 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/docx/ContentTypesXmlTests.java b/src/test/java/org/zwobble/mammoth/tests/docx/ContentTypesXmlTests.java
new file mode 100644
index 0000000..6b46456
--- /dev/null
+++ b/src/test/java/org/zwobble/mammoth/tests/docx/ContentTypesXmlTests.java
@@ -0,0 +1,27 @@
+package org.zwobble.mammoth.tests.docx;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import org.junit.Test;
+import org.zwobble.mammoth.docx.ContentTypes;
+import org.zwobble.mammoth.xml.XmlElement;
+
+import java.util.Optional;
+
+import static org.junit.Assert.assertEquals;
+import static org.zwobble.mammoth.docx.ContentTypesXml.readContentTypesXmlElement;
+import static org.zwobble.mammoth.xml.XmlNodes.element;
+
+public class ContentTypesXmlTests {
+    @Test
+    public void contentTypesBasedOnDefaultForExtensionIfThereIsNoOverride() {
+        XmlElement element = element("content-types:Types", ImmutableList.of(
+            element("content-types:Default", ImmutableMap.of(
+                "Extension", "png",
+                "ContentType", "image/png"))));
+        ContentTypes contentTypes = readContentTypesXmlElement(element);
+        assertEquals(
+            Optional.of("image/png"),
+            contentTypes.findContentType("word/media/hat.png"));
+    }
+}