From 6efa9397505e42b318879230d3b10a00f1ff2761 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 13 Feb 2016 21:19:54 +0000
Subject: Check size of elements in read

---
 src/test/java/org/zwobble/mammoth/tests/docx/BodyXmlTests.java | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/test/java/org')

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) {
-- 
cgit v1.2.3