Full router coverage

This commit is contained in:
jrootjunior 2019-11-26 11:01:54 +02:00
parent e814d1aefc
commit b5ab2fd153
2 changed files with 149 additions and 9 deletions

View file

@ -137,18 +137,19 @@ class Router:
return result
for router in self.sub_routers:
kwargs.update(event_router=router)
async for result in router.update_handler.trigger(update, **kwargs):
return result
raise SkipHandler
def emit_startup(self, *args, **kwargs):
self.startup.trigger(*args, **kwargs)
async def emit_startup(self, *args, **kwargs):
async for _ in self.startup.trigger(*args, **kwargs): # pragma: no cover
pass
for router in self.sub_routers:
router.emit_startup(*args, **kwargs)
await router.emit_startup(*args, **kwargs)
def emit_shutdown(self, *args, **kwargs):
self.startup.trigger(*args, **kwargs)
async def emit_shutdown(self, *args, **kwargs):
async for _ in self.shutdown.trigger(*args, **kwargs): # pragma: no cover
pass
for router in self.sub_routers:
router.emit_startup(*args, **kwargs)
await router.emit_shutdown(*args, **kwargs)