Exemplo n.º 1
0
def testRentController():
    """
    Test RentController
    """
    client = ClientsRepository()
    client.add(Client(1, "a", 11111))
    client.add(Client(2, "aa", 12121))

    movie = MoviesRepository()
    movie.add(Movie(3, "b", "bb", "bbb"))
    movie.add(Movie(4, "c", "cc", "ccc"))

    rent = RentRepository()
    unre = UnreController()
    controller = RentController(client, movie, rent, unre)

    controller.rentMovie(Rent(1, 3))
    controller.rentMovie(Rent(2, 3))

    assert controller.findByMovie(4) == []
    assert controller.findByMovie(3) == [1, 2]
    controller.rentMovie(Rent(1, 4))

    assert controller.GetListClients() == [(1, 2), (2, 1)]
    assert controller.GetListMovies() == [(3, 2), (4, 1)]