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