From 886ec0f1332eb4a74e38ed2c415659209947c5e8 Mon Sep 17 00:00:00 2001 From: Oleg Matviichuk Date: Tue, 29 Jul 2025 23:05:12 +0200 Subject: [PATCH] Fix Sqlite tests --- aiogram/fsm/storage/sqlite.py | 6 +++--- tests/conftest.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/aiogram/fsm/storage/sqlite.py b/aiogram/fsm/storage/sqlite.py index c5954973..c5da9917 100644 --- a/aiogram/fsm/storage/sqlite.py +++ b/aiogram/fsm/storage/sqlite.py @@ -32,10 +32,10 @@ class SqliteStorage(BaseStorage): if key_builder is None: key_builder = DefaultKeyBuilder() self._key_builder = key_builder - self.connection = connection + self._connection = connection @classmethod - async def connect(cls, db_filename: str = SQLITE_FILENAME) -> Connection: + async def connect(cls, db_filename: str = SQLITE_FILENAME) -> "SqliteStorage": """ Create an instance of :class:`SqliteStorage` with specifying the DB filename @@ -53,7 +53,7 @@ class SqliteStorage(BaseStorage): return cls(connection=connection) async def close(self) -> None: - await self.connection.close() + await self._connection.close() async def set_state(self, key: StorageKey, state: StateType = None) -> None: pass diff --git a/tests/conftest.py b/tests/conftest.py index 7c4ccead..db3c7a7b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -114,7 +114,7 @@ async def memory_storage(): @pytest.fixture() async def squlite_storage(): - storage = SqliteStorage() + storage = SqliteStorage.connect("aiogram_fsm_test.sqlite") try: yield storage finally: