Using lifespan manager to trigger startup and shutdown events

prefeitura-rio · Oct 18, 2024 · ac5397d · ac5397d
1 parent 11cfb6d
commit ac5397d
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,6 +35,7 @@ qrcode = "^7.4.2"
 pyotp = "^2.9.0"
 fastapi-simple-rate-limiter = "^0.0.4"
 fastapi-limiter = "^0.1.6"
+asgi-lifespan = "^2.1.0"
 
 
 [tool.poetry.group.dev.dependencies]

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -4,6 +4,7 @@
 import random
 from httpx import AsyncClient
 from tortoise import Tortoise
+from asgi_lifespan import LifespanManager
 
 import scripts.database_init_table
 import scripts.create_user
@@ -36,8 +37,9 @@ def event_loop():
 
 @pytest.fixture(scope="session")
 async def client():
-    async with AsyncClient(app=app, base_url="http://test") as client_object:
-        yield client_object
+    async with LifespanManager(app):
+        async with AsyncClient(app=app, base_url="http://test") as client_object:
+            yield client_object
 
 
 @pytest.fixture(scope="session", autouse=True)