def create_event(
cls, title, status, event_type="Conference", event_category="Music", id=None
):
"""Return an instance of the Event class."""
event = Event(
title=title,
description="The best description ever",
event_type=EventType(name=event_type),
event_category=EventCategory(name=event_category),
)
if status.lower() in EventStatus.LIVE:
event.published = True
event.start_datetime = datetime.now()
event.end_datetime = datetime.now() + timedelta(days=1)
elif status.lower() == EventStatus.PAST:
event.published = True
event.start_datetime = datetime.now() - timedelta(days=3)
event.end_datetime = datetime.now() - timedelta(days=1)
elif status.lower() == EventStatus.DRAFT:
event.published = False
start_datetime = (datetime.now(),)
end_datetime = datetime.now() + timedelta(days=1)
if (
id is not None
): # used for elasticsearch tests where the model isn't added to the db
event.id = id
return event
def test_event_end_datetimes_none_with_incomplete_data(self):
"""Test that when events are created without both of end_date and
end_time, end_datetime() returns None.
"""
e = Event(title=self.TITLE, creator=self.USER,
end_date=self.END.date())
self.assertIsNone(e.end_datetime())
f = Event(title=self.TITLE, creator=self.USER,
end_time=self.END.time())
self.assertIsNone(f.end_datetime())
def test_event_end_datetimes_none_with_incomplete_data(self):
"""Test that when events are created without both of end_date and
end_time, end_datetime() returns None.
"""
e = Event(title=self.TITLE,
creator=self.USER,
end_date=self.END.date())
self.assertIsNone(e.end_datetime())
f = Event(title=self.TITLE,
creator=self.USER,
end_time=self.END.time())
self.assertIsNone(f.end_datetime())
def test_event_end_datetimes(self):
"""Test that when events are created with both end_date and
end_time, end_datetime() is their value combined.
"""
e = Event(title=self.TITLE, creator=self.USER,
end_date=self.END.date(),
end_time=self.END.time())
self.assertEqual(e.end_datetime(), self.END)
def test_event_end_datetimes(self):
"""Test that when events are created with both end_date and
end_time, end_datetime() is their value combined.
"""
e = Event(title=self.TITLE,
creator=self.USER,
end_date=self.END.date(),
end_time=self.END.time())
self.assertEqual(e.end_datetime(), self.END)
