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

feat(disp): raise RuntimeError when no storage is set

Browse source 
tests: improve coverage
This commit is contained in:
mpa 2020-08-22 05:16:30 +04:00
parent 25ecc6fafc
commit 608fa5e27b
No known key found for this signature in database
GPG key ID: BCCFBFCCC9B754A8
2 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/test_dispatcher/test_dispatcher.py
View file

@ -11,6 +11,8 @@ from aiogram.api.types import Chat, Message, Update, User
from aiogram.dispatcher.dispatcher import Dispatcher
from aiogram.dispatcher.event.bases import NOT_HANDLED
from aiogram.dispatcher.router import Router
from aiogram.dispatcher.state.context import CurrentUserContext
from aiogram.dispatcher.storage.dict import DictStorage
from tests.mocked_bot import MockedBot
try:
@ -51,6 +53,22 @@ class TestDispatcher:
dp._parent_router = Router()
assert dp.parent_router is None
def test_init_storage(self):
dp = Dispatcher()
with pytest.raises(RuntimeError):
_ = dp.current_state
assert dp.storage is None
_storage = DictStorage()
dp = Dispatcher(storage=_storage)
assert dp.storage == _storage
@pytest.mark.asyncio
async def test_current_state(self):
user = User(id=109_112_97, is_bot=False, first_name="mpa")
User.set_current(user)
dp = Dispatcher(storage=DictStorage())
assert isinstance(dp.current_state, CurrentUserContext)
@pytest.mark.asyncio
async def test_feed_update(self):
dp = Dispatcher()