async def main():
    """Initialize aio_manager instace. It's an entry point function."""
    logging.basicConfig(level=logging.WARNING)

    app = await build_application()
    manager = Manager(app)

    sqlalchemy.configure_manager(
        manager,
        app,
        app['declarative_base'],
        url='mysql+pymysql://{usr}:{pwd}@{host}:{port}/{db}'.format(
            usr=app.config['DATABASE_USERNAME'],
            pwd=app.config['DATABASE_PASSWORD'],
            db=app.config['DATABASE_NAME'],
            host=app.config['DATABASE_HOST'],
            port=app.config['DATABASE_PORT']))
    _logger.info('SQLAlchemy manager extention configured.')
    return manager
Beispiel #2
0
import asyncio
import logging

from aiohttp import web
from aio_manager import Manager, Command
from aio_manager.commands.runserver import RunServer as BaseRunServer
import pytest

from app import create_app
from utils import create_database, get_config, drop_database, migrate, get_test_config

app = create_app()
manager = Manager(app)
config = get_config()


class RunTests(Command):
    def run(self, app, args):
        loop = asyncio.get_event_loop()
        test_config = get_test_config()
        loop.run_until_complete(create_database(test_config))
        run_command = './tests'

        if args.test_file:
            run_command += '/' + args.test_file

        pytest.main(['-x', '-s', run_command])

        loop.run_until_complete(drop_database(test_config))

    def configure_parser(self, parser):