1 files changed, 36 insertions, 19 deletions
diff --git a/src/test/java/org/zwobble/mammoth/tests/docx/DocumentReaderTests.java b/src/test/java/org/zwobble/mammoth/tests/docx/DocumentReaderTests.java
index 1c07679..db24404 100644
--- a/src/test/java/org/zwobble/mammoth/tests/docx/DocumentReaderTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/docx/DocumentReaderTests.java
@@ -38,30 +38,47 @@ public class DocumentReaderTests {
 
     @Test
     public void mainDocumentIsFoundUsingPackageRelationships() {
-        InternalResult<Document> result = DocumentReader.readDocument(Optional.empty(), InMemoryArchive.fromStrings(map(
+        Archive archive = InMemoryArchive.fromStrings(map(
             "word/document2.xml", XmlWriter.toString(
-                element("w:document", list(
-                    element("w:body", list(
-                        element("w:p", list(
-                            element("w:r", list(
-                                element("w:t", list(text("Hello.")))
-                            ))
-                        ))
-                    ))
-                )),
+                element(
+                    "w:document", list(
+                        element(
+                            "w:body", list(
+                                element(
+                                    "w:p", list(
+                                        element(
+                                            "w:r", list(
+                                                element("w:t", list(text("Hello.")))
+                                            )
+                                        )
+                                    )
+                                )
+                            )
+                        )
+                    )
+                ),
                 mainDocumentNamespaces
             ),
             "_rels/.rels", XmlWriter.toString(
-                element("Relationships", list(
-                    element("Relationship", map(
-                        "Id", "rId1",
-                        "Target", "/word/document2.xml",
-                        "Type", "http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"
-                    ))
-                )),
+                element(
+                    "Relationships", list(
+                        element(
+                            "Relationship", map(
+                                "Id", "rId1",
+                                "Target", "/word/document2.xml",
+                                "Type", "http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument"
+                            )
+                        )
+                    )
+                ),
                 relationshipsNamespaces
             )
-        )));
+        ));
+        InternalResult<Document> result = DocumentReader.readDocument(
+            Optional.empty(),
+            archive,
+            false
+        );
 
         assertThat(result, isInternalSuccess(document(
             withChildren(paragraphWithText("Hello."))
@@ -84,7 +101,7 @@ public class DocumentReaderTests {
         ));
         PassThroughException exception = assertThrows(
             PassThroughException.class,
-            () -> DocumentReader.readDocument(Optional.empty(), archive)
+            () -> DocumentReader.readDocument(Optional.empty(), archive, false)
         );
         assertThat(exception.getMessage(), equalTo("java.io.IOException: Could not find main document part. Are you sure this is a valid .docx file?"));
     }