def get_events(self):
event = Event()
event.id = 'unique'
event.title = 'Øl øl øl!'
event.notification_sent = False
event.date = datetime.now() + timedelta(days=1)
event.enrollment_date = datetime.now() + timedelta(minutes=10)
time.sleep(5)
return [event]
def to_event(cls, row):
event = Event()
if row['id']:
event.id = row['id']
if row['title']:
event.title = row['title']
if row['notification_sent']:
event.notification_sent = row['notification_sent']
if row['date']:
event.date = row['date']
if row['enrollment_date']:
event.enrollment_date = row['enrollment_date']
return event
def test_notification_status(self):
event = Event()
event.notification_sent = True
assert Mapper.to_row(event)['notification_sent'] == True
