def test_movie_world_rent_dvd_already_rented_by_user(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.rent_dvd(4, 1)
self.assertEqual(result, "Pesho has already rented A")
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)
def test_movie_world_repr(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)
actual = repr(movie_world).strip('\n')
expected = "4: Pesho of age 20 has 1 rented DVD's (A)\n1: A (February 1254) has age restriction 10. Status: rented"
self.assertEqual(actual, expected)
def test_movie_world_rent_dvd_no_permision(self):
movie_world = MovieWorld("Test")
d = DVD("A", 1, 1254, "February", 18)
c = Customer("Pesho", 16, 4)
movie_world.add_customer(c)
movie_world.add_dvd(d)
result = movie_world.rent_dvd(4, 1)
self.assertEqual(result, "Pesho should be at least 18 to rent this movie")
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_rent_dvd_already_rented(self):
movie_world = MovieWorld("Test")
d = DVD("A", 1, 1254, "February", 10)
c = Customer("Pesho", 20, 4)
c2 = Customer("Gosho", 26, 2)
movie_world.add_customer(c)
movie_world.add_customer(c2)
movie_world.add_dvd(d)
movie_world.rent_dvd(4, 1)
result = movie_world.rent_dvd(2, 1)
self.assertEqual(result, "DVD is already rented")
def test_movie_world_add_dvd_overflow(self):
movie_world = MovieWorld("Test")
for _ in range(16):
movie_world.add_dvd(DVD("A", 1, 1254, "February", 10))
self.assertEqual(len(movie_world.dvds), 15)
def test_movie_world_add_dvd_success(self):
movie_world = MovieWorld("Test")
d = DVD("A", 1, 1254, "February", 10)
movie_world.add_dvd(d)
self.assertEqual(movie_world.dvds, [d])
def test_movie_world_add_customer_overflow(self):
movie_world = MovieWorld("Test")
for _ in range(11):
movie_world.add_customer(Customer("Pesho", 20, 4))
self.assertEqual(len(movie_world.customers), 10)
def test_movie_world_add_customer_success(self):
movie_world = MovieWorld("Test")
c = Customer("Pesho", 20, 4)
movie_world.add_customer(c)
self.assertEqual(movie_world.customers, [c])
def tests_movie_world_customer_capacity(self):
self.assertEqual(MovieWorld.customer_capacity(), 10)
def tests_movie_world_dvd_capacity(self):
self.assertEqual(MovieWorld.dvd_capacity(), 15)
def test_movie_init(self):
movie = MovieWorld("Test")
self.assertEqual(movie.name, "Test")
self.assertEqual(movie.customers, [])
self.assertEqual(movie.dvds, [])