def test_watch_movie(movie): user1 = User('Martin', 'pw12345') movie.runtime_minutes = 107 movie1 = Movie("IT", 2017) movie1.runtime_minutes = 146 movie2 = Movie("Joker", 2019) movie2.runtime_minutes = 122 movie3 = "Fake movie" assert repr(user1.watched_movies) == "[]" user1.watch_movie(movie) user1.watch_movie(movie1) user1.watch_movie(movie2) user1.watch_movie(movie3) assert repr( user1.watched_movies ) == "[<Movie Moana, 2016>, <Movie IT, 2017>, <Movie Joker, 2019>]" assert user1.time_spent_watching_movies_minutes == 375
def test_user_watch_movie(): user = User('Martin', 'pw12345') movie1 = Movie('Moana', 2016) movie1.runtime_minutes = 113 movie2 = Movie('Frozen', 2014) movie2.runtime_minutes = 109 user.watch_movie(movie1) assert len(user.watched_movies) == 1 assert user.watched_movies[0] == movie1 assert user.time_spent_watching_movies_minutes == 113 user.watch_movie(movie2) assert len(user.watched_movies) == 2 assert user.watched_movies[1] == movie2 assert user.time_spent_watching_movies_minutes == 222 user.watch_movie(movie1) assert len(user.watched_movies) == 2 assert user.time_spent_watching_movies_minutes == 222