class TestHtmlRequestHandler(RequestHandlerTestBase): def setUp(self): self.initialise_app_mock() self.initialise_search_handler_mock() self.initialise_book_handler_mock() self.initialise_first_letter_handler_mock() self.initialise_index_handler_mock() self.initialise_download_handler_mock() self.initialise_html_request_handler() def initialise_app_mock(self): RequestHandlerTestBase.initialise_app_mock(self) self.app.harvest_books = Mock(return_value=None) def initialise_search_handler_mock(self): self.__search_handler = HtmlRequestHandlerSearch(self.app) self.__search_handler.handle = Mock() def initialise_book_handler_mock(self): self.__book_handler = HtmlRequestHandlerBook(self.app) self.__book_handler.handle = Mock() def initialise_first_letter_handler_mock(self): self.__first_letter_handler = HtmlRequestHandlerFirstLetter(self.app) self.__first_letter_handler.handle = Mock() def initialise_index_handler_mock(self): self.__index_handler = HtmlRequestHandlerIndex(self.app) self.__index_handler.handle = Mock() def initialise_download_handler_mock(self): self.__download_handler = HtmlRequestHandlerDownload(self.app) self.__download_handler.handle = Mock() def initialise_html_request_handler(self): self.__h = HtmlRequestHandler(self.app) self.__h.set_search_handler(self.__search_handler) self.__h.set_book_handler(self.__book_handler) self.__h.set_first_letter_handler(self.__first_letter_handler) self.__h.set_index_handler(self.__index_handler) self.__h.set_download_handler(self.__download_handler) def test_search_handler_calls_search_handler_object(self): self.__h.search_handler("searchTerm") self.assert_called(self.__search_handler.handle) def test_harvest_handler_calls_harvest_books(self): self.__h.harvest_handler() self.assert_called(self.app.harvest_books) def test_book_handler_calls_book_handler_object(self): self.__h.book_handler("1") self.assert_called(self.__book_handler.handle) def test_download_handler_calls_download_handler_object(self): self.__h.download_handler(None, None) self.assert_called(self.__download_handler.handle) def test_first_letter_calls_first_letter_handler_object(self): self.__h.first_letter_handler("T") self.assert_called(self.__first_letter_handler.handle) def test_index_calls_index_handler_object(self): self.__h.index_handler() self.assert_called(self.__index_handler.handle)