mirror of
https://github.com/pdf2htmlEX/pdf2htmlEX.git
synced 2024-09-29 01:31:29 +00:00
126 lines
4.8 KiB
Python
126 lines
4.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import subprocess
|
|
import shutil
|
|
import unittest
|
|
|
|
#from selenium.common.exceptions import WebDriverException
|
|
from PIL import Image, ImageChops
|
|
from test import Common
|
|
|
|
class BrowserTests(Common):
|
|
TEST_DATA_DIR = os.path.join(Common.TEST_DIR, 'browser_tests')
|
|
|
|
DEFAULT_PDF2HTMLEX_ARGS = [
|
|
'--fit-width', 800,
|
|
'--last-page', 1,
|
|
# avoid base64 to make it faster
|
|
]
|
|
|
|
BROWSER_WIDTH=800
|
|
BROWSER_HEIGHT=1200
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
pass
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
pass
|
|
|
|
def run_test_case(self, filename, args=[], page_must_load=True):
|
|
basefilename, extension = os.path.splitext(filename)
|
|
self.assertEqual(extension.lower(), '.pdf', 'Input file is not PDF')
|
|
|
|
htmlfilename = basefilename + '.html'
|
|
ref_htmlfolder = os.path.join(self.TEST_DATA_DIR, basefilename)
|
|
ref_htmlfilename = os.path.join(ref_htmlfolder, htmlfilename)
|
|
out_htmlfilename = os.path.join(self.OUTDIR, htmlfilename)
|
|
|
|
pdf2htmlEX_args = self.DEFAULT_PDF2HTMLEX_ARGS + args + [
|
|
os.path.join(self.TEST_DATA_DIR, filename),
|
|
htmlfilename ]
|
|
result = self.run_pdf2htmlEX(pdf2htmlEX_args)
|
|
|
|
self.assertIn(htmlfilename, result['output_files'], 'HTML file is not generated')
|
|
|
|
if self.GENERATING_MODE:
|
|
# copy generated html files
|
|
shutil.rmtree(ref_htmlfolder, True)
|
|
shutil.copytree(self.OUTDIR, ref_htmlfolder)
|
|
return
|
|
|
|
# keep a record of the HTML files (and any differences)
|
|
# for later reporting of test results
|
|
#
|
|
try:
|
|
os.makedirs(self.HTMDIR, 0o755, True)
|
|
outHtmlFile = os.path.join(self.HTMDIR, basefilename+'.out.html')
|
|
refHtmlFile = os.path.join(self.HTMDIR, basefilename+'.ref.html')
|
|
difHtmlFile = os.path.join(self.HTMDIR, basefilename+'.diff.html')
|
|
shutil.copy(out_htmlfilename, outHtmlFile)
|
|
shutil.copy(ref_htmlfilename, refHtmlFile)
|
|
os.system("diff "+outHtmlFile+" "+refHtmlFile+" > "+difHtmlFile+" 2>&1")
|
|
except:
|
|
pass
|
|
|
|
pngfilename_out = os.path.join(self.PNGDIR, basefilename + '.out.png')
|
|
self.generate_image(out_htmlfilename, pngfilename_out)
|
|
out_img = Image.open(pngfilename_out).convert('RGB')
|
|
|
|
pngfilename_ref = os.path.join(self.PNGDIR, basefilename + '.ref.png')
|
|
self.generate_image(ref_htmlfilename, pngfilename_ref, page_must_load=page_must_load)
|
|
ref_img = Image.open(pngfilename_ref).convert('RGB')
|
|
|
|
diff_img = ImageChops.difference(ref_img, out_img);
|
|
|
|
# ALWAYS save the diff image so we can manually check the diff
|
|
# see: (http://stackoverflow.com/questions/15721484):
|
|
diff_file_name = os.path.join(self.PNGDIR, basefilename + '.diff.png')
|
|
diff_img.convert('RGB').save(diff_file_name)
|
|
|
|
diff_bbox = diff_img.getbbox()
|
|
print("\nTesting at: [", basefilename, "]")
|
|
|
|
if diff_bbox is None:
|
|
print(" passed")
|
|
else:
|
|
print(" diff bounding box: ", diff_bbox, " should be None")
|
|
diff_size = (diff_bbox[2] - diff_bbox[0]) * (diff_bbox[3] - diff_bbox[1])
|
|
img_size = ref_img.size[0] * ref_img.size[1]
|
|
self.fail(('PNG files %s and %s differ by at most %d pixels, '+
|
|
'(%f%% of %d pixels in total), difference: %s') %
|
|
(pngfilename_out, pngfilename_ref,
|
|
diff_size, 100.0*diff_size/img_size, img_size, diff_file_name))
|
|
|
|
@unittest.skipIf(Common.GENERATING_MODE, 'Do not auto generate reference for test_fail')
|
|
def test_fail(self):
|
|
# The HTML reference is generated manually, which mismatches the PDF
|
|
# To test if the environment can detect any errors
|
|
# E.g. when network is down, 404 message is shown for any HTML message
|
|
with self.assertRaises(AssertionError):
|
|
self.run_test_case('test_fail.pdf', page_must_load=False)
|
|
|
|
def test_basic_text(self):
|
|
self.run_test_case('basic_text.pdf')
|
|
|
|
def test_geneve_1564(self):
|
|
self.run_test_case('geneve_1564.pdf')
|
|
|
|
@unittest.skip("Skipping text_visibility test due to clipping issues")
|
|
def test_text_visibility(self):
|
|
self.run_test_case('text_visibility.pdf', ['--correct-text-visibility', 1])
|
|
|
|
def test_process_form(self):
|
|
self.run_test_case('with_form.pdf', ['--process-form', 1])
|
|
|
|
def test_invalid_unicode_issue477(self):
|
|
self.run_test_case('invalid_unicode_issue477.pdf')
|
|
|
|
def test_svg_background_with_page_rotation_issue402(self):
|
|
self.run_test_case('svg_background_with_page_rotation_issue402.pdf', ['--bg-format', 'svg'])
|
|
|
|
def test_fontfile3_opentype(self):
|
|
self.run_test_case('fontfile3_opentype.pdf', ['-l', 1])
|