aiogram/tests/test_dispatcher/test_fsm_context.py
Cyril Margorin d646e19852
Fix fsmcontextproxy update (#755)
* Create test case for invalid 'FSMContextProxy.update()' function

* Fix invalid 'FSMContextProxy.update()' function

* Verify that written data has been stored and read back correctly
2021-12-01 02:43:12 +02:00

14 lines
496 B
Python

import pytest
from aiogram.contrib.fsm_storage.memory import MemoryStorage
from aiogram.dispatcher import FSMContext
class TestFSMContext:
@pytest.mark.asyncio
async def test_update_data(self):
context = FSMContext(MemoryStorage(), chat=1, user=1)
async with context.proxy() as data:
data.update(key1="value1", key2="value2")
async with context.proxy() as data:
assert data['key1'] == "value1"
assert data['key2'] == "value2"