svn commit: r1741317 - /pdfbox/branches/2.0/pdfbox/src/test/java/org/apache/pdfbox/pdmodel/TestPDDocument.java

[ti...@apache.org]

Author: tilman
Date: Wed Apr 27 19:10:02 2016
New Revision: 1741317

URL: http://svn.apache.org/viewvc?rev=1741317&view=rev
Log:
PDFBOX-3280: improve test with rendering comparison

Modified:
    pdfbox/branches/2.0/pdfbox/src/test/java/org/apache/pdfbox/pdmodel/TestPDDocument.java

Modified: pdfbox/branches/2.0/pdfbox/src/test/java/org/apache/pdfbox/pdmodel/TestPDDocument.java
URL: http://svn.apache.org/viewvc/pdfbox/branches/2.0/pdfbox/src/test/java/org/apache/pdfbox/pdmodel/TestPDDocument.java?rev=1741317&r1=1741316&r2=1741317&view=diff
==============================================================================
--- pdfbox/branches/2.0/pdfbox/src/test/java/org/apache/pdfbox/pdmodel/TestPDDocument.java (original)
+++ pdfbox/branches/2.0/pdfbox/src/test/java/org/apache/pdfbox/pdmodel/TestPDDocument.java Wed Apr 27 19:10:02 2016
@@ -16,6 +16,9 @@
  */
 package org.apache.pdfbox.pdmodel;
 
+import java.awt.Color;
+import java.awt.Font;
+import java.awt.Graphics;
 import java.awt.image.BufferedImage;
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
@@ -34,8 +37,11 @@ import junit.framework.TestCase;
 import static junit.framework.TestCase.assertNull;
 import static junit.framework.TestCase.assertTrue;
 import static junit.framework.TestCase.fail;
+import org.apache.pdfbox.pdmodel.font.PDType1Font;
 import org.apache.pdfbox.pdmodel.graphics.image.LosslessFactory;
 import org.apache.pdfbox.pdmodel.graphics.image.PDImageXObject;
+import org.apache.pdfbox.rendering.PDFRenderer;
+import org.junit.Assert;
 
 /**
  * Testcase introduced with PDFBOX-1581.
@@ -264,16 +270,38 @@ public class TestPDDocument extends Test
         PDDocument doc1 = new PDDocument();
         PDPage page = new PDPage();
         PDPageContentStream pageContentStream = new PDPageContentStream(doc1, page);
-        BufferedImage bim = new BufferedImage(20, 20, BufferedImage.TYPE_INT_RGB);
+        BufferedImage bim = new BufferedImage(100, 50, BufferedImage.TYPE_INT_RGB);
+        Graphics g = bim.getGraphics();
+        Font font = new Font("Dialog", Font.PLAIN, 20);   
+        g.setFont(font);
+        g.drawString("PDFBox", 10, 30);
+        g.dispose();
         PDImageXObject img = LosslessFactory.createFromImage(doc1, bim);
-        pageContentStream.drawImage(img, 0, 0);
+        pageContentStream.drawImage(img, 200, 500);
+        pageContentStream.setFont(PDType1Font.HELVETICA, 20);
+        pageContentStream.beginText();
+        pageContentStream.setNonStrokingColor(Color.blue);
+        pageContentStream.newLineAtOffset(200, 600);
+        pageContentStream.showText("PDFBox");
+        pageContentStream.endText();
         pageContentStream.close();
         doc1.addPage(page);
-
+        BufferedImage bim1 = new PDFRenderer(doc1).renderImage(0);
+        
         PDDocument doc2 = new PDDocument();
         doc2.importPage(doc1.getPage(0));
         doc1.close();
+        BufferedImage bim2 = new PDFRenderer(doc2).renderImage(0);
         doc2.save(new ByteArrayOutputStream());
         doc2.close();
+        
+        assertEquals(bim1.getWidth(), bim2.getWidth());
+        assertEquals(bim1.getHeight(), bim2.getHeight());
+        int w = bim1.getWidth();
+        int h = bim1.getHeight();
+        int[] pixels1 = bim1.getRaster().getPixels(0, 0, w, h, (int[]) null);
+        int[] pixels2 = bim2.getRaster().getPixels(0, 0, w, h, (int[]) null);
+        assertEquals(w * h * 3, pixels1.length);
+        Assert.assertArrayEquals(pixels1, pixels2);
     }
 }