async def test_health_check_db_down(web_app_client):
persist.get_database().drop()
resp = await web_app_client.get('/health')
assert resp.status == 500
j = await resp.json()
assert j['code'] == 'unhealthy'
def peewee_moves_cli_proxy(ctx, **kwargs):
class ScriptInfo:
def __init__(self):
self.data = {'manager': None}
ctx.obj = ctx.obj or ScriptInfo()
ctx.obj.data['manager'] = DatabaseManager(database=persist.get_database(),
directory=persist.get_migrations_dir())
def setup_db(models):
settings = {
'backend': 'colibris.persist.backends.SQLiteBackend',
'database': ':memory:',
'regexp_function': True
}
persist.DatabaseBackend.configure(settings)
db = persist.get_database()
db.connect()
persist.models.set_database(db)
db.create_tables(models)
def database(models, **settings):
# Use in-memory SQLite db by default
settings.setdefault('backend',
'colibris.persist.backends.SQLiteBackend')
if settings['backend'] == 'colibris.persist.backends.SQLiteBackend':
settings.setdefault('database', ':memory:')
persist.DatabaseBackend.configure(settings)
db = persist.get_database()
db.connect()
persist.models.set_database(db)
db.create_tables(models)
return db
def pytest_runtest_setup():
colibris.setup()
if persist.is_enabled():
router = Router(persist.get_database(),
migrate_dir=persist.get_migrations_dir())
router.run()
def pytest_runtest_teardown():
if persist.is_created() and not persist.get_database().is_dropped():
persist.get_database().drop()
def execute(self, options):
router = Router(persist.get_database(),
migrate_dir=persist.get_migrations_dir())
router.create(options.name,
auto=importlib.import_module(
settings.PROJECT_PACKAGE_NAME))
def execute(self, options):
router = Router(persist.get_database(),
migrate_dir=persist.get_migrations_dir())
router.run()
def pytest_runtest_setup():
colibris.setup()
if persist.is_enabled():
manager = DatabaseManager(persist.get_database(), directory=persist.get_migrations_dir())
manager.upgrade()