update lifespan to be an async context manager

generator lifespans were deprectated in starlette 0.16.0 https://github.com/encode/starlette/releases/tag/0.16.0
tiangolo · Nov 5, 2021 · f5f1bfc · f5f1bfc
1 parent 524abdc
commit f5f1bfc
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/tests/test_router_events.py b/tests/test_router_events.py
@@ -81,12 +81,15 @@ def sub_router_shutdown():
 
 
 def test_app_lifespan_state(state):
-    def lifespan(app):
-        state.app_startup = True
-        yield
-        state.app_shutdown = True
-
-    app = FastAPI(lifespan=lifespan)
+    class Lifespan:
+        def __init__(self, app):
+            pass
+        async def __aenter__(self):
+            state.app_startup = True
+        async def __aexit__(self, exc_type, exc_value, exc_tb):
+            state.app_shutdown = True
+
+    app = FastAPI(lifespan=Lifespan)
 
     @app.get("/")
     def main():