/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.tom_roush.pdfbox.pdmodel;
import com.tom_roush.pdfbox.io.IOUtils;
import junit.framework.TestCase;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
/**
* Testcase introduced with PDFBOX-1581.
*
*/
public class TestPDDocument extends TestCase
{
private File testResultsDir = new File("target/test-output");
@Override
protected void setUp() throws Exception
{
super.setUp();
testResultsDir.mkdirs();
}
/**
* Test document save/load using a stream.
* @throws IOException if something went wrong
*/
public void testSaveLoadStream() throws IOException
{
// Create PDF with one blank page
PDDocument document = new PDDocument();
document.addPage(new PDPage());
// Save
ByteArrayOutputStream baos = new ByteArrayOutputStream();
document.save(baos);
document.close();
// Verify content
byte[] pdf = baos.toByteArray();
assertTrue(pdf.length > 200);
assertEquals("%PDF-1.4", new String(Arrays.copyOfRange(pdf, 0, 8), "UTF-8"));
assertEquals("%%EOF\n", new String(Arrays.copyOfRange(pdf, pdf.length - 6, pdf.length), "UTF-8"));
// Load
PDDocument loadDoc = PDDocument.load(new ByteArrayInputStream(pdf));
assertEquals(1, loadDoc.getNumberOfPages());
loadDoc.close();
}
/**
* Test document save/load using a file.
* @throws IOException if something went wrong
*/
public void testSaveLoadFile() throws IOException
{
// Create PDF with one blank page
PDDocument document = new PDDocument();
document.addPage(new PDPage());
// Save
File targetFile = new File(testResultsDir, "pddocument-saveloadfile.pdf");
document.save(targetFile);
document.close();
// Verify content
assertTrue(targetFile.length() > 200);
InputStream in = new FileInputStream(targetFile);
byte[] pdf = IOUtils.toByteArray(in);
in.close();
assertTrue(pdf.length > 200);
assertEquals("%PDF-1.4", new String(Arrays.copyOfRange(pdf, 0, 8), "UTF-8"));
assertEquals("%%EOF\n", new String(Arrays.copyOfRange(pdf, pdf.length - 6, pdf.length), "UTF-8"));
// Load
PDDocument loadDoc = PDDocument.load(targetFile);
assertEquals(1, loadDoc.getNumberOfPages());
loadDoc.close();
}
/**
* Test document save/loadNonSeq using a stream.
* @throws IOException if something went wrong
*/
public void testSaveLoadNonSeqStream() throws IOException
{
// Create PDF with one blank page
PDDocument document = new PDDocument();
document.addPage(new PDPage());
// Save
ByteArrayOutputStream baos = new ByteArrayOutputStream();
document.save(baos);
document.close();
// Verify content
byte[] pdf = baos.toByteArray();
assertTrue(pdf.length > 200);
assertEquals("%PDF-1.4", new String(Arrays.copyOfRange(pdf, 0, 8), "UTF-8"));
assertEquals("%%EOF\n", new String(Arrays.copyOfRange(pdf, pdf.length - 6, pdf.length), "UTF-8"));
// Load
PDDocument loadDoc = PDDocument.load(new ByteArrayInputStream(pdf));
assertEquals(1, loadDoc.getNumberOfPages());
loadDoc.close();
}
/**
* Test document save/loadNonSeq using a file.
* @throws IOException if something went wrong
*/
public void testSaveLoadNonSeqFile() throws IOException
{
// Create PDF with one blank page
PDDocument document = new PDDocument();
document.addPage(new PDPage());
// Save
File targetFile = new File(testResultsDir, "pddocument-saveloadnonseqfile.pdf");
document.save(targetFile);
document.close();
// Verify content
assertTrue(targetFile.length() > 200);
InputStream in = new FileInputStream(targetFile);
byte[] pdf = IOUtils.toByteArray(in);
in.close();
assertTrue(pdf.length > 200);
assertEquals("%PDF-1.4", new String(Arrays.copyOfRange(pdf, 0, 8), "UTF-8"));
assertEquals("%%EOF\n", new String(Arrays.copyOfRange(pdf, pdf.length - 6, pdf.length), "UTF-8"));
// Load
PDDocument loadDoc = PDDocument.load(targetFile);
assertEquals(1, loadDoc.getNumberOfPages());
loadDoc.close();
}
/**
* Test get/setVersion.
* @throws IOException if something went wrong
*/
public void testVersions() throws IOException
{
PDDocument document = new PDDocument();
// test default version
assertEquals(1.4f, document.getVersion());
assertEquals(1.4f, document.getDocument().getVersion());
assertEquals("1.4", document.getDocumentCatalog().getVersion());
// force downgrading version (header)
document.getDocument().setVersion(1.3f);
document.getDocumentCatalog().setVersion(null);
// test new version (header)
assertEquals(1.3f, document.getVersion());
assertEquals(1.3f, document.getDocument().getVersion());
assertNull(document.getDocumentCatalog().getVersion());
document.close();
// check if version downgrade is denied
document = new PDDocument();
document.setVersion(1.3f);
// all versions shall have their default value
assertEquals(1.4f, document.getVersion());
assertEquals(1.4f, document.getDocument().getVersion());
assertEquals("1.4", document.getDocumentCatalog().getVersion());
// check version upgrade
document.setVersion(1.5f);
// overall version has to be 1.5f
assertEquals(1.5f, document.getVersion());
// header version has to be unchanged
assertEquals(1.4f, document.getDocument().getVersion());
// catalog version version has to be 1.5
assertEquals("1.5", document.getDocumentCatalog().getVersion());
document.close();
}
}