def get_router_view_definition(router_type, router=None):
# Scoped import to avoid circular deps.
from go.router.view_definition import RouterViewDefinitionBase
try:
router_pkg = get_router_pkg(
router_type, ['definition', 'view_definition'])
except UnknownRouterType:
# To handle obsolete routers that are still viewable
if router_type not in obsolete_router_types():
raise
from go.vumitools.router.definition import (
RouterDefinitionBase)
router_def = RouterDefinitionBase(router)
router_def.router_type = router_type
return RouterViewDefinitionBase(router_def)
router_def = router_pkg.definition.RouterDefinition(router)
if not hasattr(router_pkg, 'view_definition'):
return RouterViewDefinitionBase(router_def)
return router_pkg.view_definition.RouterViewDefinition(router_def)
def test_obsolete_router_types(self):
obsolete_types = obsolete_router_types()
self.assertEqual(obsolete_types, set([]))
def test_obsolete_router_types(self):
obsolete_types = obsolete_router_types()
self.assertEqual(obsolete_types, set([
]))
