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)
