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, )