def app(base_app):
"""Flask application fixture.
Scope: function
This fixture offers a Flask application with already a database connection
and all the models created. When finished it will delete all models.
.. code-block:: python
def create_ninja_turtle()
with app.test_client() as client:
somedata = 'ninja turtle'
res = client.post(url_for('api.create_object'),
content_type='application/json',
data=json.dumps(somedata))
assert res.status_code == 200
"""
from reana_db.database import Session
from reana_db.models import Base, Resource
engine = create_engine(base_app.config["SQLALCHEMY_DATABASE_URI"])
base_app.session.bind = engine
with base_app.app_context():
if not engine.dialect.has_schema(engine, "__reana"):
engine.execute(CreateSchema("__reana"))
if not database_exists(engine.url):
create_database(engine.url)
Base.metadata.create_all(bind=engine)
Resource.initialise_default_resources()
yield base_app
Session.close() # close hanging connections
Base.metadata.drop_all(bind=engine)
def session():
"""Create a SQL Alchemy session.
Scope: function
This fixture offers a SQLAlchemy session which has been created from the
``db_engine`` fixture.
.. code-block:: python
from reana_db.models import Workflow
def test_create_workflow(session):
workflow = Workflow(...)
session.add(workflow)
session.commit()
"""
from reana_db.database import Session
yield Session
Session.close()