def test_register_rules(self): app = Mock() controller = WorkController() url_rules = [['/work/request', 'work_request', {'type_is': 'decorated_view', 'name': 'work_request', 'co_filename': '{}'.format(rapid.lib.__file__.replace('.pyc', '.py'))}], ['/work/execute', 'work_execute', {'type_is': 'decorated_view', 'name': 'work_execute', 'co_filename': '{}'.format(rapid.lib.__file__.replace('.pyc', '.py'))}, {'methods': ['POST']}]] registered_rules = {} def set_url(*args, **kwargs): arguments = list(args) if sys.version.startswith('2'): # TODO: Remove after moved to python3 arguments[2] = {'type_is': args[2].func_code.co_name, 'name': args[2].func_name, 'co_filename': args[2].func_code.co_filename} else: arguments[2] = {'type_is': args[2].__code__.co_name, 'name': args[2].__wrapped__.__func__.__name__, 'co_filename': args[2].__code__.co_filename} if kwargs: arguments.append(kwargs) registered_rules[args[0]] = arguments app.add_url_rule = set_url controller.register_url_rules(app) for assertions in url_rules: eq_(registered_rules[assertions[0]], assertions)
def test_register_url_rules(self): mock_app = Mock() mock_app.add_url_rule = self._register_helper self.dal.register_url_rules(mock_app) eq_(self.dal.app, mock_app) eq_( self.registry['/api/pipelines/create'], { 'name': 'create_pipeline', 'func': 'create_pipeline', 'methods': ['POST'] }) eq_( self.registry['/api/pipelines/<int:pipeline_id>/start'], { 'name': 'start_pipeline_instance', 'func': 'start_pipeline_instance', 'methods': ['POST'] })