From 64c7ee4284a048b7b6a7c7b1c42ebfb201bf765a Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Fri, 26 Mar 2021 23:46:21 +0000 Subject: Fix blank pages bug (fixes #98) --- tests/test_pdf_ops.py | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 tests/test_pdf_ops.py (limited to 'tests/test_pdf_ops.py') diff --git a/tests/test_pdf_ops.py b/tests/test_pdf_ops.py new file mode 100644 index 0000000..bed763a --- /dev/null +++ b/tests/test_pdf_ops.py @@ -0,0 +1,37 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +"""Tests for the PDF operations""" + +import os +import shutil +import tempfile +import unittest + +from pikepdf import Pdf + +from paper2remarkable.providers import LocalFile + +from _constants import TEST_FILE + + +class PdfOpsTestCase(unittest.TestCase): + def setUp(self): + self._tmpdir = tempfile.mkdtemp(prefix="p2r_test_blank_") + + def tearDown(self): + shutil.rmtree(self._tmpdir) + + def test_blank_pages(self): + local_filename = os.path.join(self._tmpdir, "test_blank.pdf") + with open(local_filename, "w") as fp: + fp.write(TEST_FILE) + prov = LocalFile(upload=False, blank=True) + out_filename = os.path.join(self._tmpdir, "test_blank1.pdf") + filename = prov.run(local_filename, filename=out_filename) + pdf = Pdf.open(filename) + self.assertEqual(len(pdf.pages), 2) + + +if __name__ == "__main__": + unittest.main() -- cgit v1.2.3