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_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