def test_get_rental_books_no_table(self): with patch( "mmlibrary.html_parser.HtmlParser._get_books_table") as method: method.return_value = None books = HtmlParser.get_rental_books(None) assert isinstance(books, RentalBooks) assert books.len == 0
def _get_rental_books(self, user: User, type: str = BookFilter.TYPE_RENTAL_NORMAL, param: Dict = {}) -> RentalBooks: html = self._html_page.fetch_login_page(Library.LIBRALY_HOME_URL, user) all_books = HtmlParser.get_rental_books(html) filtered_books = BookFilter.do(all_books, type=type, param=param) return filtered_books
def test_get_rental_books(self): page = HtmlPage() user = User(os.environ["USER1"]) html = page.fetch_login_page(Library.LIBRALY_HOME_URL, user) HtmlParser.get_rental_books(html) page.release_resource()