From 6328bb3af5a577c99be105b4365c024c0465df4d Mon Sep 17 00:00:00 2001 From: kievzenit Date: Mon, 28 Jul 2025 17:21:19 +0300 Subject: [PATCH] added test for PyMongoStorage that checks if storage could be properly closed --- tests/test_fsm/storage/test_pymongo.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/test_fsm/storage/test_pymongo.py b/tests/test_fsm/storage/test_pymongo.py index 96204bfd..b2071860 100644 --- a/tests/test_fsm/storage/test_pymongo.py +++ b/tests/test_fsm/storage/test_pymongo.py @@ -16,6 +16,14 @@ async def test_get_storage_passing_only_url(mongo_server): pytest.fail(str(e)) +async def test_pymongo_storage_close_does_not_throw(mongo_server): + storage = PyMongoStorage.from_url(url=mongo_server) + try: + assert await storage.close() is None + except Exception as e: + pytest.fail(f"close() raised an exception: {e}") + + async def test_update_not_existing_data_with_empty_dictionary( mongo_storage: PyMongoStorage, storage_key: StorageKey,