From 8567496b77835151ee360023e7b1b6a7be9c636b Mon Sep 17 00:00:00 2001 From: Rishat Fayzullin Date: Wed, 13 Mar 2024 22:39:07 +0300 Subject: [PATCH] Refactoring while review --- aiogram/fsm/storage/mongo.py | 6 +++--- tests/conftest.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/aiogram/fsm/storage/mongo.py b/aiogram/fsm/storage/mongo.py index ca0346ac..44ea0430 100644 --- a/aiogram/fsm/storage/mongo.py +++ b/aiogram/fsm/storage/mongo.py @@ -1,6 +1,6 @@ from typing import Any, Dict, Optional, cast -from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorCollection +from motor.motor_asyncio import AsyncIOMotorClient from aiogram.fsm.state import State from aiogram.fsm.storage.base import ( @@ -33,8 +33,8 @@ class MongoStorage(BaseStorage): if key_builder is None: key_builder = DefaultKeyBuilder() self._client = client - self._db_name = db_name - self._collection: AsyncIOMotorCollection = self._client[db_name][collection_name] + self._database = self._client[db_name] + self._collection = self._database[collection_name] self._key_builder = key_builder @classmethod diff --git a/tests/conftest.py b/tests/conftest.py index 91ee9975..f13fec10 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -86,7 +86,7 @@ async def mongo_storage(mongo_server): pytest.skip(str(e)) else: yield storage - await storage._client.drop_database(storage._db_name) + await storage._client.drop_database(storage._database) finally: await storage.close()