mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
* Create test case for invalid 'FSMContextProxy.update()' function * Fix invalid 'FSMContextProxy.update()' function * Verify that written data has been stored and read back correctly
14 lines
496 B
Python
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"
|