def test_all_options(self):
generator = BareResourceGenerator()
normal_options = generator.get_options()
local_options = generator.get_local_options()
# Order should be normal options, then local options
options_order = list(normal_options) + list(local_options)
self.assertListEqual(options_order, list(generator.options))
for option in local_options.values():
self.assertIsInstance(option, ResourceParameter)
def test_generate_thumbnail_valid_multiple_pages(self):
generator = BareResourceGenerator()
generator.data = MagicMock(
return_value=[
{"type": "html", "data": "Page 1"},
{"type": "html", "data": "Page 2", "thumbnail": True}
]
)
thumbnail_data = generator.generate_thumbnail()
self.assertEqual(thumbnail_data["data"], "Page 2")
def test_pdf_multiple_pages(self):
generator = BareResourceGenerator()
generator.data = MagicMock(
return_value=[
{"type": "html", "data": "Page 1"},
{"type": "html", "data": "Page 2"},
]
)
(pdf_file, filename) = generator.pdf("Test")
pdf = PdfFileReader(BytesIO(pdf_file))
self.assertEqual(pdf.getNumPages(), 2)
def test_generate_thumbnail_more_than_one_given(self):
generator = BareResourceGenerator()
generator.data = MagicMock(
return_value=[
{"type": "html", "data": "", "thumbnail": True},
{"type": "html", "data": "", "thumbnail": True}
]
)
self.assertRaises(
MoreThanOneThumbnailPageFoundError,
generator.generate_thumbnail,
)
def test_generate_thumbnail_none_given(self):
generator = BareResourceGenerator()
generator.data = MagicMock(
return_value=[
{"type": "html", "data": ""},
{"type": "html", "data": ""}
]
)
self.assertRaises(
ThumbnailPageNotFoundError,
generator.generate_thumbnail,
)
def test_generate_thumbnail_valid_single_page(self):
generator = BareResourceGenerator()
thumbnail_data = generator.generate_thumbnail()
self.assertEqual(thumbnail_data["data"], "Page 1")
def test_pdf_single_page(self):
generator = BareResourceGenerator()
(pdf_file, filename) = generator.pdf("Test")
pdf = PdfFileReader(BytesIO(pdf_file))
self.assertEqual(pdf.getNumPages(), 1)