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() == '2016-12-14'
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 == 1000
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) == 1
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() == '2017-01-11'
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 not 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) == 2
assert articles[0].title == 'Prometheus'
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_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 not None
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 == 'Prometheus'
assert articles[1].title == "Suicide Squad"
assert articles[2].title == 'The Great Wall'
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) == 0
# 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) == 0
def test_repository_can_retrieve_tags(session_factory):
repo = SqlAlchemyRepository(session_factory)
tags = repo.get_tags()
assert len(tags) == 20
tag_one = [tag for tag in tags if tag.tag_name == 'Action'][0]
tag_two = [tag for tag in tags if tag.tag_name == 'Sci-Fi'][0]
tag_three = [tag for tag in tags if tag.tag_name == 'Mystery'][0]
tag_four = [tag for tag in tags if tag.tag_name == 'Horror'][0]
assert tag_one.number_of_tagged_articles == 303
assert tag_two.number_of_tagged_articles == 120
assert tag_three.number_of_tagged_articles == 106
assert tag_four.number_of_tagged_articles == 119
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):
print(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 == 'Guardians of the Galaxy'
# 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('Action'))
assert article.is_tagged_by(Tag('Adventure'))
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 = Movies(
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
def test_repository_can_get_first_article(session_factory):
repo = SqlAlchemyRepository(session_factory)
article = repo.get_first_article()
assert article.title == 'Guardians of the Galaxy'
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_last_article(session_factory):
repo = SqlAlchemyRepository(session_factory)
article = repo.get_last_article()
assert article.title == 'Nine Lives'
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 == []
def test_repository_does_not_retrieve_a_non_existent_article(session_factory):
repo = SqlAlchemyRepository(session_factory)
article = repo.get_article(201)
assert article is not None
def test_repository_can_retrieve_a_user(session_factory):
repo = SqlAlchemyRepository(session_factory)
user = repo.get_user('fmercury')
assert user == User('fmercury', '8734gfe2058v')
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