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')