sanic/tests/test_auto_reload.py

55 lines
1.6 KiB
Python

import os
import subprocess
import secrets
import sys
from tempfile import TemporaryDirectory
from textwrap import dedent
import pytest
def write_app(filename, **runargs):
text = secrets.token_urlsafe()
with open(filename, "w") as f:
f.write(dedent(f"""\
from sanic import Sanic
from sanic.response import text
app = Sanic(__name__)
started = False
@app.listener("after_server_start")
def complete(*args):
global started
started = True
print("complete", {text!r})
try:
app.run(**{runargs!r})
except Exception as e:
print("complete", repr(e))
else:
if not started:
print("complete DID NOT START")
"""
))
return text
@pytest.mark.parametrize("runargs,port", [
dict(port=42102, auto_reload=True),
dict(port=42103, debug=True),
])
async def test_reloader_live(runargs):
with TemporaryDirectory() as tmpdir:
filename = os.path.join(tmpdir, "reloaderapp.py")
text = write_app(filename, **runargs)
with subprocess.Popen([sys.executable, filename], stdout=subprocess.PIPE) as proc:
line = (l.decode().strip() for l in proc.stdout if l.startswith(b"complete"))
try:
assert next(line) == f"complete {text}"
# Edit source code and try again
text = write_app(filename)
assert next(line) == f"complete {text}"
finally:
proc.terminate()