Produits Entr'ouvert » Chrono

from django.utils.timezone import make_aware, now

                                   TimePeriodException, Booking)

    def clean(self):

        cleaned_data = super(TimePeriodForm, self).clean()

        if self.has_booking_out_of_bounds():

            raise ValidationError(_('One or several bookings exist within this time period.'))

        return cleaned_data

    def has_booking_out_of_bounds(self):

        if not self.instance.desk_id:

            return False

        bookings = Booking.objects.filter(

            event__start_datetime__gte=now(),

            event__desk=self.instance.desk, cancellation_datetime__isnull=True).select_related('event')

        for booking in bookings:

            event_startdt = booking.event.start_datetime

            event_enddt = booking.event.end_datetime

            if event_startdt.weekday() != self.instance.weekday:

                continue

            start = make_aware(datetime.datetime.combine(event_startdt, self.cleaned_data['start_time']))

            end = make_aware(datetime.datetime.combine(event_enddt, self.cleaned_data['end_time']))

            if (event_startdt < start) or (event_enddt) > end:

                return True

            if self.cleaned_data['weekday'] != self.instance.weekday:

                return True

        return False

@mock.patch('chrono.manager.forms.now', lambda: datetime.datetime(2017, 5, 20, 10, 23))

def test_prevent_time_period_changes_when_bookings_exist(app, admin_user):

    agenda = Agenda.objects.create(label='Foo bar', kind='meetings')

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

    meeting_type = MeetingType(agenda=agenda, label='Blah', duration=30)

    meeting_type.save()

    timeperiod = TimePeriod.objects.create(

        weekday=1, desk=desk, start_time=datetime.time(8, 00), end_time=datetime.time(11, 0))

    event = Event.objects.create(agenda=agenda, places=1, desk=desk, meeting_type=meeting_type,

                                 start_datetime=make_aware(datetime.datetime(2017, 5, 22, 8, 30)))

    Booking.objects.create(event=event)

    event = Event.objects.create(agenda=agenda, places=1, desk=desk, meeting_type=meeting_type,

                                 start_datetime=make_aware(datetime.datetime(2017, 5, 23, 8, 30)))

    Booking.objects.create(event=event)

    login(app)

    timeperiod_edit_link = "/manage/timeperiods/%d/edit" % timeperiod.pk

    # change time period weekday

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['weekday'] = '2'

    resp = resp.form.submit()

    error_message = resp.html.find('ul', {'class': 'errorlist'}).text.strip()

    assert error_message == 'One or several bookings exist within this time period.'

    # extend time period start

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['start_time'] = '7:30'

    resp = resp.form.submit().follow()

    assert resp.html.find('a', {'href': timeperiod_edit_link}).text.strip() == u'Tuesday / 7:30 a.m. \u2192 11 a.m.'

    # shorten time period start at booking's bound

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['start_time'] = '8:30'

    resp = resp.form.submit().follow()

    assert resp.html.find('a', {'href': timeperiod_edit_link}).text.strip() == u'Tuesday / 8:30 a.m. \u2192 11 a.m.'

    # shorten time period start

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['start_time'] = '9:30'

    resp = resp.form.submit()

    error_message = resp.html.find('ul', {'class': 'errorlist'}).text.strip()

    assert error_message == 'One or several bookings exist within this time period.'

    # extend time period end

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['end_time'] = '12:30'

    resp = resp.form.submit().follow()

    assert resp.html.find('a', {'href': timeperiod_edit_link}).text.strip() == u'Tuesday / 8:30 a.m. \u2192 12:30 p.m.'

    event = Event.objects.create(agenda=agenda, places=1, desk=desk, meeting_type=meeting_type,

                                 start_datetime=make_aware(datetime.datetime(2017, 5, 23, 11, 30)))

    Booking.objects.create(event=event)

    # shorten time period end at booking's bound

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['end_time'] = '12:00'

    resp = resp.form.submit().follow()

    assert resp.html.find('a', {'href': timeperiod_edit_link}).text.strip() == u'Tuesday / 8:30 a.m. \u2192 noon'

    # shorten time period end

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['end_time'] = '11:00'

    resp = resp.form.submit()

    error_message = resp.html.find('ul', {'class': 'errorlist'}).text.strip()

    assert error_message == 'One or several bookings exist within this time period.'

    # extend both time period start and end

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['start_time'] = '7:00'

    resp.form['end_time'] = '12:30'

    resp = resp.form.submit().follow()

    assert resp.html.find('a', {'href': timeperiod_edit_link}).text.strip() == u'Tuesday / 7 a.m. \u2192 12:30 p.m.'

    # shorten both time period start and end

    Booking.objects.create(event=event)

    resp = app.get('/manage/agendas/%d/' % agenda.pk)

    resp = resp.click(href=timeperiod_edit_link)

    resp.form['start_time'] = '8:00'

    resp.form['end_time'] = '11:00'

    resp = resp.form.submit()

    error_message = resp.html.find('ul', {'class': 'errorlist'}).text.strip()

    assert error_message == 'One or several bookings exist within this time period.'