From ba6f2b840d4d1b529e937edb48a24b06d391f690 Mon Sep 17 00:00:00 2001 From: yucheng Date: Fri, 30 Oct 2026 19:59:19 -0700 Subject: [PATCH] test(fips): close the database test clients with aclose so the engine stops --- tests/integration/database/test_fips_database.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/database/test_fips_database.py b/tests/integration/database/test_fips_database.py index 21f86308fe2..54238bc6876 100644 --- a/tests/integration/database/test_fips_database.py +++ b/tests/integration/database/test_fips_database.py @@ -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