Fix Sqlite tests

This commit is contained in:
Oleg Matviichuk 2025-07-29 23:05:12 +02:00
parent 12e04cd2cf
commit 886ec0f133
2 changed files with 4 additions and 4 deletions

View file

@ -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

View file

@ -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: