Check size of elements in read

1 files changed, 3 insertions, 3 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 73b609a..d78faa1 100644
--- a/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java
@@ -399,9 +399,9 @@ public class BodyXmlTests {
     }
 
     private static Result<DocumentElement> read(Maker<BodyXmlReader> reader, XmlElement element) {
-        // TODO: check exactly 1 element
-        return readAll(reader, element)
-            .map(elements -> elements.get(0));
+        Result<List<DocumentElement>> result = readAll(reader, element);
+        assertThat(result.getValue(), Matchers.hasSize(1));
+        return result.map(elements -> elements.get(0));
     }
 
     private static Result<List<DocumentElement>> readAll(Maker<BodyXmlReader> reader, XmlElement element) {