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