test(fips): close the database test clients with aclose so the engine stops
Some checks failed
Cost map pins recheck / rerun-cost-map-guard (push) Waiting to run
LiteLLM Rust DB / db-queries (push) Waiting to run
LiteLLM Rust / rust-lint (push) Waiting to run
LiteLLM Rust / rust-test (push) Waiting to run
LiteLLM Rust / rust-wheel (push) Waiting to run
Terraform Provider / gofmt, vet, build, test (push) Has been cancelled
Terraform Provider / Provider endpoints vs proxy OpenAPI schema (push) Has been cancelled

This commit is contained in:
yucheng 2026-10-30 19:59:19 -07:00
parent f770015279
commit ba6f2b840d

View file

@ -68,7 +68,7 @@ async def test_fips_database_check(
with pytest.raises(FipsModeError, match=refusal) if refusal else nullcontext():
await enforce_fips_database(FipsRepository(client), database_is_configured=True)
finally:
await client.disconnect()
await client.aclose()
if marked_after is not None:
assert _marker_rows(database_url) == marked_after
@ -89,5 +89,5 @@ async def test_pods_starting_together_in_fips_mode_on_a_new_database_both_start(
*(enforce_fips_database(FipsRepository(pod), database_is_configured=True) for pod in pods)
)
finally:
await asyncio.gather(*(pod.disconnect() for pod in pods))
await asyncio.gather(*(pod.aclose() for pod in pods))
assert _marker_rows(database_url) == 1