def test_titles_generates_none_if_no_title(): """Ensure the titles produces None if no title is fetched..""" def body_provider(url): return '<html></html>' def title_provider(body): return None hrefs = ( 'https://www.reddit.com/' 'http://digg.com/', 'https://news.ycombinator.com/', ) titles = tuple( href.titles( hrefs, body_provider=body_provider, title_provider=title_provider, ) ) for title in titles: assert title is None
def test_titles_uses_given_providers(): """Ensure the titles generator uses configurable data providers.""" def body_provider(url): return '<html></html>' def title_provider(body): return 'TEST' hrefs = ( 'https://www.reddit.com/' 'http://digg.com/', 'https://news.ycombinator.com/', ) titles = tuple( href.titles( hrefs, body_provider=body_provider, title_provider=title_provider, ) ) for title in titles: assert title == 'TEST'