svn commit: r1881249 - /pdfbox/trunk/examples/src/test/java/org/apache/pdfbox/examples/pdmodel/TestRubberStampWithImage.java

[ti...@apache.org]

Author: tilman
Date: Thu Aug 27 17:07:58 2020
New Revision: 1881249

URL: http://svn.apache.org/viewvc?rev=1881249&view=rev
Log:
PDFBOX-4892: increase test coverage

Modified:
    pdfbox/trunk/examples/src/test/java/org/apache/pdfbox/examples/pdmodel/TestRubberStampWithImage.java

Modified: pdfbox/trunk/examples/src/test/java/org/apache/pdfbox/examples/pdmodel/TestRubberStampWithImage.java
URL: http://svn.apache.org/viewvc/pdfbox/trunk/examples/src/test/java/org/apache/pdfbox/examples/pdmodel/TestRubberStampWithImage.java?rev=1881249&r1=1881248&r2=1881249&view=diff
==============================================================================
--- pdfbox/trunk/examples/src/test/java/org/apache/pdfbox/examples/pdmodel/TestRubberStampWithImage.java (original)
+++ pdfbox/trunk/examples/src/test/java/org/apache/pdfbox/examples/pdmodel/TestRubberStampWithImage.java Thu Aug 27 17:07:58 2020
@@ -16,16 +16,32 @@
  */
 package org.apache.pdfbox.examples.pdmodel;
 
-import junit.framework.TestCase;
+import java.awt.image.BufferedImage;
 
 import java.io.File;
 import java.io.IOException;
 
+import javax.imageio.ImageIO;
+
+import org.apache.pdfbox.Loader;
+import org.apache.pdfbox.cos.COSName;
+import org.apache.pdfbox.pdmodel.PDDocument;
+import org.apache.pdfbox.pdmodel.graphics.image.PDImageXObject;
+import org.apache.pdfbox.pdmodel.interactive.annotation.PDAnnotationRubberStamp;
+import org.apache.pdfbox.pdmodel.interactive.annotation.PDAppearanceDictionary;
+import org.apache.pdfbox.pdmodel.interactive.annotation.PDAppearanceEntry;
+import org.apache.pdfbox.pdmodel.interactive.annotation.PDAppearanceStream;
+import org.apache.pdfbox.rendering.PDFRenderer;
+
+import org.junit.Assert;
+import org.junit.Test;
+
 /**
  * Test for RubberStampWithImage
  */
-public class TestRubberStampWithImage extends TestCase
+public class TestRubberStampWithImage
 {
+    @Test
     public void test() throws IOException
     {
         String documentFile = "src/test/resources/org/apache/pdfbox/examples/pdmodel/document.pdf";
@@ -34,8 +50,51 @@ public class TestRubberStampWithImage ex
 
         new File("target/test-output").mkdirs();
 
+        BufferedImage bim1;
+        try (PDDocument doc1 = Loader.loadPDF(new File(documentFile)))
+        {
+            bim1 = new PDFRenderer(doc1).renderImage(0);
+        }
+
         String[] args = new String[] { documentFile, outFile, stampFile };
         RubberStampWithImage rubberStamp = new RubberStampWithImage();
         rubberStamp.doIt(args);
+
+        try (PDDocument doc2 = Loader.loadPDF(new File(outFile)))
+        {
+            BufferedImage bim2 = new PDFRenderer(doc2).renderImage(0);
+            Assert.assertFalse(compareImages(bim1, bim2));
+            PDAnnotationRubberStamp rubberStampAnnotation = (PDAnnotationRubberStamp) doc2.getPage(0).getAnnotations().get(0);
+            PDAppearanceDictionary appearance = rubberStampAnnotation.getAppearance();
+            PDAppearanceEntry normalAppearance = appearance.getNormalAppearance();
+            PDAppearanceStream appearanceStream = normalAppearance.getAppearanceStream();
+            PDImageXObject ximage = (PDImageXObject) appearanceStream.getResources().getXObject(COSName.getPDFName("Im1"));
+            BufferedImage actualStampImage = ximage.getImage();
+            BufferedImage expectedStampImage = ImageIO.read(new File(stampFile));
+            Assert.assertTrue(compareImages(expectedStampImage, actualStampImage));
+        }
+    }
+
+    private boolean compareImages(BufferedImage bim1, BufferedImage bim2)
+    {
+        if (bim1.getWidth() != bim2.getWidth())
+        {
+            return false;
+        }
+        if (bim1.getHeight() != bim2.getHeight())
+        {
+            return false;
+        }
+        for (int x = 0; x < bim1.getWidth(); ++x)
+        {
+            for (int y = 0; y < bim1.getHeight(); ++y)
+            {
+                if (bim1.getRGB(x, y) != bim2.getRGB(x, y))
+                {
+                    return false;
+                }
+            }
+        }
+        return true;
     }
 }