From c1af6cc4fce8a2807ac4b678afb7eeb3762da24a Mon Sep 17 00:00:00 2001 From: Willem Jiang Date: Tue, 26 May 2026 10:56:05 +0800 Subject: [PATCH] Potential fix for pull request finding Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> --- backend/tests/test_checkpointer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/tests/test_checkpointer.py b/backend/tests/test_checkpointer.py index fc8633b98..814322465 100644 --- a/backend/tests/test_checkpointer.py +++ b/backend/tests/test_checkpointer.py @@ -339,11 +339,11 @@ class TestAsyncCheckpointer: mock_saver_cls = MagicMock(return_value=mock_saver) mock_pool_instance = AsyncMock() - mock_pool_instance.check_connection = AsyncMock() + mock_pool_instance.__aenter__.return_value = mock_pool_instance + mock_pool_instance.__aexit__.return_value = False mock_pool_cls = MagicMock(return_value=mock_pool_instance) mock_pool_cls.check_connection = AsyncMock() - mock_dict_row = MagicMock() mock_pg_module = MagicMock()