Add test. DEVSIX-2007

Author: JonPReilly

kernel/src/test/java/com/itextpdf/kernel/pdf/PdfStampingTest.java

@@ -44,10 +44,18 @@ This file is part of the iText (R) project.
 
 import com.itextpdf.io.source.ByteUtils;
 import com.itextpdf.io.util.DateTimeUtil;
+import com.itextpdf.kernel.events.Event;
+import com.itextpdf.kernel.events.IEventHandler;
+import com.itextpdf.kernel.events.PdfDocumentEvent;
+import com.itextpdf.kernel.font.PdfFontFactory;
+import com.itextpdf.kernel.geom.Rectangle;
+import com.itextpdf.kernel.pdf.canvas.PdfCanvas;
+import com.itextpdf.kernel.utils.CompareTool;
 import com.itextpdf.kernel.xmp.XMPException;
 import com.itextpdf.kernel.xmp.XMPMetaFactory;
 import com.itextpdf.test.ExtendedITextTest;
 import com.itextpdf.test.annotations.type.IntegrationTest;
+import org.junit.Assert;
 import org.junit.BeforeClass;
 import org.junit.Ignore;
 import org.junit.Test;
@@ -673,7 +681,7 @@ public void stampingStreamsCompression01() throws IOException {
 
         assertEquals(5731884, lengthBefore);
         float expected = 5731884;
-        float coef = Math.abs((expected - lengthAfter)/expected);
+        float coef = Math.abs((expected - lengthAfter) / expected);
         assertTrue(coef < 0.01);
     }
 
@@ -698,7 +706,7 @@ public void stampingStreamsCompression02() throws IOException {
 
         assertEquals(5731884, lengthBefore);
         float expected = 11321910;
-        float coef = Math.abs((expected - lengthAfter)/expected);
+        float coef = Math.abs((expected - lengthAfter) / expected);
         assertTrue(coef < 0.01);
     }
 
@@ -721,7 +729,7 @@ public void stampingStreamsCompression03() throws IOException {
 
         assertEquals(5731884, lengthBefore);
         float expected = 5729270;
-        float coef = Math.abs((expected - lengthAfter)/expected);
+        float coef = Math.abs((expected - lengthAfter) / expected);
         assertTrue(coef < 0.01);
     }
 
@@ -796,7 +804,7 @@ public void stampingXmp2() throws IOException, XMPException {
         for (int i = 0; i < pdfDoc3.getNumberOfPages(); i++) {
             pdfDoc3.getPage(i + 1);
         }
-        assertNotNull("XmpMetadata not found",  XMPMetaFactory.parseFromBuffer(pdfDoc3.getXmpMetadata()));
+        assertNotNull("XmpMetadata not found", XMPMetaFactory.parseFromBuffer(pdfDoc3.getXmpMetadata()));
         assertEquals("Number of pages", pageCount, pdfDoc3.getNumberOfPages());
         assertEquals("Rebuilt", false, reader3.hasRebuiltXref());
         assertEquals("Fixed", false, reader3.hasFixedXref());
@@ -1250,19 +1258,19 @@ public void stampingTestWithFullCompression01() throws IOException, InterruptedE
         pdfDoc.close();
         float result = new File(destinationFolder + "stampingTestWithFullCompression01.pdf").length();
         float expected = new File(sourceFolder + "cmp_stampingTestWithFullCompression01.pdf").length();
-        float coef = Math.abs((expected - result)/expected);
+        float coef = Math.abs((expected - result) / expected);
         assertTrue(coef < 0.01);
     }
 
     @Test
     public void stampingTestWithFullCompression02() throws IOException, InterruptedException {
         PdfDocument pdfDoc = new PdfDocument(new PdfReader(sourceFolder + "fullCompressedDocument.pdf"),
                 new PdfWriter(destinationFolder + "stampingTestWithFullCompression02.pdf",
-                new WriterProperties().setFullCompressionMode(false)));
+                        new WriterProperties().setFullCompressionMode(false)));
         pdfDoc.close();
         float result = new File(destinationFolder + "stampingTestWithFullCompression02.pdf").length();
         float expected = new File(sourceFolder + "cmp_stampingTestWithFullCompression02.pdf").length();
-        float coef = Math.abs((expected - result)/expected);
+        float coef = Math.abs((expected - result) / expected);
         assertTrue(coef < 0.01);
     }
 
@@ -1284,4 +1292,42 @@ static void verifyPdfPagesCount(PdfObject root) {
             verifyPdfPagesCount(kids);
         }
     }
+
+    @Test
+    //TODO: DEVSIX-2007
+    public void stampingStreamNoEndingWhitespace01() throws IOException, InterruptedException {
+        System.out.println(destinationFolder);
+        PdfDocument pdfDoc = new PdfDocument(new PdfReader(sourceFolder + "stampingStreamNoEndingWhitespace01.pdf"),
+                new PdfWriter(destinationFolder + "stampingStreamNoEndingWhitespace01.pdf",
+                        new WriterProperties().setCompressionLevel(0)));
+
+        PdfDocument pdfDocInput = new PdfDocument(new PdfReader(sourceFolder + "stampingStreamNoEndingWhitespace01.pdf"));
+        PdfDocument pdfDocOutput = new PdfDocument(new PdfWriter(destinationFolder + "stampingStreamNoEndingWhitespace01.pdf", new WriterProperties().setCompressionLevel(0)));
+
+        class WatermarkEventHandler implements IEventHandler {
+
+            @Override
+            public void handleEvent(Event event) {
+                PdfDocumentEvent pdfEvent = (PdfDocumentEvent) event;
+                PdfPage page = pdfEvent.getPage();
+                PdfCanvas pdfCanvas = new PdfCanvas(page);
+                try {
+                    pdfCanvas.beginText()
+                            .setFontAndSize(PdfFontFactory.createFont(), 12.0f)
+                            .showText("Text")
+                            .endText();
+                } catch (IOException e) {
+                }
+            }
+        }
+        ;
+        pdfDocOutput.addEventHandler(PdfDocumentEvent.END_PAGE, new WatermarkEventHandler());
+
+        pdfDocInput.copyPagesTo(1, pdfDocInput.getNumberOfPages(), pdfDocOutput);
+
+        pdfDocInput.close();
+        pdfDocOutput.close();
+
+        Assert.assertNull(new CompareTool().compareByContent(destinationFolder + "stampingStreamNoEndingWhitespace01.pdf", sourceFolder + "cmp_stampingStreamNoEndingWhitespace01.pdf", destinationFolder, "diff_"));
+    }
 }