Read relationship type

2 files changed, 52 insertions, 4 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java b/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java
index 690ae98..9a15676 100644
--- a/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java
@@ -1427,10 +1427,10 @@ public class BodyXmlTests {
     }
 
     private static Relationship hyperlinkRelationship(String relationshipId, String target) {
-        return new Relationship(relationshipId, target);
+        return new Relationship(relationshipId, target, "http://schemas.openxmlformats.org/officeDocument/2006/relationships/hyperlink");
     }
 
     private static Relationship imageRelationship(String relationshipId, String target) {
-        return new Relationship(relationshipId, target);
+        return new Relationship(relationshipId, target, "http://schemas.openxmlformats.org/officeDocument/2006/relationships/image");
     }
 }
diff --git a/src/test/java/org/zwobble/mammoth/tests/docx/RelationshipsXmlTests.java b/src/test/java/org/zwobble/mammoth/tests/docx/RelationshipsXmlTests.java
index fb268b5..e896dd1 100644
--- a/src/test/java/org/zwobble/mammoth/tests/docx/RelationshipsXmlTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/docx/RelationshipsXmlTests.java
@@ -12,15 +12,63 @@ import static org.zwobble.mammoth.internal.xml.XmlNodes.element;
 
 public class RelationshipsXmlTests {
     @Test
-    public void targetIsReadFromRelationshipElement() {
+    public void relationshipTargetsCanBeFoundById() {
         XmlElement element = element("relationship:Relationships", list(
             element("relationships:Relationship", map(
                 "Id", "rId8",
                 "Type", "http://schemas.openxmlformats.org/officeDocument/2006/relationships/hyperlink",
-                "Target", "http://example.com"))));
+                "Target", "http://example.com"
+            )),
+            element("relationships:Relationship", map(
+                "Id", "rId2",
+                "Type", "http://schemas.openxmlformats.org/officeDocument/2006/relationships/hyperlink",
+                "Target", "http://example.net"
+            ))
+        ));
 
         Relationships relationships = readRelationshipsXmlElement(element);
 
         assertEquals("http://example.com", relationships.findTargetByRelationshipId("rId8"));
     }
+
+    @Test
+    public void relationshipTargetsCanBeFoundByType() {
+        XmlElement element = element("relationship:Relationships", list(
+            element("relationships:Relationship", map(
+                "Id", "rId2",
+                "Target", "docProps/core.xml",
+                "Type", "http://schemas.openxmlformats.org/package/2006/relationships/metadata/core-properties"
+            )),
+            element("relationships:Relationship", map(
+                "Id", "rId1",
+                "Target", "word/document.xml",
+                "Type", "http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"
+            )),
+            element("relationships:Relationship", map(
+                "Id", "rId3",
+                "Target", "word/document2.xml",
+                "Type", "http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"
+            ))
+        ));
+
+        Relationships relationships = readRelationshipsXmlElement(element);
+
+        assertEquals(
+            list("word/document.xml", "word/document2.xml"),
+            relationships.findTargetsByType("http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument")
+        );
+    }
+
+    @Test
+    public void whenThereAreNoRelationshipsOfRequestedTypeThenEmptyListIsReturned() {
+
+        XmlElement element = element("relationship:Relationships");
+
+        Relationships relationships = readRelationshipsXmlElement(element);
+
+        assertEquals(
+            list(),
+            relationships.findTargetsByType("http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument")
+        );
+    }
 }