Skip to content

Commit 093d087

Browse files
committed
[FIX] correct lifespan
1 parent 46b8a34 commit 093d087

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

api/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ async def cron_backup_database(db: Database) -> None:
3333

3434
@asynccontextmanager
3535
async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
36-
db = app.dependency_overrides.get(dependencies.db_dep, dependencies.db_dep)()
36+
db = await app.dependency_overrides.get(dependencies.db_dep, dependencies.db_dep)()
3737
assert isinstance(db, Database)
3838
db.create()
3939
task_backup = asyncio.create_task(cron_backup_database(db))

tests/integration/conftest.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,10 +133,13 @@ def override_db_dep(
133133
rds_testing_instance: RDSTestingInstance,
134134
monkeypatch: pytest.MonkeyPatch,
135135
) -> Generator[None, None, None]:
136+
async def _override_db() -> Database:
137+
return db
138+
136139
monkeypatch.setitem(
137140
app.dependency_overrides, # type: ignore
138141
db_dep,
139-
lambda: db,
142+
_override_db,
140143
)
141144
yield
142145
# Cleanup after every test

0 commit comments

Comments
 (0)