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(
2014,
'James Gunn',
['Chris Pratt', 'Vin Diesel', 'Bradley Cooper', 'Zoe Saldana'],
121,
8.1,
'Guardians of the Galaxy',
'A group of intergalactic criminals are forced to work together to stop a fanatical warrior from taking control of the universe.',
)
repo.add_article(article)
assert repo.get_article(new_article_id) == article