diff --git a/src/deutschland/bundesanzeiger/bundesanzeiger.py b/src/deutschland/bundesanzeiger/bundesanzeiger.py index 31d10c3..19fd2ef 100644 --- a/src/deutschland/bundesanzeiger/bundesanzeiger.py +++ b/src/deutschland/bundesanzeiger/bundesanzeiger.py @@ -166,8 +166,7 @@ def __get_next_page_link(self, content: str): if not next_link: return None - return next_link.attrs.get('href') - + return next_link.attrs.get("href") def __generate_result(self, url: str, page_limit: int): results = dict() diff --git a/tests/bundesanzeiger/test_parsing.py b/tests/bundesanzeiger/test_parsing.py index fc90fbb..48f5c2f 100644 --- a/tests/bundesanzeiger/test_parsing.py +++ b/tests/bundesanzeiger/test_parsing.py @@ -1,9 +1,12 @@ +import os from deutschland.bundesanzeiger import Bundesanzeiger def test_get_next_page_link(): - with open('response.html') as file: - ba = Bundesanzeiger() + html_file = os.path.join(os.path.dirname(__file__), "response.html") + with open(html_file) as file: html = file.read() - link = ba._Bundesanzeiger__get_next_page_link(html) - assert link == 'https://www.bundesanzeiger.de/pub/de/suchen2?4-1.-top~nav-pager-navigation-1-pagination~link' + ba = Bundesanzeiger() + link = ba._Bundesanzeiger__get_next_page_link(html) + expected = "https://www.bundesanzeiger.de/pub/de/suchen2?4-1.-top~nav-pager-navigation-1-pagination~link" + assert link == expected