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_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_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_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 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 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()})
