Add backend lifecycle regression tests and fix lifespan cleanup
This commit is contained in:
@@ -121,18 +121,18 @@ async def _lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
|
||||
from app.services import geo_service # noqa: PLC0415
|
||||
|
||||
log.debug("database_directory_ensured", directory=str(db_path.parent))
|
||||
db = await open_db(settings.database_path)
|
||||
startup_db = await open_db(settings.database_path)
|
||||
try:
|
||||
await init_db(db)
|
||||
await geo_service.load_cache_from_db(db)
|
||||
unresolved_count = await geo_service.count_unresolved(db)
|
||||
await init_db(startup_db)
|
||||
await geo_service.load_cache_from_db(startup_db)
|
||||
unresolved_count = await geo_service.count_unresolved(startup_db)
|
||||
from app.services import setup_service # noqa: PLC0415
|
||||
|
||||
setup_complete = await setup_service.is_setup_complete(db)
|
||||
setup_complete = await setup_service.is_setup_complete(startup_db)
|
||||
set_setup_complete_cache(app, setup_complete)
|
||||
log.debug("setup_completion_cached", completed=setup_complete)
|
||||
finally:
|
||||
await db.close()
|
||||
await startup_db.close()
|
||||
|
||||
if unresolved_count > 0:
|
||||
log.warning("geo_cache_unresolved_ips", unresolved=unresolved_count)
|
||||
@@ -172,7 +172,6 @@ async def _lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
|
||||
log.info("bangui_shutting_down")
|
||||
scheduler.shutdown(wait=False)
|
||||
await http_session.close()
|
||||
await db.close()
|
||||
log.info("bangui_shut_down")
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user