2 files changed, 2 insertions, 30 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 2c1d505..8b31452 100644
--- a/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java
@@ -437,7 +437,7 @@ public class BodyXmlTests {
         Relationships relationships = new Relationships(map(
             "rId5", new Relationship("media/hat.png")));
         String imageBytes = "Not an image at all!";
-        DocxFile file = new InMemoryDocxFile(map("word/media/hat.png", imageBytes));
+        DocxFile file = InMemoryDocxFile.fromStrings(map("word/media/hat.png", imageBytes));
 
         Image image = (Image) readSuccess(
             a(bodyReader, with(RELATIONSHIPS, relationships), with(DOCX_FILE, file)),
@@ -465,7 +465,7 @@ public class BodyXmlTests {
         XmlElement element = inlineImageXml(embeddedBlipXml("rId5"), "");
         Relationships relationships = new Relationships(map(
             "rId5", new Relationship("media/hat.emf")));
-        DocxFile file = new InMemoryDocxFile(map("word/media/hat.emf", "Not an image at all!"));
+        DocxFile file = InMemoryDocxFile.fromStrings(map("word/media/hat.emf", "Not an image at all!"));
         ContentTypes contentTypes = new ContentTypes(map("emf", "image/x-emf"), map());
 
         Result<?> result = read(
diff --git a/src/test/java/org/zwobble/mammoth/tests/docx/InMemoryDocxFile.java b/src/test/java/org/zwobble/mammoth/tests/docx/InMemoryDocxFile.java
deleted file mode 100644
index 4aef42e..0000000
--- a/src/test/java/org/zwobble/mammoth/tests/docx/InMemoryDocxFile.java
+++ /dev/null
@@ -1,28 +0,0 @@
-package org.zwobble.mammoth.tests.docx;
-
-import org.zwobble.mammoth.docx.DocxFile;
-
-import java.io.ByteArrayInputStream;
-import java.io.IOException;
-import java.io.InputStream;
-import java.nio.charset.StandardCharsets;
-import java.util.Map;
-import java.util.Optional;
-
-public class InMemoryDocxFile implements DocxFile {
-    private final Map<String, String> entries;
-
-    public InMemoryDocxFile(Map<String, String> entries) {
-        this.entries = entries;
-    }
-
-    @Override
-    public Optional<InputStream> tryGetInputStream(String name) throws IOException {
-        return Optional.ofNullable(entries.get(name))
-            .map(value -> new ByteArrayInputStream(value.getBytes(StandardCharsets.UTF_8)));
-    }
-
-    @Override
-    public void close() throws IOException {
-    }
-}