mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
add type hint, tests, changes file
This commit is contained in:
parent
2e13c28ddd
commit
53a1bcf5e1
3 changed files with 14 additions and 1 deletions
1
CHANGES/1117.feature.rst
Normal file
1
CHANGES/1117.feature.rst
Normal file
|
|
@ -0,0 +1 @@
|
|||
Added a method that allows you to compactly register routers
|
||||
|
|
@ -179,7 +179,7 @@ class Router:
|
|||
self._parent_router = router
|
||||
router.sub_routers.append(self)
|
||||
|
||||
def include_routers(self, *routers):
|
||||
def include_routers(self, *routers) -> Router:
|
||||
if not routers:
|
||||
raise ValueError("You must provide routers")
|
||||
for router in routers:
|
||||
|
|
|
|||
|
|
@ -33,6 +33,18 @@ class TestRouter:
|
|||
assert router3.parent_router is router2
|
||||
assert router3.sub_routers == []
|
||||
|
||||
def test_including_many_routers(self):
|
||||
router = Router()
|
||||
router1 = Router()
|
||||
router2 = Router()
|
||||
|
||||
with pytest.raises(ValueError, match="You must provide routers"):
|
||||
router.include_routers()
|
||||
|
||||
router.include_routers(router1, router2)
|
||||
|
||||
assert router.sub_routers == [router1, router2]
|
||||
|
||||
def test_include_router_by_string_bad_type(self):
|
||||
router = Router()
|
||||
with pytest.raises(ValueError, match=r"router should be instance of Router"):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue