def testReturnVideo(self):
"""tests the returning of a video"""
video_title = "Braveheart"
video = Video(video_title)
c1 = Customer(first_name, last_name, phone_number, dob, email)
c1.rent_video(video.ID)
self.assertEqual(1, len(c1.rented_video_IDs))
c1.return_video(video.ID)
self.assertEqual(0, len(c1.rented_video_IDs))
def testRentReturnVideo(self):
"""tests the renting and returning of a video (tests duplicate rents and returns)"""
video1 = Video("Braveheart")
video2 = Video("The Matrix")
video3 = Video("Forrest Gump")
c1 = Customer(first_name, last_name, phone_number, dob, email)
c1.rent_video(video1.ID)
c1.rent_video(video2.ID)
c1.rent_video(video3.ID)
self.assertEqual(3, len(c1.rented_video_IDs))
c1.return_video(video2.ID)
self.assertEqual(2, len(c1.rented_video_IDs))
c1.rent_video(video2.ID)
c1.rent_video(video2.ID) # duplicate rental videos (same UUID)
self.assertEqual(3, len(c1.rented_video_IDs))
c1.return_video(video3.ID)
c1.return_video(video3.ID) # duplicate return videos (same UUID)
self.assertEqual(2, len(c1.rented_video_IDs))
