def test_app_lifespan():
startup_complete = False
cleanup_complete = False
app = Starlette()
@app.on_event("startup")
def run_startup():
nonlocal startup_complete
startup_complete = True
@app.on_event("shutdown")
def run_cleanup():
nonlocal cleanup_complete
cleanup_complete = True
assert not startup_complete
assert not cleanup_complete
with LifespanContext(app):
assert startup_complete
assert not cleanup_complete
assert startup_complete
assert cleanup_complete
def test_lifespan_handler():
startup_complete = False
cleanup_complete = False
handler = LifespanHandler()
@handler.on_event("startup")
def run_startup():
nonlocal startup_complete
startup_complete = True
@handler.on_event("shutdown")
def run_cleanup():
nonlocal cleanup_complete
cleanup_complete = True
assert not startup_complete
assert not cleanup_complete
with LifespanContext(handler):
assert startup_complete
assert not cleanup_complete
assert startup_complete
assert cleanup_complete
def test_app_add_event_handler():
startup_complete = False
cleanup_complete = False
app = Starlette()
def run_startup():
nonlocal startup_complete
startup_complete = True
def run_cleanup():
nonlocal cleanup_complete
cleanup_complete = True
app.add_event_handler("startup", run_startup)
app.add_event_handler("shutdown", run_cleanup)
assert not startup_complete
assert not cleanup_complete
with LifespanContext(app):
assert startup_complete
assert not cleanup_complete
assert startup_complete
assert cleanup_complete