class TestBookFactory(unittest.TestCase): def setUp(self): self.__f = BookFactory() self.__epubPath = "aquarius/bookformats/tests/data/TreasureIsland.epub" self.__pdfpath = "aquarius/bookformats/tests/data/1.pdf" def test_get_non_existant_book_returns_none(self): b = self.__f.get_book("MyBook.rubbish") self.assertIsNone(b) def test_get_epub_calls_epub_creator(self): epub_creator = EpubCreator() epub_creator.create = Mock() self.__f.epub_creator = epub_creator self.__f.get_book(self.__epubPath) self.assertTrue(epub_creator.create.called) def test_get_pdf_calls_pdf_creator(self): pdf_creator = PdfCreator() pdf_creator.create = Mock() self.__f.pdf_creator = pdf_creator self.__f.get_book(self.__pdfpath) self.assertTrue(pdf_creator.create.called) def test_epub_creator_property(self): self.__f.epub_creator = "moo" self.assertEquals("moo", self.__f.epub_creator) def test_pdf_creator_property(self): self.__f.pdf_creator = "moo" self.assertEquals("moo", self.__f.pdf_creator)
def setUp(self): self.__f = BookFactory() self.__epubPath = "aquarius/bookformats/tests/data/TreasureIsland.epub" self.__pdfpath = "aquarius/bookformats/tests/data/1.pdf"