def test_pagination_previous_only(): """ Test that webpage.pagination_html() returns previous page correctly when no next page """ webpage = Webpage(TEST_WEBSITE) webpage.url_previous = 'page-1.html' result = '<nav class="magnetizer-pagination"><ul>' result += '<li><a href="page-1.html" class="magnetizer-previous">Newer posts</a></li>' result += '</ul></nav>' assert webpage.pagination_html() == result
def test_pagination_previous_and_next(): """ Test that webpage.pagination_html() returns next and previous pages correctly when both are available """ webpage = Webpage(TEST_WEBSITE) webpage.url_previous = 'page-3.html' webpage.url_next = 'page-5.html' result = '<nav class="magnetizer-pagination"><ul>' result += '<li><a href="page-3.html" class="magnetizer-previous">Newer posts</a></li>' result += '<li><a href="page-5.html" class="magnetizer-next">Older posts</a></li>' result += '</ul></nav>' assert webpage.pagination_html() == result
def test_pagination_none(): """ Test that webpage.pagination_html() returns None when no pagination needed """ webpage = Webpage(TEST_WEBSITE) assert webpage.pagination_html() is None