Support content type overrides

1 files changed, 16 insertions, 1 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
index 6b46456..573103b 100644
--- a/src/test/java/org/zwobble/mammoth/tests/docx/ContentTypesXmlTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/docx/ContentTypesXmlTests.java
@@ -14,7 +14,7 @@ import static org.zwobble.mammoth.xml.XmlNodes.element;
 
 public class ContentTypesXmlTests {
     @Test
-    public void contentTypesBasedOnDefaultForExtensionIfThereIsNoOverride() {
+    public void contentTypeIsBasedOnDefaultForExtensionIfThereIsNoOverride() {
         XmlElement element = element("content-types:Types", ImmutableList.of(
             element("content-types:Default", ImmutableMap.of(
                 "Extension", "png",
@@ -24,4 +24,19 @@ public class ContentTypesXmlTests {
             Optional.of("image/png"),
             contentTypes.findContentType("word/media/hat.png"));
     }
+
+    @Test
+    public void contentTypeIsBasedOnOverrideIfPresent() {
+        XmlElement element = element("content-types:Types", ImmutableList.of(
+            element("content-types:Default", ImmutableMap.of(
+                "Extension", "png",
+                "ContentType", "image/png")),
+            element("content-types:Override", ImmutableMap.of(
+                "PartName", "/word/media/hat.png",
+                "ContentType", "image/hat"))));
+        ContentTypes contentTypes = readContentTypesXmlElement(element);
+        assertEquals(
+            Optional.of("image/hat"),
+            contentTypes.findContentType("word/media/hat.png"));
+    }
 }