def test_repository_can_retrieve_genres(session_factory):
repo = SqlAlchemyRepository(session_factory)
genres = repo.get_genres()
assert len(genres) == 14
genre_one = [genre for genre in genres if genre.genre_name == 'Sci-Fi'][0] # Should have 2 matches
genre_two = [genre for genre in genres if genre.genre_name == 'Comedy'][0] # Should have 3 matches
assert genre_one.number_of_classified_movies == 2
assert genre_two.number_of_classified_movies == 3
def test_repository_can_retrieve_the_top_movie_classified_by_a_genre(session_factory):
repo = SqlAlchemyRepository(session_factory)
list_of_genres = repo.get_genres()
action = [genre for genre in list_of_genres if genre.genre_name == "Action"]
action = action[0]
top_action = repo.get_top_movie_by_genre(action)
assert top_action.title == "Guardians of the Galaxy"
fantasy = [genre for genre in list_of_genres if genre.genre_name == "Fantasy"]
fantasy = fantasy[0]
top_fantasy = repo.get_top_movie_by_genre(fantasy)
assert top_fantasy.title == "Suicide Squad"
