def test_movie_world_return_dvd_unsuccessful(self): movie_world = MovieWorld("Test") d = DVD("A", 1, 1254, "February", 10) c = Customer("Pesho", 20, 4) movie_world.add_customer(c) movie_world.add_dvd(d) result = movie_world.return_dvd(4, 1) self.assertEqual(result, "Pesho does not have that DVD")
def test_movie_world_return_dvd_success(self): movie_world = MovieWorld("Test") d = DVD("A", 1, 1254, "February", 10) c = Customer("Pesho", 20, 4) movie_world.add_customer(c) movie_world.add_dvd(d) movie_world.rent_dvd(4, 1) result = movie_world.return_dvd(4, 1) self.assertEqual(result, "Pesho has successfully returned A") self.assertEqual(c.rented_dvds, []) self.assertEqual(d.is_rented, False)
from project.customer import Customer from project.dvd import DVD from project.movie_world import MovieWorld customer = Customer('Atanas', 18, 123) dvd = DVD('CONG: In the Wild', 555, 1998, 'August', 18) dvd2 = DVD.from_date(777, 'CHE', '12.09.1988', 18) # dvd.is_rented = True movie_world = MovieWorld("Gaco") movie_world.add_customer(customer) movie_world.add_dvd(dvd) movie_world.add_dvd(dvd2) print(movie_world.rent_dvd(123, 555)) print(movie_world.rent_dvd(123, 555)) print(movie_world.return_dvd(123, 555)) print(movie_world.return_dvd(123, 555)) print(movie_world.rent_dvd(123, 777)) print(movie_world)