mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-15 20:02:22 +00:00
1ef7655fd7
* Mongo storage included to storages test * Added few additional checks in storages test * Added MongoStorage for FSM * Added changes description * Fixed error message syntax Co-authored-by: Alex Root Junior <jroot.junior@gmail.com> * Resolved mypy check error * IF/ELSE statement simplified * Fix ruff linter error: RET505 Unnecessary `elif` after `return` statement * Fix ruff linter error: E501 Line too long (100 > 99) * Added mongo storage testing in CI * Refactoring while review * Refactoring while review * Storing FSM state and data together in MongoDB-storage * Fix CI - MongoDB container action is only supported on Linux * Refactoring while review * Enable Macos in pypy-tests section of CI * Refactoring while review * Makefile updated * redis and mongo storages tests do not run in pypy-tests job of CI * Fix docstring of DefaultKeyBuilder --------- Co-authored-by: Alex Root Junior <jroot.junior@gmail.com>
59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
import pytest
|
|
|
|
from aiogram.fsm.storage.base import BaseStorage, StorageKey
|
|
from tests.mocked_bot import MockedBot
|
|
|
|
|
|
@pytest.fixture(name="storage_key")
|
|
def create_storage_key(bot: MockedBot):
|
|
return StorageKey(chat_id=-42, user_id=42, bot_id=bot.id)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"storage",
|
|
[
|
|
pytest.lazy_fixture("redis_storage"),
|
|
pytest.lazy_fixture("mongo_storage"),
|
|
pytest.lazy_fixture("memory_storage"),
|
|
],
|
|
)
|
|
class TestStorages:
|
|
async def test_set_state(self, bot: MockedBot, storage: BaseStorage, storage_key: StorageKey):
|
|
assert await storage.get_state(key=storage_key) is None
|
|
|
|
await storage.set_state(key=storage_key, state="state")
|
|
assert await storage.get_state(key=storage_key) == "state"
|
|
await storage.set_state(key=storage_key, state=None)
|
|
assert await storage.get_state(key=storage_key) is None
|
|
|
|
async def test_set_data(self, bot: MockedBot, storage: BaseStorage, storage_key: StorageKey):
|
|
assert await storage.get_data(key=storage_key) == {}
|
|
|
|
await storage.set_data(key=storage_key, data={"foo": "bar"})
|
|
assert await storage.get_data(key=storage_key) == {"foo": "bar"}
|
|
await storage.set_data(key=storage_key, data={})
|
|
assert await storage.get_data(key=storage_key) == {}
|
|
|
|
async def test_update_data(
|
|
self, bot: MockedBot, storage: BaseStorage, storage_key: StorageKey
|
|
):
|
|
assert await storage.get_data(key=storage_key) == {}
|
|
assert await storage.update_data(key=storage_key, data={"foo": "bar"}) == {"foo": "bar"}
|
|
assert await storage.update_data(key=storage_key, data={}) == {"foo": "bar"}
|
|
assert await storage.get_data(key=storage_key) == {"foo": "bar"}
|
|
assert await storage.update_data(key=storage_key, data={"baz": "spam"}) == {
|
|
"foo": "bar",
|
|
"baz": "spam",
|
|
}
|
|
assert await storage.get_data(key=storage_key) == {
|
|
"foo": "bar",
|
|
"baz": "spam",
|
|
}
|
|
assert await storage.update_data(key=storage_key, data={"baz": "test"}) == {
|
|
"foo": "bar",
|
|
"baz": "test",
|
|
}
|
|
assert await storage.get_data(key=storage_key) == {
|
|
"foo": "bar",
|
|
"baz": "test",
|
|
}
|