def test_user_attended(self): user = "******" event = Event( title="Test event", description="Test description", start=datetime.datetime.utcnow().strftime('%m/%d/%Y %I:%M %p'), end=(datetime.datetime.utcnow() + datetime.timedelta(hours=1)).strftime('%m/%d/%Y %I:%M %p'), event_type="action", points=10) event.save_to_db() EventRegister.register_for_event(user, event.get_id()) EventRegister.set_user_attended(user, event.get_id().hex) self.assertEquals( EventRegister.get_user_attended(user, event.get_id().hex), True) EventRegister.set_user_attended(user, event.get_id().hex) self.assertEquals( EventRegister.get_user_attended(user, event.get_id().hex), False) Database.remove(Event.COLLECTION, {'_id': event.get_id()})
def test_remove_from_db(self): event = Event("Test", "lk", "virtual", 10, datetime.now(), datetime.now()) event.save_to_db() try: event.remove_from_db() except NoSuchEventExistException: self.fail("Error occurred when tried to delete existing event")
def test_is_synced(self): event = Event("Test", "lk", "virtual", 10, datetime.now(), datetime.now()) event.save_to_db() event.set_title("TestUpdated") event.sync_to_db() self.assertTrue(event.is_synced(), "event marked un-synced when it is") event.remove_from_db()
def test_save_to_db(self): event = Event("Test", "lk", "virtual", 10, datetime.now(), datetime.now()) event.save_to_db() try: test_result = Event.get_by_id(event.get_id()) except NoSuchEventExistException: self.fail() event.remove_from_db() self.assertEqual(test_result, event, "Saved and retrieved event is not the same")
def test_sync_to_db(self): event = Event("Test", "lk", "virtual", 10, datetime.now(), datetime.now()) event.save_to_db() event.set_title("TestUpdated") event.sync_to_db() try: test_result = Event.get_by_id(event.get_id()) except NoSuchEventExistException: self.fail() event.remove_from_db() self.assertEqual(test_result, event, "Sync event with database failed")
def event_add_post(): try: start = datetime.strptime(request.form.get('start'), '%m/%d/%Y %I:%M %p') end = datetime.strptime(request.form.get('end'), '%m/%d/%Y %I:%M %p') new_event = Event(request.form.get('title'), request.form.get('description'), request.form.get('event_type'), int(request.form.get('points')), start, end) if not new_event.is_valid_model(): abort(500) new_event.save_to_db() return "Done" except ValueError: abort(500)
def test_event_register(self): user = "******" event = Event(title="Test event", description="Test description", start=datetime.datetime.utcnow().strftime('%m/%d/%Y %I:%M %p'), end=(datetime.datetime.utcnow() + datetime.timedelta(hours=1)).strftime('%m/%d/%Y %I:%M %p'), event_type="action", points=10) event.save_to_db() EventRegister.register_for_event(user, event.get_id()) self.assertIsNotNone(EventRegister.check_if_registered(user, event.get_id())) Database.remove("registrations", {"user": user, "event": event.get_id()}) Database.remove(Event.COLLECTION, {'_id': event.get_id()})
def test_event_unregister(self): user = "******" event = Event( title="Test event", description="Test description", start=datetime.datetime.utcnow().strftime('%m/%d/%Y %I:%M %p'), end=(datetime.datetime.utcnow() + datetime.timedelta(hours=1)).strftime('%m/%d/%Y %I:%M %p'), event_type="action", points=10) event.save_to_db() EventRegister.register_for_event(user, event.get_id()) EventRegister.unregister_for_event(user, event.get_id()) self.assertIsNone( EventRegister.check_if_registered(user, event.get_id())) Database.remove(Event.COLLECTION, {'_id': event.get_id()})
def test_user_attended(self): user = "******" event = Event(title="Test event", description="Test description", start=datetime.datetime.utcnow().strftime('%m/%d/%Y %I:%M %p'), end=(datetime.datetime.utcnow() + datetime.timedelta(hours=1)).strftime('%m/%d/%Y %I:%M %p'), event_type="action", points=10) event.save_to_db() EventRegister.register_for_event(user, event.get_id()) EventRegister.set_user_attended(user, event.get_id().hex) self.assertEquals(EventRegister.get_user_attended(user, event.get_id().hex), True) EventRegister.set_user_attended(user, event.get_id().hex) self.assertEquals(EventRegister.get_user_attended(user, event.get_id().hex), False) Database.remove(Event.COLLECTION, {'_id': event.get_id()})