def __init__(self, method, uri=None, args=None):
self._method = method
self._uri = uri
# Register argument annotations
if args:
is_map = isinstance(args, collections.Mapping)
args, kwargs = ((), args) if is_map else (args, {})
self._add_args = decorators.args(*args, **kwargs)
def test_args_decorate_function(mocker):
handler = mocker.Mock(spec=["set_annotations"])
@classmethod
def patched(*_):
return handler
mocker.patch("uplink.arguments.ArgumentAnnotationHandlerBuilder.from_func",
patched)
args = decorators.args(str, str, name=str)
func = mocker.stub()
args(func)
handler.set_annotations.assert_called_with((str, str), name=str)
def test_args(request_definition_builder):
args = decorators.args(str, str, name=str)
args.modify_request_definition(request_definition_builder)
builder = request_definition_builder.argument_handler_builder
builder.set_annotations.assert_called_with((str, str), name=str)
def test_args_call_old(request_definition_builder):
annotation = decorators.args(str, str, name=str)
annotation(request_definition_builder)
handler = request_definition_builder.method_handler_builder
handler.add_annotation.assert_called_with(annotation)
