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([ ]))