Skip to content

Commit 28cc3d8

Browse files
febus982claude
andcommitted
Add test for atexit cleanup handler
Ensures _cleanup_all_managers() is covered by tests, restoring 100% test coverage. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
1 parent 5be12c7 commit 28cc3d8

1 file changed

Lines changed: 15 additions & 0 deletions

File tree

tests/test_sqlalchemy_bind_manager.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,3 +105,18 @@ def test_engine_is_disposed_on_cleanup(multiple_config):
105105

106106
mocked_dispose.assert_called_once()
107107
mocked_async_sync_dispose.assert_called_once()
108+
109+
110+
def test_atexit_cleanup_disposes_all_managers(multiple_config):
111+
"""Test that the atexit handler disposes all tracked manager instances."""
112+
from sqlalchemy_bind_manager._bind_manager import _cleanup_all_managers
113+
114+
sa_manager = SQLAlchemyBindManager(multiple_config)
115+
116+
with patch.object(
117+
sa_manager,
118+
"_dispose_sync",
119+
) as mocked_dispose_sync:
120+
_cleanup_all_managers()
121+
122+
mocked_dispose_sync.assert_called_once()

0 commit comments

Comments
 (0)