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'
