def main():
database_config = read_config(database_config_template(),
filter_section='sqlalchemy')
setup_db(database_config)
render_er(
"{}://{}:{}@{}:{}/{}".format(database_config['drivername'],
database_config['username'],
database_config['password'],
database_config['host'],
database_config['port'],
database_config['database']),
os.path.join('..', '_static', 'img', 'schema_db.png'))
def teardown_function(function):
database_config = read_config(database_config_template())
# FIXME: we are recreating the deployment directory but it should be
# replaced by an temporary creation of folder.
shutil.rmtree(function.deployment_dir, ignore_errors=True)
db, _ = setup_db(database_config['sqlalchemy'])
Model.metadata.drop_all(db)
def test_setup_db(database):
database_config = read_config(database_config_template(),
filter_section='sqlalchemy')
db, Session = setup_db(database_config)
with db.connect() as conn:
session = Session(bind=conn)
file_type = session.query(SubmissionFileType).all()
assert len(file_type) > 0
def session_scope_function(database_config, ramp_config, database_connection):
try:
deployment_dir = create_test_db(database_config, ramp_config)
with session_scope(database_config['sqlalchemy']) as session:
yield session
finally:
shutil.rmtree(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_scope_function():
database_config = read_config(database_config_template())
ramp_config = read_config(ramp_config_template())
try:
yield
finally:
shutil.rmtree(ramp_config['ramp']['deployment_dir'],
ignore_errors=True)
db, _ = setup_db(database_config['sqlalchemy'])
Model.metadata.drop_all(db)
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:
yield session
finally:
shutil.rmtree(deployment_dir, ignore_errors=True)
db, _ = setup_db(database_config['sqlalchemy'])
Model.metadata.drop_all(db)
def base_db():
database_config = read_config(database_config_template())
ramp_config = read_config(ramp_config_template())
try:
create_test_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 session_scope_function(database_connection):
database_config = read_config(database_config_template())
ramp_config = read_config(ramp_config_template())
try:
yield
finally:
# FIXME: we are recreating the deployment directory but it should be
# replaced by an temporary creation of folder.
deployment_dir = os.path.commonpath(
[ramp_config['ramp']['kit_dir'], ramp_config['ramp']['data_dir']])
shutil.rmtree(deployment_dir, ignore_errors=True)
db, _ = setup_db(database_config['sqlalchemy'])
Model.metadata.drop_all(db)
def session_scope_function():
database_config = read_config(database_config_template())
ramp_config = ramp_config_template()
try:
deployment_dir = create_test_db(database_config, ramp_config)
with session_scope(database_config['sqlalchemy']) as session:
add_users(session)
add_problems(session)
add_events(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 = ramp_config_template()
try:
deployment_dir = 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(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 teardown_module(module):
database_config = read_config(database_config_template())
ramp_config = read_config(ramp_config_template())
shutil.rmtree(ramp_config['ramp']['deployment_dir'], ignore_errors=True)
db, _ = setup_db(database_config['sqlalchemy'])
Model.metadata.drop_all(db)
