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')