def test_register_or_update(self):
# Creates an event, when there's no event with such name in DB
Event.register_or_update(**self.default_attrs())
self.assertEqual(Event.count(), 1)
created_event = next(Event.query())
self.assertIsInstance(created_event.registered_at, datetime)
self.assertEqual(created_event.name, 'Test event')
# Does not create a new event, if one with current name exists in DB
Event.register_or_update(**self.default_attrs())
self.assertEqual(Event.count(), 1)
def test_register(self):
Event.register(**self.default_attrs())
self.assertEqual(Event.count(), 1)
created_event = next(Event.query())
self.assertIsInstance(created_event.registered_at, datetime)
self.assertEqual(created_event.name, 'Test event')
