def test_set_director():
movie1 = Movie("Moana", 2016)
director = Director("Director")
movie1.director = director
assert movie1.director == director
director_invalid = "Director"
movie1.director = director_invalid
assert movie1.director == director
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
def test_get_movie_by_director(memory_repo):
movie = Movie('Test Movie', 2020)
movie.director = Director('Director1')
memory_repo.add_movie(movie)
assert movie in memory_repo.get_movies_by_director('Director1')
