def new_init(self, *args, **kwargs):
init(self, *args, **kwargs)
call_args = utils.get_call_args(init, self, *args, **kwargs)
f = functools.partial(handler.handle_call_args,
call_args=call_args)
hook = hooks.RequestAuditor(f)
self._builder.add_hook(hook)
def new_init(self, *args, **kwargs):
init(self, *args, **kwargs)
call_args = utils.get_call_args(init, self, *args, **kwargs)
f = functools.partial(handler.handle_call_args,
call_args=call_args)
hook = hooks_.RequestAuditor(f)
self.session.inject(hook)
def with_value(self, value):
"""
Creates an object that can be used with the
:py:class:`Session.inject` method or
:py:class:`~uplink.inject` decorator to inject request properties
with specific values.
.. versionadded:: 0.4.0
"""
auditor = functools.partial(self.modify_request, value=value)
return hooks.RequestAuditor(auditor)
def test_delegate_handle_response_multiple(self, mocker):
# Include one hook that can't handle responses
mock_response_handler = mocker.Mock()
mock_request_auditor = mocker.Mock()
chain = hooks.TransactionHookChain(
hooks.RequestAuditor(mock_request_auditor),
hooks.ResponseHandler(mock_response_handler),
hooks.ResponseHandler(mock_response_handler),
)
chain.handle_response("consumer", {})
assert mock_response_handler.call_count == 2
assert mock_request_auditor.call_count == 0
def _hook(self):
if self.__hook is None:
self.__hook = hooks.RequestAuditor(self.set_json_body)
return self.__hook
def test_audit_request(self, mocker):
auditor = mocker.Mock()
ra = hooks.RequestAuditor(auditor)
ra.audit_request("consumer", "request")
auditor.assert_called_with("request")
def test_audit_request(self, mocker):
auditor = mocker.stub()
ra = hooks.RequestAuditor(auditor)
ra.audit_request(1, 2, 3)
auditor.assert_called_with(1, 2, 3)
