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