Fix Comments from MR

DDS-GmbH · Jan 18, 2023 · a8e0382 · a8e0382
1 parent 07937c0
commit a8e0382
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/main/java/com/docutools/poipath/xwpf/XWPFBuilder.java b/src/main/java/com/docutools/poipath/xwpf/XWPFBuilder.java
@@ -50,13 +50,15 @@ public XWPFBuilder<XWPFParagraph> paragraph() {
    */
   public XWPFBuilder<T> paragraph(String text) {
     var paragraphTexts = text.split("\n");
+    XWPFParagraph paragraph;
     if (bodyElement instanceof XWPFDocument document) {
-      Arrays.stream(paragraphTexts).forEach(t -> document.createParagraph().createRun().setText(t));
+      paragraph = document.createParagraph();
     } else if (bodyElement instanceof XWPFTableCell cell) {
-      Arrays.stream(paragraphTexts).forEach(t -> cell.addParagraph().createRun().setText(t));
+      paragraph = cell.addParagraph();
     } else {
       throw isUnsupported(XWPFParagraph.class);
     }
+    paragraph.createRun().setText(text);
     return this;
   }
 

diff --git a/src/test/java/com/docutools/poipath/xwpf/XWPFBuildingTests.java b/src/test/java/com/docutools/poipath/xwpf/XWPFBuildingTests.java
@@ -4,6 +4,7 @@
 import static com.docutools.poipath.xwpf.XWPFBuilder.buildDocument;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.empty;
+import static org.hamcrest.Matchers.endsWith;
 import static org.hamcrest.Matchers.equalTo;
 import static org.hamcrest.Matchers.is;
 import static org.hamcrest.Matchers.notNullValue;
@@ -78,7 +79,7 @@ void addMultipleParagraphs() {
 
     // Assert
     assertThat(Objects.requireNonNull(wrap(document).bodyElement(0).asParagraph()).text(), startsWith("Lorem ipsum"));
-    assertThat(Objects.requireNonNull(wrap(document).bodyElement(1).asParagraph()).text(), startsWith("Tellus id"));
+    assertThat(Objects.requireNonNull(wrap(document).bodyElement(0).asParagraph()).text(), endsWith("arcu."));
   }
 
   @Test