python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(types): use StorageDataT in storage.dict

Browse source 
fix(tests): name setter properly
This commit is contained in:
mpa 2020-08-22 21:22:36 +04:00
parent 608fa5e27b
commit 5b479b3845
No known key found for this signature in database
GPG key ID: BCCFBFCCC9B754A8
3 changed files with 25 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/test_dispatcher/test_state/test_context.py
View file

@ -58,6 +58,21 @@ class TestCurrentUserContext:
ctx = CurrentUserContext(storage, chat_id, user_id, key_maker=my_key_maker)
assert ctx.key == my_key_maker(chat_id, user_id) == key_maker_const_result
@pytest.mark.asyncio
@pytest.mark.parametrize(
("setter_method", "getter_method", "value"),
(
("set_state", "get_state", "some state"),
("set_data", "get_data", {"some": "data"}),
)
)
async def test_setters_getters(self, storage, setter_method, getter_method, value):
chat_id, user_id = 1, 2
ctx = CurrentUserContext(storage, chat_id, user_id)
assert await getattr(ctx, setter_method)(value) is None
assert await getattr(ctx, getter_method)() == value
@pytest.mark.asyncio
@pytest.mark.parametrize("setter_method", ("set_state", "set_data"))
async def test_setters(self, storage, setter_method):
@ -84,7 +99,7 @@ class TestCurrentUserContext:
@pytest.mark.asyncio
@pytest.mark.parametrize("reseter_method", ("reset_data", "reset_state", "finish"))
async def test_setters(self, storage, reseter_method):
async def test_resetters(self, storage, reseter_method):
chat_id, user_id = 1, 2
ctx = CurrentUserContext(storage, chat_id, user_id)