def test_create_event_recurring_dtstart(session):
timezone = 'Europe/Zurich'
event = Event(state='initiated')
event.timezone = timezone
event.start = tzdatetime(2008, 2, 7, 10, 15, timezone)
event.end = tzdatetime(2008, 2, 7, 16, 00, timezone)
event.title = 'Squirrel Park Visit'
event.recurrence = ('DTSTART:19970902T090000\n'
'RRULE:FREQ=DAILY;INTERVAL=2;COUNT=5')
session.add(event)
event.submit()
event.publish()
transaction.commit()
occurrences = session.query(Event).one().occurrences
assert len(occurrences) == 5
assert occurrences[0].start == tzdatetime(1997, 9, 2, 9, 0, timezone)
assert occurrences[0].end == tzdatetime(1997, 9, 2, 14, 45, timezone)
assert occurrences[1].start == tzdatetime(1997, 9, 4, 9, 0, timezone)
assert occurrences[1].end == tzdatetime(1997, 9, 4, 14, 45, timezone)
assert occurrences[2].start == tzdatetime(1997, 9, 6, 9, 0, timezone)
assert occurrences[2].end == tzdatetime(1997, 9, 6, 14, 45, timezone)
assert occurrences[3].start == tzdatetime(1997, 9, 8, 9, 0, timezone)
assert occurrences[3].end == tzdatetime(1997, 9, 8, 14, 45, timezone)
assert occurrences[4].start == tzdatetime(1997, 9, 10, 9, 0, timezone)
assert occurrences[4].end == tzdatetime(1997, 9, 10, 14, 45, timezone)
def test_create_event_recurring_limit(session):
timezone = 'Europe/Zurich'
event = Event(state='initiated')
year = datetime.today().year
event.timezone = timezone
event.start = tzdatetime(year, 2, 7, 10, 15, timezone)
event.end = tzdatetime(year, 2, 7, 16, 00, timezone)
event.title = 'Squirrel Park Visit'
event.recurrence = 'RRULE:FREQ=YEARLY;INTERVAL=1;COUNT=5'
session.add(event)
event.submit()
event.publish()
transaction.commit()
occurrences = session.query(Event).one().occurrences
assert len(occurrences) == 2
assert occurrences[0].start == tzdatetime(year, 2, 7, 10, 15, timezone)
assert occurrences[0].end == tzdatetime(year, 2, 7, 16, 00, timezone)
assert occurrences[1].start == tzdatetime(year + 1, 2, 7, 10, 15, timezone)
assert occurrences[1].end == tzdatetime(year + 1, 2, 7, 16, 00, timezone)
def test_as_ical():
url = 'https://example.org/my-event'
event = Event(
state='initiated',
timezone='Europe/Zurich',
start=tzdatetime(2008, 2, 7, 10, 15, 'Europe/Zurich'),
end=tzdatetime(2008, 2, 7, 16, 00, 'Europe/Zurich'),
recurrence='RRULE:FREQ=DAILY;INTERVAL=2;COUNT=5',
title='Squirrel Park Visit',
content={'description': '<em>Furry</em> things will happen!'},
location='Squirrel Park',
tags=['fun', 'animals', 'furry'],
meta={'submitter': '*****@*****.**'},
name='event',
modified=tzdatetime(2008, 2, 7, 10, 15, 'Europe/Zurich'),
)
ical = event.as_ical(url=url).decode().strip().split('\r\n')
assert sorted(ical) == sorted([
'BEGIN:VCALENDAR',
'VERSION:2.0',
'PRODID:-//OneGov//onegov.event//',
'BEGIN:VEVENT',
'SUMMARY:Squirrel Park Visit',
'DTSTART;VALUE=DATE-TIME:20080207T091500Z',
'DTEND;VALUE=DATE-TIME:20080207T150000Z',
'DESCRIPTION:<em>Furry</em> things will happen!',
'LAST-MODIFIED;VALUE=DATE-TIME:20080207T091500Z',
'LOCATION:Squirrel Park',
'RRULE:FREQ=DAILY;COUNT=5;INTERVAL=2',
'URL:https://example.org/my-event',
'END:VEVENT',
'END:VCALENDAR',
])
event.submit()
event.publish()
occurrence = event.occurrences[0]
occurrence.modified = tzdatetime(2008, 2, 7, 10, 15, 'Europe/Zurich')
ical = event.occurrences[0].as_ical(url=url).decode().strip().split('\r\n')
assert sorted(ical) == sorted([
'BEGIN:VCALENDAR',
'VERSION:2.0',
'PRODID:-//OneGov//onegov.event//',
'BEGIN:VEVENT',
'SUMMARY:Squirrel Park Visit',
'DTSTART;VALUE=DATE-TIME:20080207T091500Z',
'DTEND;VALUE=DATE-TIME:20080207T150000Z',
'DESCRIPTION:<em>Furry</em> things will happen!',
'LAST-MODIFIED;VALUE=DATE-TIME:20080207T091500Z',
'LOCATION:Squirrel Park',
'URL:https://example.org/my-event',
'END:VEVENT',
'END:VCALENDAR',
])
def test_create_event(session):
timezone = 'Europe/Zurich'
start = tzdatetime(2008, 2, 7, 10, 15, timezone)
end = tzdatetime(2008, 2, 7, 16, 00, timezone)
title = 'Squirrel Park Visit'
description = '<em>Furry</em> things will happen!'
location = 'Squirrel Park'
tags = ['fun', 'animals', 'furry']
event = Event(state='initiated')
event.timezone = timezone
event.start = start
event.end = end
event.title = title
event.content = {'description': description}
event.location = location
event.tags = tags
event.meta = {'submitter': '*****@*****.**'}
event.name = 'event'
session.add(event)
event.submit()
event.publish()
transaction.commit()
event = session.query(Event).one()
assert event.timezone == timezone
assert event.start == start
assert event.localized_start == start
assert str(event.start.tzinfo) == 'UTC'
assert str(event.localized_start.tzinfo) == timezone
assert event.end == end
assert event.localized_end == end
assert str(event.end.tzinfo) == 'UTC'
assert str(event.localized_end.tzinfo) == timezone
assert event.title == title
assert event.location == location
assert sorted(event.tags) == sorted(tags)
assert event.description == description
assert event.content == {'description': description}
assert event.meta == {'submitter': '*****@*****.**'}
assert event.name == 'event'
occurrence = session.query(Occurrence).one()
assert occurrence.timezone == timezone
assert occurrence.start == start
assert occurrence.localized_start == start
assert str(occurrence.start.tzinfo) == 'UTC'
assert str(occurrence.localized_start.tzinfo) == timezone
assert occurrence.end == end
assert occurrence.localized_end == end
assert str(occurrence.end.tzinfo) == 'UTC'
assert str(occurrence.localized_end.tzinfo) == timezone
assert occurrence.title == title
assert occurrence.location == location
assert sorted(occurrence.tags) == sorted(tags)
assert occurrence.event.description == description
assert occurrence.name == 'event-2008-02-07'
assert [o.id for o in event.occurrences] == [occurrence.id]
assert occurrence.event.id == event.id
def test_delete_event(session):
timezone = 'Europe/Zurich'
event = Event(
state='initiated',
title='event',
start=tzdatetime(2008, 2, 7, 10, 15, timezone),
end=tzdatetime(2008, 2, 7, 16, 00, timezone),
timezone=timezone
)
session.add(event)
session.delete(session.query(Event).one())
transaction.commit()
session.expire_all()
assert session.query(Event).count() == 0
assert session.query(Occurrence).count() == 0
event = Event(
state='initiated',
title='event',
start=tzdatetime(2008, 2, 7, 10, 15, timezone),
end=tzdatetime(2008, 2, 7, 16, 00, timezone),
timezone=timezone
)
session.add(event)
event.submit()
session.delete(session.query(Event).one())
transaction.commit()
session.expire_all()
assert session.query(Event).count() == 0
assert session.query(Occurrence).count() == 0
event = Event(
state='initiated',
title='event',
start=tzdatetime(2008, 2, 7, 10, 15, timezone),
end=tzdatetime(2008, 2, 7, 16, 00, timezone),
timezone=timezone
)
session.add(event)
event.submit()
event.publish()
session.delete(session.query(Event).one())
transaction.commit()
session.expire_all()
assert session.query(Event).count() == 0
assert session.query(Occurrence).count() == 0
event = Event(
state='initiated',
title='event',
start=tzdatetime(2008, 2, 7, 10, 15, timezone),
end=tzdatetime(2008, 2, 7, 16, 00, timezone),
timezone=timezone
)
session.add(event)
event.submit()
event.publish()
event.withdraw()
session.delete(session.query(Event).one())
transaction.commit()
session.expire_all()
assert session.query(Event).count() == 0
assert session.query(Occurrence).count() == 0
def test_update_event(session):
timezone = 'Europe/Zurich'
event = Event(state='initiated')
event.timezone = timezone
event.start = tzdatetime(2008, 2, 7, 10, 15, timezone)
event.end = tzdatetime(2008, 2, 7, 16, 00, timezone)
event.title = 'Squirrel Park Visit'
event.content = {'description': '<em>Furry</em> things will happen!'}
event.location = 'Squirrel Park'
event.tags = ['fun', 'animals', 'furry']
event.name = 'event'
session.add(event)
event.submit()
event.publish()
occurence = session.query(Event).one().occurrences[0]
assert occurence.timezone == event.timezone
assert occurence.start == event.start
assert occurence.end == event.end
assert occurence.event.title == event.title
assert occurence.title == event.title
assert occurence.event.content == event.content
assert occurence.event.description == event.content['description']
assert occurence.event.location == event.location
assert occurence.location == event.location
assert sorted(occurence.event.tags) == sorted(event.tags)
assert sorted(occurence.tags) == sorted(event.tags)
assert occurence.name == 'event-2008-02-07'
event.start = tzdatetime(2009, 2, 7, 10, 15, timezone)
event.end = tzdatetime(2009, 2, 7, 10, 15, timezone)
event.title = 'Squirrel Park Visit - Cancelled'
event.content = {'description': 'No <em>Furry</em> things will happen! :('}
event.location = '-'
event.tags = ['no fun', 'no animals']
assert len(session.query(Event).one().occurrences) == 1
occurence = session.query(Event).one().occurrences[0]
assert occurence.timezone == event.timezone
assert occurence.start == event.start
assert occurence.end == event.end
assert occurence.event.title == event.title
assert occurence.title == event.title
assert occurence.event.content == event.content
assert occurence.event.description == event.content['description']
assert occurence.event.location == event.location
assert occurence.location == event.location
assert sorted(occurence.event.tags) == sorted(event.tags)
assert sorted(occurence.tags) == sorted(event.tags)
event.recurrence = 'RRULE:FREQ=DAILY;INTERVAL=1;COUNT=2'
occurrences = session.query(Event).one().occurrences
assert len(occurrences) == 2
assert occurrences[0].start == tzdatetime(2009, 2, 7, 10, 15, timezone)
assert occurrences[0].end == tzdatetime(2009, 2, 7, 10, 15, timezone)
assert occurrences[1].start == tzdatetime(2009, 2, 8, 10, 15, timezone)
assert occurrences[1].end == tzdatetime(2009, 2, 8, 10, 15, timezone)
event.timezone = 'US/Eastern'
assert session.query(Event).one().timezone == 'US/Eastern'
occurrences = session.query(Event).one().occurrences
assert len(occurrences) == 2
assert occurrences[0].start == tzdatetime(2009, 2, 7, 10, 15, timezone)
assert occurrences[0].end == tzdatetime(2009, 2, 7, 10, 15, timezone)
assert occurrences[0].timezone == 'US/Eastern'
assert occurrences[1].start == tzdatetime(2009, 2, 8, 10, 15, timezone)
assert occurrences[1].end == tzdatetime(2009, 2, 8, 10, 15, timezone)
assert occurrences[1].timezone == 'US/Eastern'
event.name = 'new-event'
assert session.query(Event).one().name == 'new-event'
occurrences = session.query(Event).one().occurrences
assert len(occurrences) == 2
assert occurrences[0].name == 'new-event-2009-02-07'
assert occurrences[1].name == 'new-event-2009-02-08'
event.recurrence = ''
assert session.query(Occurrence).count() == 1
assert len(session.query(Event).one().occurrences) == 1
def test_create_event_recurring(session):
timezone = 'Europe/Zurich'
start = tzdatetime(2008, 2, 7, 10, 15, timezone)
end = tzdatetime(2008, 2, 7, 16, 00, timezone)
title = 'Squirrel Park Visit'
description = '<em>Furry</em> things will happen!'
location = 'Squirrel Park'
tags = ['fun', 'animals', 'furry']
event = Event(state='initiated')
event.timezone = timezone
event.start = start
event.end = end
event.recurrence = 'RRULE:FREQ=DAILY;INTERVAL=2;COUNT=5'
event.title = title
event.content = {'description': description}
event.location = location
event.tags = tags
event.meta = {'submitter': '*****@*****.**'}
event.name = 'event'
session.add(event)
event.submit()
event.publish()
transaction.commit()
event = session.query(Event).one()
assert event.timezone == timezone
assert event.start == start
assert event.localized_start == start
assert str(event.start.tzinfo) == 'UTC'
assert str(event.localized_start.tzinfo) == timezone
assert event.end == end
assert event.localized_end == end
assert str(event.end.tzinfo) == 'UTC'
assert str(event.localized_end.tzinfo) == timezone
assert event.recurrence == 'RRULE:FREQ=DAILY;INTERVAL=2;COUNT=5'
assert event.title == title
assert event.location == location
assert sorted(event.tags) == sorted(tags)
assert event.description == description
assert event.content == {'description': description}
assert event.meta == {'submitter': '*****@*****.**'}
assert event.name == 'event'
occurrences = session.query(Occurrence).all()
assert len(occurrences) == 5
for occurrence in occurrences:
assert occurrence.timezone == timezone
assert str(occurrence.start.tzinfo) == 'UTC'
assert str(occurrence.localized_start.tzinfo) == timezone
assert str(occurrence.end.tzinfo) == 'UTC'
assert str(occurrence.localized_end.tzinfo) == timezone
assert occurrence.title == title
assert occurrence.location == location
assert sorted(occurrence.tags) == sorted(tags)
assert occurrence.event.description == description
assert occurrences[0].start == start
assert occurrences[0].localized_start == start
assert occurrences[0].end == end
assert occurrences[0].localized_end == end
assert occurrences[0].name == 'event-2008-02-07'
start = tzdatetime(2008, 2, 9, 10, 15, timezone)
end = tzdatetime(2008, 2, 9, 16, 00, timezone)
assert occurrences[1].start == start
assert occurrences[1].localized_start == start
assert occurrences[1].end == end
assert occurrences[1].localized_end == end
assert occurrences[1].name == 'event-2008-02-09'
start = tzdatetime(2008, 2, 11, 10, 15, timezone)
end = tzdatetime(2008, 2, 11, 16, 00, timezone)
assert occurrences[2].start == start
assert occurrences[2].localized_start == start
assert occurrences[2].end == end
assert occurrences[2].localized_end == end
assert occurrences[2].name == 'event-2008-02-11'
start = tzdatetime(2008, 2, 13, 10, 15, timezone)
end = tzdatetime(2008, 2, 13, 16, 00, timezone)
assert occurrences[3].start == start
assert occurrences[3].localized_start == start
assert occurrences[3].end == end
assert occurrences[3].localized_end == end
assert occurrences[3].name == 'event-2008-02-13'
start = tzdatetime(2008, 2, 15, 10, 15, timezone)
end = tzdatetime(2008, 2, 15, 16, 00, timezone)
assert occurrences[4].start == start
assert occurrences[4].localized_start == start
assert occurrences[4].end == end
assert occurrences[4].localized_end == end
assert occurrences[4].name == 'event-2008-02-15'
assert (sorted([o.id for o in event.occurrences]) ==
sorted([o.id for o in occurrences]))
assert occurrences[0].event.id == event.id
assert occurrences[1].event.id == event.id
assert occurrences[2].event.id == event.id
assert occurrences[3].event.id == event.id
assert occurrences[4].event.id == event.id
def test_transitions():
event = Event(state='initiated')
assert event.state == 'initiated'
with pytest.raises(AssertionError):
event.publish()
with pytest.raises(AssertionError):
event.withdraw()
event.submit()
assert event.state == 'submitted'
with pytest.raises(AssertionError):
event.submit()
with pytest.raises(AssertionError):
event.withdraw()
event.publish()
assert event.state == 'published'
with pytest.raises(AssertionError):
event.submit()
with pytest.raises(AssertionError):
event.publish()
event.withdraw()
assert event.state == 'withdrawn'
with pytest.raises(AssertionError):
event.submit()
with pytest.raises(AssertionError):
event.withdraw()
event.publish()
assert event.state == 'published'
def test_occurrence_dates(session):
year = datetime.today().year
event = Event(state='initiated')
event.timezone = 'Europe/Zurich'
event.start = tzdatetime(year, 2, 7, 10, 15, 'Europe/Zurich')
event.end = tzdatetime(year, 2, 7, 16, 00, 'Europe/Zurich')
event.recurrence = 'RRULE:FREQ=YEARLY;INTERVAL=1;COUNT=5'
assert len(event.occurrence_dates(limit=False)) == 5
dates = event.occurrence_dates()
assert len(dates) == 2
assert dates[0] == tzdatetime(year, 2, 7, 10, 15, 'Europe/Zurich')
assert dates[1] == tzdatetime(year + 1, 2, 7, 10, 15, 'Europe/Zurich')
assert str(dates[0].tzinfo) == 'UTC'
assert str(dates[1].tzinfo) == 'UTC'
dates = event.occurrence_dates(localize=True)
assert len(dates) == 2
assert dates[0] == tzdatetime(year, 2, 7, 10, 15, 'Europe/Zurich')
assert dates[1] == tzdatetime(year + 1, 2, 7, 10, 15, 'Europe/Zurich')
assert str(dates[0].tzinfo) == 'Europe/Zurich'
assert str(dates[1].tzinfo) == 'Europe/Zurich'
event.title = 'Event'
session.add(event)
transaction.commit()
event = session.query(Event).one()
assert len(event.occurrence_dates(limit=False)) == 5
dates = event.occurrence_dates()
assert len(dates) == 2
assert dates[0] == tzdatetime(year, 2, 7, 10, 15, 'Europe/Zurich')
assert dates[1] == tzdatetime(year + 1, 2, 7, 10, 15, 'Europe/Zurich')
assert str(dates[0].tzinfo) == 'UTC'
assert str(dates[1].tzinfo) == 'UTC'
dates = event.occurrence_dates(localize=True)
assert len(dates) == 2
assert dates[0] == tzdatetime(year, 2, 7, 10, 15, 'Europe/Zurich')
assert dates[1] == tzdatetime(year + 1, 2, 7, 10, 15, 'Europe/Zurich')
assert str(dates[0].tzinfo) == 'Europe/Zurich'
assert str(dates[1].tzinfo) == 'Europe/Zurich'
def occurrences(form):
event = Event()
form.update_model(event)
return [occurrence.date() for occurrence in event.occurrence_dates()]
