def sqlalchemy_session(sqlalchemy_engine):
session = setup_database(sqlalchemy_engine)
recreate_database(sqlalchemy_engine)
try:
yield session
finally:
session.close()
from app import create_app, setup_database, seed_db
from flask import render_template
def add_vue_routes(app):
@app.route('/')
def serve_vue_app():
"""
Serve our vue app
"""
return(render_template('index.html'))
@app.after_request
def add_header(req):
"""
Clear Cache for hot-reloading
"""
req.headers["Cache-Control"] = "no-cache"
return req
if __name__ == '__main__':
app = create_app()
add_vue_routes(app)
setup_database(app)
seed_db(app)
app.run(debug=True)
event = create_event(verb, publisher, consumer_id=user)
self.events.append(event)
EventProcessor.add_event(event)
sleep(1)
for event in sample(self.events, k=4):
self.events.remove(event)
self.assertTrue(EventProcessor.retract_event(event))
sleep(1)
notifications = list(
EventProcessor.consume('notification', user, limit=10))
self.assertEqual(len(notifications), len(self.events))
for item in notifications:
self.assertTrue(
int(item['item_id']) in list(
map(lambda x: x['item_id'], self.events)))
if __name__ == '__main__':
clear_ns()
setup_system()
setup_workers(2)
setup_database(drop=True)
unittest.main()
