def test_get_movie_by_actor(memory_repo):
movie = Movie('Test Movie', 2020)
movie.add_actor(Actor('Actor1'))
movie.add_actor(Actor('Actor2'))
memory_repo.add_movie(movie)
assert movie in memory_repo.get_movies_by_actor('Actor1')
assert movie in memory_repo.get_movies_by_actor('Actor2')
def read_csv_file(self):
with open(self._data_path, mode='r', encoding='utf-8-sig') as f:
records = csv.DictReader(f)
for record in records:
movie = Movie(record.get('Title'), int(record.get('Year', 0)))
for actor in self._read_field(record, 'Actors', ','):
actor = Actor(actor)
movie.add_actor(actor)
self._dataset_of_actors.add(actor)
for genre in self._read_field(record, 'Genre', ','):
genre = Genre(genre)
movie.add_genre(genre)
self._dataset_of_genres.add(genre)
for director in self._read_field(record, 'Director', ','):
director = Director(director)
movie.set_director(director)
self._dataset_of_directors.add(director)
movie.description = record.get('Description')
movie.runtime_minutes = int(record.get('Runtime (Minutes)'))
self._dataset_of_movies.add(movie)
def memory_repo():
repo = memory_repository.MemoryRepository()
movie_1 = Movie('Movie1', 2000)
movie_2 = Movie('Movie2', 2001)
movie_3 = Movie('Movie3', 2002)
movie_4 = Movie('Movie4', 2003)
movie_5 = Movie('Movie5', 2004)
actor1 = Actor('Actor1')
actor2 = Actor('Actor2')
actor3 = Actor('Actor3')
actor4 = Actor('Actor4')
actor5 = Actor('Actor5')
director1 = Director('Director1')
director2 = Director('Director2')
movie_1.add_actor(actor1)
movie_1.add_actor(actor2)
movie_1.add_actor(actor4)
movie_1.add_actor(actor5)
movie_2.add_actor(actor2)
movie_2.add_actor(actor4)
movie_2.add_actor(actor5)
movie_3.add_actor(actor3)
movie_3.add_actor(actor5)
movie_4.add_actor(actor3)
movie_4.add_actor(actor4)
movie_4.add_actor(actor5)
movie_5.add_actor(actor3)
movie_5.add_actor(actor4)
movie_5.add_actor(actor5)
movie_1.director = director1
movie_2.director = director1
movie_3.director = director1
movie_4.director = director2
movie_5.director = director2
genre1 = Genre('Genre1')
genre2 = Genre('Genre2')
movie_1.add_genre(genre1)
movie_2.add_genre(genre2)
movie_3.add_genre(genre1)
movie_4.add_genre(genre2)
movie_5.add_genre(genre1)
repo.add_movie(movie_1)
repo.add_movie(movie_2)
repo.add_movie(movie_3)
repo.add_movie(movie_4)
repo.add_movie(movie_5)
user = User(username='******', password='******')
repo.add_user(user)
return repo
