def test_repository_returns_date_of_previous_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_movie(6)
previous_date = repo.get_date_of_previous_movie(movie)
assert previous_date.isoformat() == '2015-01-01'
def test_repository_can_add_a_genre(session_factory):
repo = SqlAlchemyRepository(session_factory)
genre = Genre('Motoring')
repo.add_genre(genre)
assert genre in repo.get_genres()
def test_repository_returns_an_empty_list_for_non_existent_genre(
session_factory):
repo = SqlAlchemyRepository(session_factory)
movie_ids = repo.get_movie_ids_for_genre('United States')
assert len(movie_ids) == 0
def test_repository_can_retrieve_movie_count(session_factory):
repo = SqlAlchemyRepository(session_factory)
number_of_movies = repo.get_number_of_movies()
# Check that the query returned 177 Movies.
assert number_of_movies == 1000
def test_repository_returns_an_empty_list_for_non_existent_ids(
session_factory):
repo = SqlAlchemyRepository(session_factory)
movies = repo.get_movies_by_id([0, 199])
assert len(movies) == 1
def test_repository_returns_date_of_next_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_movie(3)
next_date = repo.get_date_of_next_movie(movie)
assert next_date.isoformat() == '2017-01-01'
def test_repository_returns_none_when_there_are_no_previous_movies(
session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_movie(1)
previous_date = repo.get_date_of_previous_movie(movie)
assert previous_date is not 0
def test_repository_does_not_retrieve_movie_for_non_existent_id(
session_factory):
repo = SqlAlchemyRepository(session_factory)
movies = repo.get_movies_by_id([2, 209])
assert len(movies) == 2
assert movies[0].title == 'Prometheus'
def test_repository_does_not_add_a_review_without_a_user(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_movie(2)
review = Review(None, movie, "Trump's onto it!", datetime.today())
with pytest.raises(RepositoryException):
repo.add_review(review)
def test_repository_returns_none_when_there_are_no_subsequent_movies(
session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_movie(177)
next_date = repo.get_date_of_next_movie(movie)
assert next_date is not 0
def test_repository_can_get_movies_by_ids(session_factory):
repo = SqlAlchemyRepository(session_factory)
movies = repo.get_movies_by_id([2, 5, 6])
assert len(movies) == 3
assert movies[0].title == 'Prometheus'
assert movies[1].title == "Suicide Squad"
assert movies[2].title == 'The Great Wall'
def test_repository_can_retrieve_movies_by_date(session_factory):
repo = SqlAlchemyRepository(session_factory)
movies = repo.get_movies_by_date(date(2020, 3, 1))
# Check that the query returned 0 Movies.
assert len(movies) == 0
# these movies are no jokes...
movies = repo.get_movies_by_date(date(2019, 1, 1))
# Check that the query returned 3 Movies.
assert len(movies) == 0
def test_repository_returns_movie_ids_for_existing_genre(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie_ids = repo.get_movie_ids_for_genre('Mystery')
assert movie_ids == [
2, 20, 62, 63, 65, 84, 89, 91, 98, 99, 104, 110, 119, 122, 128, 139,
140, 146, 148, 155, 160, 179, 182, 205, 214, 223, 224, 255, 276, 281,
330, 365, 372, 406, 418, 428, 429, 446, 453, 462, 464, 467, 468, 498,
530, 537, 541, 542, 543, 546, 556, 557, 563, 579, 580, 591, 593, 596,
599, 626, 629, 637, 639, 652, 653, 659, 663, 667, 677, 682, 690, 710,
714, 718, 728, 734, 743, 751, 752, 764, 769, 774, 781, 782, 799, 813,
816, 817, 819, 822, 836, 839, 848, 860, 865, 883, 914, 941, 946, 960,
961, 966, 972, 974, 977, 996
]
def test_repository_can_retrieve_genres(session_factory):
repo = SqlAlchemyRepository(session_factory)
genres = repo.get_genres()
assert len(genres) == 20
genre_one = [genre for genre in genres
if genre.genre_name == 'Adventure'][0]
genre_two = [genre for genre in genres if genre.genre_name == 'Mystery'][0]
genre_three = [genre for genre in genres
if genre.genre_name == 'Horror'][0]
genre_four = [genre for genre in genres
if genre.genre_name == 'Thriller'][0]
assert genre_one.number_of_genreged_movies == 259
assert genre_two.number_of_genreged_movies == 106
assert genre_three.number_of_genreged_movies == 119
assert genre_four.number_of_genreged_movies == 195
def test_repository_can_add_a_review(session_factory):
repo = SqlAlchemyRepository(session_factory)
user = repo.get_user('thorke')
movie = repo.get_movie(2)
review = make_review("Trump's onto it!", user, movie)
repo.add_review(review)
assert review in repo.get_reviews()
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_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_movie(1)
# Check that the Movie has the expected title.
assert movie.title == 'Guardians of the Galaxy'
# Check that the Movie is reviewed as expected.
review_one = [
review for review in movie.reviews
if review.review == 'Oh no, COVID-19 has hit New Zealand'
][0]
review_two = [
review for review in movie.reviews
if review.review == 'Yeah Freddie, bad news'
][0]
assert review_one.user.username == 'fmercury'
assert review_two.user.username == "thorke"
# Check that the Movie is genreged as expected.
assert movie.is_genreged_by(Genre('Adventure'))
assert movie.is_genreged_by(Genre('Adventure'))
def test_repository_can_add_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
number_of_movies = repo.get_number_of_movies()
new_movie_id = number_of_movies + 1
movie = Movie(date.fromisoformat('2020-03-09'), "wa", "ga", "", "img",
5000, "as,df", 124, "asdf", 44, new_movie_id)
repo.add_movie(movie)
assert repo.get_movie(new_movie_id) == movie
def test_can_retrieve_an_movie_and_add_a_review_to_it(session_factory):
repo = SqlAlchemyRepository(session_factory)
# Fetch Movie and User.
movie = repo.get_movie(5)
author = repo.get_user('thorke')
# Create a new Review, connecting it to the Movie and User.
review = make_review('First death in Australia', author, movie)
movie_fetched = repo.get_movie(5)
author_fetched = repo.get_user('thorke')
assert review in movie_fetched.reviews
assert review in author_fetched.reviews
def test_repository_does_not_retrieve_an_movie_when_there_are_no_movies_for_a_given_date(
session_factory):
repo = SqlAlchemyRepository(session_factory)
movies = repo.get_movies_by_date(date(2020, 3, 8))
assert len(movies) == 0
def test_repository_does_not_retrieve_a_non_existent_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_movie(9999)
assert movie is None
def test_repository_can_get_first_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_first_movie()
assert movie.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_reviews(session_factory):
repo = SqlAlchemyRepository(session_factory)
assert len(repo.get_reviews()) == 3
def test_repository_can_get_last_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = repo.get_last_movie()
assert movie.title == 'Nine Lives'
def test_repository_can_retrieve_a_user(session_factory):
repo = SqlAlchemyRepository(session_factory)
user = repo.get_user('fmercury')
assert user == User('fmercury', '8734gfe2058v')