def test_repository_can_retrieve_article_count(session_factory):
repo = SqlAlchemyRepository(session_factory)
number_of_articles = repo.get_number_of_articles()
# Check that the query returned 177 Articles.
assert number_of_articles == 177
def test_repository_can_add_article(session_factory):
repo = SqlAlchemyRepository(session_factory)
number_of_articles = repo.get_number_of_articles()
new_article_id = number_of_articles + 1
article = Article(
date.fromisoformat('2020-03-09'),
'Second US coronavirus cruise tests negative amid delays and cancellations',
'It was revealed ...',
'https://www.nzherald.co.nz/travel/news/article.cfm?c_id=7&objectid=12315024',
'https://www.nzherald.co.nz/resizer/ix7hy3lzkMWUkD8hE6kdZ-8oaOM=/620x349/smart/filters:quality(70)/arc-anglerfish-syd-prod-nzme.s3.amazonaws.com/public/7VFOBLCBCNDHLICBY3CTPFR2L4.jpg',
new_article_id
)
repo.add_article(article)
assert repo.get_article(new_article_id) == article
