def real_db():
"""A fixture to get access to the real mysql db. We need this
to log in to providers like gmail to check that events changes
are synced back."""
engine = main_engine()
session = InboxSession(engine)
yield session
session.rollback()
session.close()
def __init__(self, config, dumpfile):
from inbox.models.session import InboxSession
from inbox.ignition import main_engine
engine = main_engine()
# Set up test database
self.session = InboxSession(engine)
self.engine = engine
self.config = config
self.dumpfile = dumpfile
# Populate with test data
self.populate()
def start():
g.db_session = InboxSession(engine)
g.log = get_logger()
try:
g.namespace = g.db_session.query(Namespace) \
.filter(Namespace.public_id == g.namespace_public_id).one()
g.encoder = APIEncoder(g.namespace.public_id)
except NoResultFound:
return err(404, "Couldn't find namespace with id `{0}` ".format(
g.namespace_public_id))
g.parser = reqparse.RequestParser(argument_class=ValidatableArgument)
g.parser.add_argument('limit', default=DEFAULT_LIMIT, type=limit,
location='args')
g.parser.add_argument('offset', default=0, type=int, location='args')
def start():
g.db_session = InboxSession(engine, ignore_soft_deletes=False)
def db(dbloader):
from inbox.models.session import InboxSession
dbloader.session = InboxSession(dbloader.engine)
yield dbloader
dbloader.session.close()
def new_session(self, ignore_soft_deletes=True):
from inbox.models.session import InboxSession
self.session.close()
self.session = InboxSession(self.engine,
versioned=False,
ignore_soft_deletes=ignore_soft_deletes)
