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)