def test_repository_returns_date_of_previous_article(session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_article(6) previous_date = repo.get_date_of_previous_article(article) assert previous_date.isoformat() == '2020-03-01'
def test_repository_can_add_a_tag(session_factory): repo = SqlAlchemyRepository(session_factory) tag = Tag('Motoring') repo.add_tag(tag) assert tag in repo.get_tags()
def test_repository_returns_an_empty_list_for_non_existent_tag( session_factory): repo = SqlAlchemyRepository(session_factory) article_ids = repo.get_article_ids_for_tag('United States') assert len(article_ids) == 0
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_returns_an_empty_list_for_non_existent_ids( session_factory): repo = SqlAlchemyRepository(session_factory) articles = repo.get_articles_by_id([0, 199]) assert len(articles) == 0
def test_repository_returns_date_of_next_article(session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_article(3) next_date = repo.get_date_of_next_article(article) assert next_date.isoformat() == '2020-03-05'
def test_repository_returns_none_when_there_are_no_subsequent_articles( session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_article(177) next_date = repo.get_date_of_next_article(article) assert next_date is None
def test_repository_does_not_add_a_comment_without_a_user(session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_article(2) comment = Comment(None, article, "Trump's onto it!", datetime.today()) with pytest.raises(RepositoryException): repo.add_comment(comment)
def test_repository_returns_none_when_there_are_no_previous_articles( session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_article(1) previous_date = repo.get_date_of_previous_article(article) assert previous_date is None
def test_repository_does_not_retrieve_article_for_non_existent_id( session_factory): repo = SqlAlchemyRepository(session_factory) articles = repo.get_articles_by_id([2, 209]) assert len(articles) == 1 assert articles[ 0].title == 'Covid 19 coronavirus: US deaths double in two days, Trump says quarantine not necessary'
def test_repository_can_get_articles_by_ids(session_factory): repo = SqlAlchemyRepository(session_factory) articles = repo.get_articles_by_id([2, 5, 6]) assert len(articles) == 3 assert articles[ 0].title == 'Covid 19 coronavirus: US deaths double in two days, Trump says quarantine not necessary' assert articles[ 1].title == "Australia's first coronavirus fatality as man dies in Perth" assert articles[ 2].title == 'Coronavirus: Death confirmed as six more test positive in NSW'
def test_repository_can_retrieve_articles_by_date(session_factory): repo = SqlAlchemyRepository(session_factory) articles = repo.get_articles_by_date(date(2020, 3, 1)) # Check that the query returned 3 Articles. assert len(articles) == 3 # these articles are no jokes... articles = repo.get_articles_by_date(date(2020, 4, 1)) # Check that the query returned 5 Articles. assert len(articles) == 5
def test_repository_can_retrieve_tags(session_factory): repo = SqlAlchemyRepository(session_factory) tags = repo.get_tags() assert len(tags) == 10 tag_one = [tag for tag in tags if tag.tag_name == 'New Zealand'][0] tag_two = [tag for tag in tags if tag.tag_name == 'Health'][0] tag_three = [tag for tag in tags if tag.tag_name == 'World'][0] tag_four = [tag for tag in tags if tag.tag_name == 'Politics'][0] assert tag_one.number_of_tagged_articles == 53 assert tag_two.number_of_tagged_articles == 2 assert tag_three.number_of_tagged_articles == 64 assert tag_four.number_of_tagged_articles == 1
def test_repository_can_add_a_comment(session_factory): repo = SqlAlchemyRepository(session_factory) user = repo.get_user('thorke') article = repo.get_article(2) comment = make_comment("Trump's onto it!", user, article) repo.add_comment(comment) assert comment in repo.get_comments()
def test_repository_can_add_a_user(session_factory): repo = SqlAlchemyRepository(session_factory) user = User('Dave', '123456789') repo.add_user(user) repo.add_user(User('Martin', '123456789')) user2 = repo.get_user('Dave') assert user2 == user and user2 is user
def test_repository_can_retrieve_article(session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_article(1) # Check that the Article has the expected title. assert article.title == 'Coronavirus: First case of virus in New Zealand' # Check that the Article is commented as expected. comment_one = [ comment for comment in article.comments if comment.comment == 'Oh no, COVID-19 has hit New Zealand' ][0] comment_two = [ comment for comment in article.comments if comment.comment == 'Yeah Freddie, bad news' ][0] assert comment_one.user.username == 'fmercury' assert comment_two.user.username == "thorke" # Check that the Article is tagged as expected. assert article.is_tagged_by(Tag('Health')) assert article.is_tagged_by(Tag('New Zealand'))
def test_can_retrieve_an_article_and_add_a_comment_to_it(session_factory): repo = SqlAlchemyRepository(session_factory) # Fetch Article and User. article = repo.get_article(5) author = repo.get_user('thorke') # Create a new Comment, connecting it to the Article and User. comment = make_comment('First death in Australia', author, article) article_fetched = repo.get_article(5) author_fetched = repo.get_user('thorke') assert comment in article_fetched.comments assert comment in author_fetched.comments
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
def test_repository_does_not_retrieve_an_article_when_there_are_no_articles_for_a_given_date( session_factory): repo = SqlAlchemyRepository(session_factory) articles = repo.get_articles_by_date(date(2020, 3, 8)) assert len(articles) == 0
def test_repository_does_not_retrieve_a_non_existent_user(session_factory): repo = SqlAlchemyRepository(session_factory) user = repo.get_user('prince') assert user is None
def test_repository_can_retrieve_comments(session_factory): repo = SqlAlchemyRepository(session_factory) assert len(repo.get_comments()) == 3
def test_repository_can_get_first_article(session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_first_article() assert article.title == 'Coronavirus: First case of virus in New Zealand'
def test_repository_can_get_last_article(session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_last_article() assert article.title == 'Covid 19 coronavirus: Kiwi mum on the heartbreak of losing her baby in lockdown'
def test_repository_returns_article_ids_for_existing_tag(session_factory): repo = SqlAlchemyRepository(session_factory) article_ids = repo.get_article_ids_for_tag('Health') assert article_ids == [1, 2]
def test_repository_does_not_retrieve_a_non_existent_article(session_factory): repo = SqlAlchemyRepository(session_factory) article = repo.get_article(201) assert article is None
def test_repository_can_retrieve_a_user(session_factory): repo = SqlAlchemyRepository(session_factory) user = repo.get_user('fmercury') assert user == User('fmercury', '8734gfe2058v')