mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
use global pytest.mark.asyncio instead decorator
This commit is contained in:
parent
e4a0f78cc3
commit
4484323c4d
2 changed files with 4 additions and 10 deletions
|
|
@ -70,7 +70,7 @@ async def gender_pre_set(context: FSMContext, old_state: str, new_state: str):
|
|||
|
||||
# You can use pre_finish decorator to call decorated functions before state group will be finished
|
||||
@Form.pre_finish
|
||||
async def form_post_finish(context: FSMContext, old_state: str):
|
||||
async def form_pre_finish(context: FSMContext, old_state: str):
|
||||
async with context.proxy() as data:
|
||||
markup = types.ReplyKeyboardRemove()
|
||||
|
||||
|
|
|
|||
|
|
@ -4,6 +4,8 @@ from aiogram.contrib.fsm_storage.memory import MemoryStorage
|
|||
from aiogram.dispatcher import FSMContext
|
||||
from aiogram.dispatcher.filters.state import StatesGroup, State
|
||||
|
||||
pytestmark = pytest.mark.asyncio
|
||||
|
||||
|
||||
class FakeDispatcher:
|
||||
context = FSMContext(MemoryStorage(), chat=1, user=1)
|
||||
|
|
@ -26,7 +28,7 @@ async def reset_dispatcher():
|
|||
FakeDispatcher.reset()
|
||||
context = FakeDispatcher.get_current().current_state()
|
||||
current_state = await context.get_state()
|
||||
assert current_state == None
|
||||
assert current_state == None
|
||||
yield FakeDispatcher
|
||||
|
||||
|
||||
|
|
@ -36,47 +38,39 @@ class MyGroup(StatesGroup, dispatcher=FakeDispatcher()):
|
|||
|
||||
|
||||
class TestNavigation:
|
||||
@pytest.mark.asyncio
|
||||
async def test_first(self, reset_dispatcher):
|
||||
state = await MyGroup.first()
|
||||
assert state == 'MyGroup:state_1'
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_last(self, reset_dispatcher):
|
||||
state = await MyGroup.last()
|
||||
assert state == 'MyGroup:state_2'
|
||||
|
||||
class TestNext:
|
||||
@pytest.mark.asyncio
|
||||
async def test_next_from_none(self, reset_dispatcher):
|
||||
state = await MyGroup.next()
|
||||
assert state == 'MyGroup:state_1'
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_next_from_the_first_state(self, reset_dispatcher):
|
||||
await MyGroup.state_1.set()
|
||||
state = await MyGroup.next()
|
||||
assert state == 'MyGroup:state_2'
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_next_from_the_last_state(self, reset_dispatcher):
|
||||
await MyGroup.last()
|
||||
state = await MyGroup.next()
|
||||
assert state == None
|
||||
|
||||
class TestPrevious:
|
||||
@pytest.mark.asyncio
|
||||
async def test_previous_from_none(self, reset_dispatcher):
|
||||
state = await MyGroup.previous()
|
||||
assert state == 'MyGroup:state_1'
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_previous_from_the_first_state(self, reset_dispatcher):
|
||||
await MyGroup.first()
|
||||
state = await MyGroup.previous()
|
||||
assert state == None
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_previous_from_the_last_state(self, reset_dispatcher):
|
||||
await MyGroup.state_2.set()
|
||||
state = await MyGroup.previous()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue