def module_client(setup_database):
# Fill database with values from base db
new_test_db(DATABASE_URI)
app = create_app(testing=True, database_uri=DATABASE_URI)
with app.test_request_context():
with app.test_client() as client:
try:
yield client
finally:
# aborts any pending transactions
db.session.rollback()
# Removes all data from the database
for table in reversed(db.metadata.sorted_tables):
db.session.execute(table.delete())
db.session.commit()
db.engine.dispose()
def setup_database():
# Create database specific for tests
if not database_exists(DATABASE_URI_BASE):
create_database(DATABASE_URI_BASE)
app = create_app(testing=True, database_uri=DATABASE_URI_BASE)
with app.test_request_context():
with app.test_client():
flask_migrate.upgrade()
fill_basic_db(db)
db.engine.dispose()
if not database_exists(DATABASE_URI):
create_database(DATABASE_URI)
try:
yield None
finally:
# Drop database to save container space
if not os.environ.get("SAVE_DB", False):
drop_database(DATABASE_URI)
drop_database(DATABASE_URI_BASE)
"Web API for Data Delivery System" #################################################################################################### # IMPORTS ################################################################################ IMPORTS # #################################################################################################### # Standard library # Installed # Own modules from dds_web import create_app #################################################################################################### # Global variable for Gunicorn ###################################### Global variable for Gunicorn # #################################################################################################### app_obj = create_app()