def test_event_collection_remove_stale_events(session):
timezone = 'UTC'
events = EventCollection(session)
events.add(
title='Event', timezone=timezone,
start=datetime(2010, 6, 1, 10),
end=datetime(2010, 6, 1, 10) + timedelta(hours=1),
)
assert events.query().count() == 1
max_stale = standardize_date(datetime.today() + timedelta(days=2), 'UTC')
events.remove_stale_events(max_stale=max_stale)
assert events.query().count() == 0
def test_event_collection_pagination(session):
events = EventCollection(session)
assert events.page_index == 0
assert events.pages_count == 0
assert events.batch == []
for year in range(2008, 2011):
for month in range(1, 13):
event = events.add(
title='Event {0}-{1}'.format(year, month),
start=datetime(year, month, 18, 14, 00),
end=datetime(year, month, 18, 16, 00),
timezone='US/Eastern',
tags=['month-{0}'.format(month)],
recurrence='RRULE:FREQ=DAILY;INTERVAL=1;COUNT=4'
)
event.submit()
if year > 2008:
event.publish()
if year > 2009:
event.withdraw()
assert events.query().count() == 3 * 12
events = EventCollection(session, state='initiated')
assert events.subset_count == 0
events = EventCollection(session, state='submitted')
assert events.subset_count == 12
assert all([e.start.year == 2008 for e in events.batch])
assert all([e.start.month < 11 for e in events.batch])
assert len(events.next.batch) == 12 - events.batch_size
assert all([e.start.year == 2008 for e in events.next.batch])
assert all([e.start.month > 10 for e in events.next.batch])
events = EventCollection(session, state='published')
assert events.subset_count == 12
assert all([e.start.year == 2009 for e in events.batch])
assert all([e.start.month < 11 for e in events.batch])
assert len(events.next.batch) == 12 - events.batch_size
assert all([e.start.year == 2009 for e in events.next.batch])
assert all([e.start.month > 10 for e in events.next.batch])
events = EventCollection(session, state='withdrawn')
assert events.subset_count == 12
assert all([e.start.year == 2010 for e in events.batch])
assert all([e.start.month < 11 for e in events.batch])
assert len(events.next.batch) == 12 - events.batch_size
assert all([e.start.year == 2010 for e in events.next.batch])
assert all([e.start.month > 10 for e in events.next.batch])
def test_event_collection(session):
events = EventCollection(session)
assert events.query().count() == 0
event_1 = events.add(
title='Squirrel Park Visit',
start=datetime(2015, 6, 16, 9, 30),
end=datetime(2015, 6, 16, 18, 00),
timezone='US/Eastern',
content={
'description': '<em>Furry</em> things will happen!'
},
location='Squirrel Park',
tags=['fun', 'animals'],
recurrence='RRULE:FREQ=DAILY;INTERVAL=1;COUNT=5'
)
event_1.submit()
event_1.publish()
event_2 = events.add(
title='History of the Squirrel Park',
start=datetime(2015, 6, 18, 14, 00),
end=datetime(2015, 6, 18, 16, 00),
timezone='US/Eastern',
content={
'description': 'Learn how the Park got so <em>furry</em>!'
},
location='Squirrel Park',
tags=['history']
)
event_2.submit()
event_2.publish()
assert events.query().count() == 2
events.delete(event_1)
events.delete(event_2)
assert events.query().count() == 0
assert session.query(Occurrence).count() == 0
def test_export_import(session):
collection = EventCollection(session)
events = []
for year in range(2008, 2011):
for month in range(1, 13):
tags = ['month-{0}'.format(month)]
recurrence = 'RRULE:FREQ=DAILY;INTERVAL=1;COUNT=4'
location = 'location'
meta = {'1': '3', '2': '2'}
content = {'a': 'x', 'b': 'z', 'description': 'description'}
if month == 1:
content = None
meta = {'1': '1'}
if month == 2:
recurrence = ''
content = {'a': 'y'}
if month == 3:
tags = []
content = {'d': 'n'}
if month == 4:
recurrence = None
location = ''
if month == 5:
meta = {}
location = None
if month == 6:
meta = {'4': '7'}
content = {}
if month == 7:
meta = None
event = collection.add(
title='Event {0}-{1}'.format(year, month),
start=datetime(year, month, 18, 14, 00),
end=datetime(year, month, 18, 16, 00),
timezone='US/Eastern',
tags=tags,
recurrence=recurrence,
location=location,
meta=meta,
content=content
)
event.submit()
if year > 2008:
event.publish()
if year > 2009:
event.withdraw()
events.append({
'state': event.state,
'title': event.title,
'localized_start': event.localized_start,
'localized_end': event.localized_end,
'timezone': event.timezone,
'recurrence': event.recurrence or '',
'tags': event.tags,
'location': event.location or '',
'description': event.description,
'occurrence_dates': event.occurrence_dates(),
'meta': event.meta,
'content': event.content,
})
transaction.commit()
output = StringIO()
assert export_csv(collection, output) == 36
output.seek(0)
for event in collection.query():
collection.delete(event)
assert collection.query().count() == 0
import_csv(collection, output)
assert collection.query().count() == 36
events = sorted(events, key=lambda x: x['localized_start'])
for index, event in enumerate(collection.query().order_by(Event.start)):
assert events[index]['state'] == event.state
assert events[index]['title'] == event.title
assert events[index]['localized_start'] == event.localized_start
assert events[index]['localized_end'] == event.localized_end
assert events[index]['timezone'] == event.timezone
assert events[index]['recurrence'] == event.recurrence or ''
assert events[index]['tags'] == event.tags
assert events[index]['location'] == event.location or ''
assert events[index]['description'] == event.description
assert events[index]['occurrence_dates'] == event.occurrence_dates()
assert events[index]['meta'] == event.meta
assert events[index]['content'] == event.content
