From dc505b266f6e2b40be645e03680ba3322a8afe97 Mon Sep 17 00:00:00 2001 From: florimondmanca Date: Sun, 12 May 2019 23:14:33 +0200 Subject: [PATCH] add test for lifespan.startup.failed --- tests/middleware/test_lifespan.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/tests/middleware/test_lifespan.py b/tests/middleware/test_lifespan.py index 82246896..4f381c1a 100644 --- a/tests/middleware/test_lifespan.py +++ b/tests/middleware/test_lifespan.py @@ -42,11 +42,22 @@ def test_raise_on_startup(): def run_startup(): raise RuntimeError() - app = Router(routes=[Lifespan(on_startup=run_startup)]) + router = Router(routes=[Lifespan(on_startup=run_startup)]) + async def app(scope, receive, send): + async def _send(message): + nonlocal startup_failed + if message["type"] == "lifespan.startup.failed": + startup_failed = True + return await send(message) + + await router(scope, receive, _send) + + startup_failed = False with pytest.raises(RuntimeError): with TestClient(app): pass # pragma: nocover + assert startup_failed def test_raise_on_shutdown(): -- 2.47.2