From 52ebb5dcd6c43e5b7f7b242e7510bc74fbc35e5b Mon Sep 17 00:00:00 2001
From: Lancetnik <diementros@yandex.ru>
Date: Tue, 6 Jun 2023 21:42:14 +0300
Subject: [PATCH] increase test coverage

---
 tests/test_router_events.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/tests/test_router_events.py b/tests/test_router_events.py
index 440b4b469..76309f599 100644
--- a/tests/test_router_events.py
+++ b/tests/test_router_events.py
@@ -2,7 +2,7 @@ from contextlib import asynccontextmanager
 from typing import AsyncGenerator, Dict
 
 import pytest
-from fastapi import APIRouter, FastAPI
+from fastapi import APIRouter, FastAPI, Request
 from fastapi.testclient import TestClient
 from pydantic import BaseModel
 
@@ -112,19 +112,19 @@ def test_router_nested_lifespan_state(state: State) -> None:
     @asynccontextmanager
     async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
         state.app_startup = True
-        yield
+        yield {"app": True}
         state.app_shutdown = True
 
     @asynccontextmanager
     async def router_lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
         state.router_startup = True
-        yield
+        yield {"router": True}
         state.router_shutdown = True
     
     @asynccontextmanager
     async def subrouter_lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
         state.sub_router_startup = True
-        yield
+        yield {"sub_router": True}
         state.sub_router_shutdown = True
 
     sub_router = APIRouter(lifespan=subrouter_lifespan)
@@ -136,7 +136,10 @@ def test_router_nested_lifespan_state(state: State) -> None:
     app.include_router(router)
 
     @app.get("/")
-    def main() -> Dict[str, str]:
+    def main(request: Request) -> Dict[str, str]:
+        assert request.state.app
+        assert request.state.router
+        assert request.state.sub_router
         return {"message": "Hello World"}
 
     assert state.app_startup is False