Produits Entr'ouvert » Chrono

    def get_context_data(self, **kwargs):

        context = super(DeskDeleteView, self).get_context_data(**kwargs)

        context['cannot_delete'] = Booking.objects.filter(

                event__desk=self.get_object(),

                event__start_datetime__gt=now(),

                cancellation_datetime__isnull=True).exists()

        return context

    def delete(self, request, *args, **kwargs):

        self.object = self.get_object()

        context = self.get_context_data()

        if context['cannot_delete']:

            raise PermissionDenied()

        return super(DeskDeleteView, self).delete(request, *args, **kwargs)

def test_delete_busy_desk(app, admin_user):

    agenda = Agenda(label=u'Foo bar', kind='meetings')

    agenda.save()

    desk_a = Desk.objects.create(agenda=agenda, label='Desk A')

    desk_b = Desk.objects.create(agenda=agenda, label='Desk B')

    event = Event(start_datetime=now() + datetime.timedelta(days=10),

                  places=10, agenda=agenda, desk=desk_a)

    event.save()

    app = login(app)

    resp = app.get('/manage/', status=200)

    resp = resp.click('Foo bar').follow()

    resp = resp.click('Settings')

    desk_page = resp.click('Desk A')

    desk_delete_page = desk_page.click('Delete')

    assert 'Are you sure you want to delete this?' in desk_delete_page.text

    # make sure the deleting is not disabled

    assert 'disabled' not in desk_delete_page.text

    booking = Booking(event=event)

    booking.save()

    resp = desk_page.click('Delete')

    assert 'This cannot be removed' in resp.text

    # the button is disabled

    assert 'disabled' in resp.text

    app.post('/manage/desks/%s/delete' % desk_a.pk, status=403)