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 add():
if not current_user.is_authenticated:
flash("Please login")
return redirect(url_for('account.login'))
form = AddEventForm()
datetime1 = datetime.strptime("2019-04-25 22:37", '%Y-%m-%d %H:%M')
form.eventtype.choices = [(t.id, t.event_type) for t in Type.query.all()]
if request.method == 'POST':
v = Venue(venuename=form.venue.data,
locationaddress=form.location.data)
# db.session.bulk_save_objects(multi)
db.session.add(v)
db.session.commit()
print('*******', current_user.id)
e = Event(title=form.title.data,
description=form.description.data,
organizer_id=current_user.id,
venue_id=v.id,
eventtype_id=form.eventtype.data,
end_time=datetime1,
start_time=datetime1,
img_link=form.imglink.data,
event_owner=form.eventowner.data)
db.session.add(e)
db.session.commit()
flash('Successfully added')
return render_template('add_event.html', form=form)
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 event_add_get():
try:
return render_template('items/event_edit.html',
event=Event("", "", "", "", datetime.now(),
datetime.now()).to_json(),
action_type="Add",
event_types=Constants.EVENT_TYPES)
except NoSuchEventExistException:
abort(404)
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 add():
# print("Hello")
form = AddEventForm()
# print(form)
form.organizer.choices = [(u.id, u.email) for u in User.query.all()]
if form.validate_on_submit():
e = Event(description=form.description.data,
user_id=form.organizer.data)
db.session.add(e)
db.session.commit()
return redirect(url_for('ticket'))
return render_template('add_event.html', form=form)
def show_create_event():
if request.method == "POST":
print('create event')
new_events = Event(event_name=request.form['event_name'],
user_id=current_user.id,
location=request.form['event_location'],
address=request.form['event_address'],
event_type=request.form.get('event_dropdown'),
img_pic=request.form['event_img'],
time=request.form['event_date'])
db.session.add(new_events)
db.session.commit()
return redirect(url_for('event.render_event', id=new_events.id))
return render_template('event/create_event.html')
def test_to_json(self):
dt = datetime.now()
id_ = uuid.uuid4()
event = Event("Test", "lk", "virtual", 10, dt, dt, id_)
self.equal = self.assertEqual(
event.to_json(), {
'title': 'Test',
'description': 'lk',
"event_type": "virtual",
"points": 10,
'start': dt,
'end': dt,
'_id': id_
})
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 add():
form = AddForm()
if form.validate_on_submit():
name = form.name.data
description = form.description.data
image_url = form.image_url.data
price = form.price.data
address = form.address.data
time = form.time.data
# Add new Event to database
new_pup = Event(name, description, image_url, price, address, time)
db.session.add(new_pup)
db.session.commit()
return redirect(url_for('events.list'))
return render_template('add.html', form=form)
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 add():
form = AddForm()
formtype = AddType()
if request.method == 'POST':
if form.validate_on_submit():
newevent = Event(
name=form.name.data,
description=form.description.data,
image_url=form.image_url.data,
address=form.address.data,
datetimestart=datetime.strptime(form.datetimestart.data,
"%m/%d/%Y %H:%M"),
created=datetime.now(),
datetimeend=None if not form.datetimeend.data else
datetime.strptime(form.datetimeend.data, "%m/%d/%Y %H:%M"))
current_user.event.append(newevent)
db.session.add(newevent)
db.session.commit()
return redirect(url_for('tickettypes.addtype', id=newevent.id))
else:
for field_name, errors in form.errors.items():
flash(errors)
return render_template('add.html', form=form)
def test_is_not_valid_model(self):
event = Event(12, 12, "virtual", 10, "hello", "sd")
self.assertFalse(event.is_valid_model(), "Invalid model is valid")
def test_factory_form_json(self):
event = Event("Test", "lk", "virtual", 10, datetime.now(),
datetime.now())
self.assertEqual(event, Event.factory_form_json(event.to_json()),
"Creating event object from json failed")
def test_remove_non_existing_event_from_db(self):
event = Event("Test", "lk", "virtual", 10, datetime.now(),
datetime.now(), uuid.uuid4())
self.assertRaises(NoSuchEventExistException, event.remove_from_db())
def test_is_valid_model(self):
event = Event("Test", "lk", "virtual", 10, datetime.now(),
datetime.now(), uuid.uuid4())
self.assertTrue(event.is_valid_model(), "Valid model is invalid")
