async def test_create_app_wrong_name(tmpworkdir, event_loop):
mktree(tmpworkdir, SIMPLE_APP)
config = Config(app_path='app.py', app_factory_name='missing')
with pytest.raises(AiohttpDevConfigError) as excinfo:
config.import_app_factory()
assert excinfo.value.args[
0] == "Module 'app.py' does not define a 'missing' attribute/class"
async def test_all_options(tmpdir, test_client, loop, template_engine, session,
database, example):
StartProject(
path=str(tmpdir),
name='foobar',
template_engine=template_engine,
session=session,
database=database,
example=example,
)
assert 'app' in {p.basename for p in tmpdir.listdir()}
style_guide = flake8.get_style_guide()
report = style_guide.check_files([str(tmpdir)])
assert report.total_errors == 0
if database != 'none':
return
config = Config(app_path='app/main.py',
root_path=str(tmpdir),
static_path='.')
app_factory = config.import_app_factory()
app = app_factory()
modify_main_app(app, config)
cli = await test_client(app)
r = await cli.get('/')
assert r.status == 200
text = await r.text()
assert '<title>foobar</title>' in text
async def test_start_run(tmpdir, loop, aiohttp_client, smart_caplog):
StartProject(path=str(tmpdir.join('the-path')), name='foobar')
assert {p.basename for p in tmpdir.listdir()} == {'the-path'}
assert {p.basename
for p in tmpdir.join('the-path').listdir()} == {
'app',
'requirements.txt',
'README.md',
'static',
}
assert """\
adev.main INFO: Starting new aiohttp project "foobar" at "/<tmpdir>/the-path"
adev.main INFO: project created, 13 files generated\n""" == smart_caplog.log.replace(
str(tmpdir), '/<tmpdir>')
config = Config(app_path='the-path/app/',
root_path=str(tmpdir),
static_path='.')
app_factory = config.import_app_factory()
app = await app_factory()
modify_main_app(app, config)
assert isinstance(app, aiohttp.web.Application)
cli = await aiohttp_client(app)
r = await cli.get('/')
assert r.status == 200
text = await r.text()
assert "Success! you've setup a basic aiohttp app." in text
async def test_start_other_dir(tmpdir, loop, aiohttp_client, smart_caplog):
StartProject(path=str(tmpdir.join('the-path')), name='foobar', database=DatabaseChoice.NONE)
assert {p.basename for p in tmpdir.listdir()} == {'the-path'}
assert {p.basename for p in tmpdir.join('the-path').listdir()} == {
'app',
'Makefile',
'requirements.txt',
'README.md',
'activate.settings.sh',
'setup.cfg',
'static',
'tests',
}
assert """\
adev.main INFO: Starting new aiohttp project "foobar" at "/<tmpdir>/the-path"
adev.main INFO: config:
template_engine: jinja
session: secure
database: none
example: message-board
adev.main INFO: project created, 16 files generated\n""" == smart_caplog.log.replace(str(tmpdir), '/<tmpdir>')
config = Config(app_path='the-path/app/', root_path=str(tmpdir), static_path='.')
app_factory = config.import_app_factory()
app = await app_factory()
modify_main_app(app, config)
assert isinstance(app, aiohttp.web.Application)
cli = await aiohttp_client(app)
r = await cli.get('/')
assert r.status == 200
text = await r.text()
assert "Success! you've setup a basic aiohttp app." in text
async def test_db_creation(tmpdir, aiohttp_client, loop):
StartProject(
path=str(tmpdir),
name='foobar postgres test',
template_engine=TemplateChoice.JINJA,
session=SessionChoices.NONE,
database=DatabaseChoice.PG_SA,
example=ExampleChoice.MESSAGE_BOARD,
)
assert 'app' in {p.basename for p in tmpdir.listdir()}
style_guide = flake8.get_style_guide()
report = style_guide.check_files([str(tmpdir)])
assert report.total_errors == 0
db_password = os.getenv('APP_DB_PASSWORD', '')
env = {
'APP_DB_PASSWORD': db_password,
'PATH': os.getenv('PATH', ''),
}
p = subprocess.run(['make', 'reset-database'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
cwd=str(tmpdir), env=env, universal_newlines=True)
assert p.returncode == 0, p.stdout
assert 'creating database "foobar"...'
assert 'creating tables from model definition...'
os.environ['APP_DB_PASSWORD'] = db_password
config = Config(app_path='app/main.py', root_path=str(tmpdir), static_path='.')
app_factory = config.import_app_factory()
app = await app_factory()
modify_main_app(app, config)
cli = await aiohttp_client(app)
r = await cli.get('/')
assert r.status == 200
text = await r.text()
assert '<title>foobar postgres test</title>' in text
async def test_not_app(tmpworkdir):
mktree(tmpworkdir, {
'app.py': """\
def app_factory():
return 123
"""
})
config = Config(app_path='app.py')
with pytest.raises(AiohttpDevConfigError):
await config.load_app(config.import_app_factory())
async def test_serve_main_app(tmpworkdir, loop, mocker):
asyncio.set_event_loop(loop)
mktree(tmpworkdir, SIMPLE_APP)
mock_modify_main_app = mocker.patch('aiohttp_devtools.runserver.serve.modify_main_app')
loop.call_later(0.5, loop.stop)
config = Config(app_path='app.py')
await start_main_app(config, config.import_app_factory(), loop)
mock_modify_main_app.assert_called_with(mock.ANY, config)
async def test_not_app(tmpworkdir):
mktree(tmpworkdir, {'app.py': """\
def app_factory():
return 123
"""})
config = Config(app_path='app.py')
with pytest.raises(
AiohttpDevConfigError,
match=
r"'app_factory' returned 'int' not an aiohttp\.web\.Application"):
await config.load_app(config.import_app_factory())
async def test_run_app_aiohttp_client(tmpworkdir, aiohttp_client):
mktree(tmpworkdir, SIMPLE_APP)
config = Config(app_path='app.py')
app_factory = config.import_app_factory()
app = await config.load_app(app_factory)
modify_main_app(app, config)
assert isinstance(app, aiohttp.web.Application)
cli = await aiohttp_client(app)
r = await cli.get('/')
assert r.status == 200
text = await r.text()
assert text == 'hello world'
async def test_wrong_function_signature(tmpworkdir):
mktree(
tmpworkdir, {
'app.py': """\
def app_factory(foo):
return web.Application()
"""
})
config = Config(app_path='app.py')
with pytest.raises(
AiohttpDevConfigError,
match=r"'app\.py\.app_factory' should not have required arguments"
):
await config.load_app(config.import_app_factory())
async def test_serve_main_app(tmpworkdir, event_loop, mocker):
asyncio.set_event_loop(event_loop)
mktree(tmpworkdir, SIMPLE_APP)
mock_modify_main_app = mocker.patch(
'aiohttp_devtools.runserver.serve.modify_main_app')
event_loop.call_later(0.5, event_loop.stop)
config = Config(app_path="app.py", main_port=0)
runner = await create_main_app(config, config.import_app_factory())
await start_main_app(runner, config.main_port)
mock_modify_main_app.assert_called_with(mock.ANY, config)
await runner.cleanup()
async def test_no_loop_coroutine(tmpworkdir):
mktree(tmpworkdir, {
'app.py': """\
from aiohttp import web
async def hello(request):
return web.Response(text='<h1>hello world</h1>', content_type='text/html')
async def app_factory():
a = web.Application()
a.router.add_get('/', hello)
return a
"""
})
config = Config(app_path='app.py')
app = await config.load_app(config.import_app_factory())
assert isinstance(app, web.Application)
async def test_start_main_app_app_instance(tmpworkdir, loop, mocker):
mktree(tmpworkdir, {
'app.py': """\
from aiohttp import web
async def hello(request):
return web.Response(text='<h1>hello world</h1>', content_type='text/html')
app = web.Application()
app.router.add_get('/', hello)
"""
})
mock_modify_main_app = mocker.patch('aiohttp_devtools.runserver.serve.modify_main_app')
config = Config(app_path='app.py')
await start_main_app(config, config.import_app_factory(), loop)
mock_modify_main_app.assert_called_with(mock.ANY, config)
