package com.github.jhonnymertz.wkhtmltopdf.wrapper;
import com.github.jhonnymertz.wkhtmltopdf.wrapper.configurations.WrapperConfig;
import com.github.jhonnymertz.wkhtmltopdf.wrapper.params.Param;
import java.io.IOException;
import org.apache.pdfbox.pdfparser.PDFParser;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.util.PDFTextStripper;
import org.junit.Assert;
import org.junit.Test;
import java.io.ByteArrayInputStream;
import java.io.File;
import static org.hamcrest.core.StringContains.containsString;
public class PdfTest {
@Test
public void testCommand() throws Exception {
Pdf pdf = new Pdf();
pdf.addToc();
pdf.addParam(new Param("--enable-javascript"), new Param("--html-header", "file:///example.html"));
pdf.addPageFromUrl("http://www.google.com");
Assert.assertThat("command params should contain the --enable-javascript and --html-header", pdf.getCommand(), containsString("--enable-javascript --html-header file:///example.html"));
}
@Test
public void findExecutable() throws Exception {
WrapperConfig wc = new WrapperConfig();
Assert.assertThat("executable should be /usr/bin/wkhtmltopdf", wc.findExecutable(), containsString("/usr/bin/wkhtmltopdf"));
}
@Test
public void testPdfFromStringTo() throws Exception {
// GIVEN a html template containing special characters that java stores in utf-16 internally
Pdf pdf = new Pdf();
pdf.addPageFromString("Müller
");
// WHEN
byte[] pdfBytes = pdf.getPDF();
String pdfText = getPdfTextFromBytes(pdfBytes);
Assert.assertThat("document should contain the creditorName", pdfText, containsString("Müller"));
}
@Test
public void testMultiplePages() throws Exception {
Pdf pdf = new Pdf();
pdf.addPageFromString("Page 1
");
pdf.addPageFromString("Page 2
");
pdf.addPageFromUrl("http://www.google.com");
pdf.addPageFromString("Page 4
");
// WHEN
byte[] pdfBytes = pdf.getPDF();
String pdfText = getPdfTextFromBytes(pdfBytes);
Assert.assertThat("document should contain the fourth page name", pdfText, containsString("Page 4"));
}
@Test
public void callingGetCommandFollowedByGetPdfShouldNotInfluenceTheOutput() throws Exception {
Pdf pdf = new Pdf();
pdf.addPageFromString("Twice
");
// WHEN
pdf.getCommand();
//Followed by
byte[] pdfBytes = pdf.getPDF();//Causes the page fromString's content to have become the file path
String pdfText = getPdfTextFromBytes(pdfBytes);
Assert.assertThat("document should contain the string that was originally inserted", pdfText, containsString("Twice"));
}
@Test
public void testRemovingGeneratedFile() throws Exception {
Pdf pdf = new Pdf();
pdf.addPageFromString("Page 1
");
pdf.addPageFromString("Page 2
");
pdf.addPageFromUrl("http://www.google.com");
pdf.addPageFromString("Page 4
");
File savedPdf = pdf.saveAs("output.pdf");
Assert.assertTrue(savedPdf.delete());
}
private String getPdfTextFromBytes(byte[] pdfBytes) throws IOException {
PDFParser parser = new PDFParser(new ByteArrayInputStream(pdfBytes));
// that is a valid PDF (otherwise an IOException occurs)
parser.parse();
PDFTextStripper pdfTextStripper = new PDFTextStripper();
return pdfTextStripper.getText(new PDDocument(parser.getDocument()));
}
}