diff --git a/cinema/tests.py b/cinema/tests.py index 69a8378328b2901a3a39cca41bf62d7fbf6c4733..daa54f5a64b70c15b8c066aec1958ddc47844110 100644 --- a/cinema/tests.py +++ b/cinema/tests.py @@ -35,11 +35,14 @@ class AddCinemaTests(TestCase): # self.assertEqual(len(Cinema.objects.filter(name='Cinema 2')), 1) # Test: view the list of screens through the 'manage_screens' url - # Expected result: one screen object found + # Expected result: three screen objects found def testManageScreens(self): + cinema = Cinema.objects.get(id=1) + Screen.objects.create(cinema=cinema, screen_number=2, seating_capacity=100) + Screen.objects.create(cinema=cinema, screen_number=3, seating_capacity=100) response = self.client.get(reverse('manage_screens')) self.assertEqual(response.status_code, 200) - self.assertEqual(len(response.context['screens']), 1) + self.assertEqual(len(response.context['screens']), 3) # Test: add a new screen # Expected result: new object added to the system @@ -64,11 +67,14 @@ class AddCinemaTests(TestCase): self.assertEqual(len(Screen.objects.filter()), 0) # Test: view the list of screens through the 'screen_list' url - # Expected result: one screen object found + # Expected result: three screen objects found def testScreenListView(self): + cinema = Cinema.objects.get(id=1) + Screen.objects.create(cinema=cinema, screen_number=2, seating_capacity=100) + Screen.objects.create(cinema=cinema, screen_number=3, seating_capacity=100) response = self.client.get(reverse('screen_list')) self.assertEqual(response.status_code, 200) - self.assertEqual(len(response.context['screens']), 1) + self.assertEqual(len(response.context['screens']), 3) # Test: add a new showing # Expected result: new showing object added to the system @@ -85,11 +91,13 @@ class AddCinemaTests(TestCase): self.assertEqual(len(Film.objects.filter(title="A Film")), 1) # Test: view the list of films - # Expected result: one film object found + # Expected result: three film objects found def testFilmListView(self): + Film.objects.create(title="Film2", length=datetime.timedelta(hours=2), rating=12, genre="Comedy", poster_url="www.example.com/image.webp") + Film.objects.create(title="Film3", length=datetime.timedelta(hours=2), rating=12, genre="Comedy", poster_url="www.example.com/image.webp") response = self.client.post(reverse('film_management')) self.assertEqual(response.status_code, 200) - self.assertEqual(len(response.context['films']), 1) + self.assertEqual(len(response.context['films']), 3) # Test: delete a film # Expected result: film no longer exists @@ -109,11 +117,16 @@ class AddCinemaTests(TestCase): # print(Film.objects.get(id=1).title) # Test: view the list of showings for a film - # Expected result: one showing object found + # Expected result: three showing objects found def testFilmShowings(self): + screen = Screen.objects.get(id=1) + film = Film.objects.get(id=1) + tz = timezone.get_current_timezone() + Showing.objects.create(screen=screen, film=film, start_time=datetime.datetime(2023, 12, 1, 11, 30, 0, tzinfo=tz), available_seats=200) + Showing.objects.create(screen=screen, film=film, start_time=datetime.datetime(2023, 12, 1, 11, 30, 0, tzinfo=tz), available_seats=200) response = self.client.post(reverse('film_showings', args=[1])) self.assertEqual(response.status_code, 200) - self.assertEqual(len(response.context['showings']), 1) + self.assertEqual(len(response.context['showings']), 3) # showings_management @@ -139,11 +152,13 @@ class AddCinemaTests(TestCase): # self.assertEqual(Showing.objects.get(id=1).start_time, newtime) # Test: view the list of users - # Expected result: one user object found + # Expected result: three user objects found def testUserListView(self): + User.objects.create_user(username="test2", password="test") + User.objects.create_user(username="test3", password="test") response = self.client.post(reverse('user_management')) self.assertEqual(response.status_code, 200) - self.assertEqual(len(response.context['users']), 1) + self.assertEqual(len(response.context['users']), 3) # enable_user @@ -161,11 +176,15 @@ class AddCinemaTests(TestCase): # update_ticket - # cancellation_management + # Test: view the list of cancellations + # Expected result: three cancellation objects found def testBookingListView(self): + reservation = Reservation.objects.get(id=1) + Cancelation.objects.create(reservation=reservation) + Cancelation.objects.create(reservation=reservation) response = self.client.post(reverse('cancellation_management')) self.assertEqual(response.status_code, 200) - self.assertEqual(len(response.context['cancellations']), 1) + self.assertEqual(len(response.context['cancellations']), 3) # approve_cancellation