def test_reservations(self): for r in Reservation.get_reservations(self.user): self.assertIn(r, [self.reservation1, self.reservation2]) for r in self.user.get_reservations(): self.assertIn(r, [self.reservation1, self.reservation2]) self.assertEqual(Reservation.get_reservation_by_id(self.reservation1.id), \ self.reservation1) self.assertEqual(Reservation.get_reservation_by_id(self.reservation2.id), \ self.reservation2)
def delete_reservation(request): if request.method != 'POST': return redirect('data.views.dashboard_page') reservation_id = request.POST['reservation_id'] reservation = Reservation.get_reservation_by_id(reservation_id) get_current_user(request).remove_reservation(reservation) return redirect('data.views.dashboard_page')