Exemplo n.º 1
0
def testMovieRepository():
    """
    Test MovieRepository class
    """

    rep = MoviesRepository()
    assert len(rep) == 0
    assert rep.getAll() == []

    movie = Movie(1, "a", "aa", "aaa")
    rep.add(movie)
    assert len(rep) == 1
    assert rep.getAll() == [movie]
    assert rep.getMovie(1) == movie

    try:
        rep.add(movie)
        assert False
    except ExceptionMess:
        pass

    try:
        rep.remove(12)
        assert False
    except ExceptionMess:
        pass

    movie2 = Movie(2, "b", "bb", "bbb")
    rep.add(movie2)
    assert len(rep) == 2
    assert rep.getAll() == [movie, movie2]
    assert rep.getMovie(2) == movie2

    try:
        rep.getMovie(1123)
        assert False
    except ExceptionMess:
        pass

    rep.remove(1)
    assert len(rep) == 1
    assert rep.getAll() == [movie2]
    assert rep.getMovie(2) == movie2

    rep.updateMovie(Movie(2, "z", "zz", "zzz"))
    movie3 = Movie(2, "z", "zz", "zzz")
    assert len(rep) == 1
    assert rep.getAll() == [movie3]
    assert rep.getMovie(2) == movie3