def session_toy(): database_config = read_config(database_config_template()) ramp_config = read_config(ramp_config_template()) try: create_toy_db(database_config, ramp_config) with session_scope(database_config['sqlalchemy']) as session: yield session finally: shutil.rmtree(ramp_config['ramp']['deployment_dir'], ignore_errors=True) db, _ = setup_db(database_config['sqlalchemy']) Model.metadata.drop_all(db)
def make_toy_db(database_connection): database_config = read_config(database_config_template()) ramp_config = ramp_config_template() try: deployment_dir = create_toy_db(database_config, ramp_config) yield finally: shutil.rmtree(deployment_dir, ignore_errors=True) db, _ = setup_db(database_config['sqlalchemy']) Model.metadata.drop_all(db)
def session_toy_aws(database_connection): database_config = read_config(database_config_template()) ramp_config_aws = ramp_aws_config_template() try: deployment_dir = create_toy_db(database_config, ramp_config_aws) with session_scope(database_config['sqlalchemy']) as session: yield session finally: db, _ = setup_db(database_config['sqlalchemy']) Model.metadata.drop_all(db) shutil.rmtree(deployment_dir, ignore_errors=True)
def session_scope_module(): database_config = read_config(database_config_template()) ramp_config = ramp_config_template() try: deployment_dir = create_toy_db(database_config, ramp_config) with session_scope(database_config['sqlalchemy']) as session: _change_state_db(session) yield session finally: shutil.rmtree(deployment_dir, ignore_errors=True) db, _ = setup_db(database_config['sqlalchemy']) Model.metadata.drop_all(db)
def client(): database_config = read_config(database_config_template()) ramp_config = read_config(ramp_config_template()) try: create_toy_db(database_config, ramp_config) flask_config = generate_flask_config(database_config) app = create_app(flask_config) app.config['TESTING'] = True app.config['WTF_CSRF_ENABLED'] = False yield app.test_client() finally: shutil.rmtree(ramp_config['ramp']['deployment_dir'], ignore_errors=True) try: # In case of failure we should close the global flask engine from ramp_frontend import db as db_flask db_flask.session.close() except RuntimeError: pass db, _ = setup_db(database_config['sqlalchemy']) Model.metadata.drop_all(db)
def setup_module(module): database_config = read_config(database_config_template()) ramp_config = read_config(ramp_config_template()) create_toy_db(database_config, ramp_config)
def setup_module(module): database_config = read_config(database_config_template()) ramp_config = ramp_config_template() module.deployment_dir = create_toy_db(database_config, ramp_config)