Read relationship types

1 files changed, 43 insertions, 2 deletions

diff --git a/tests/docx/relationships_xml_tests.py b/tests/docx/relationships_xml_tests.py
index c3d0af0..5e52142 100644
--- a/tests/docx/relationships_xml_tests.py
+++ b/tests/docx/relationships_xml_tests.py
@@ -5,16 +5,57 @@ from mammoth.docx.relationships_xml import read_relationships_xml_element
 
 
 @istest
-def target_is_read_from_relationship_element():
+def relationship_targets_can_be_found_by_id():
     element = xml_element("relationships:Relationships", {}, [
         xml_element("relationships:Relationship", {
             "Id": "rId8",
             "Type": "http://schemas.openxmlformats.org/officeDocument/2006/relationships/hyperlink",
             "Target": "http://example.com",
-        })
+        }),
+        xml_element("relationships:Relationship", {
+            "Id": "rId2",
+            "Type": "http://schemas.openxmlformats.org/officeDocument/2006/relationships/hyperlink",
+            "Target": "http://example.net",
+        }),
     ])
     relationships = read_relationships_xml_element(element)
     assert_equal(
         "http://example.com",
         relationships.find_target_by_relationship_id("rId8"),
     )
+
+
+@istest
+def relationship_targets_can_be_found_by_type():
+    element = xml_element("relationships:Relationships", {}, [
+        xml_element("relationships:Relationship", {
+            "Id": "rId2",
+            "Target": "docProps/core.xml",
+            "Type": "http://schemas.openxmlformats.org/package/2006/relationships/metadata/core-properties",
+        }),
+        xml_element("relationships:Relationship", {
+            "Id": "rId1",
+            "Target": "word/document.xml",
+            "Type": "http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument",
+        }),
+        xml_element("relationships:Relationship", {
+            "Id": "rId3",
+            "Target": "word/document2.xml",
+            "Type": "http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument",
+        }),
+    ])
+    relationships = read_relationships_xml_element(element)
+    assert_equal(
+        ["word/document.xml", "word/document2.xml"],
+        relationships.find_targets_by_type("http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"),
+    )
+
+
+@istest
+def when_there_are_no_relationships_of_requested_type_then_empty_list_is_returned():
+    element = xml_element("relationships:Relationships", {}, [])
+    relationships = read_relationships_xml_element(element)
+    assert_equal(
+        [],
+        relationships.find_targets_by_type("http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"),
+    )