Esempio n. 1
0
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
Esempio n. 2
0
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