def test__message_received_interceptor_exception(self): e = RPCResponseEndpointUnit(routing_obj=self) e.send = Mock() e.send.return_value = sentinel.sent e.channel = Mock() with patch('pyon.net.endpoint.ResponseEndpointUnit._message_received', new=Mock(side_effect=exception.IonException)): retval = e._message_received(sentinel.msg, {}) self.assertEquals(retval, sentinel.sent) assert_called_once_with_header(self, e.send, {'status_code': -1, 'error_message':'', 'conv-id': '', 'conv-seq': 2, 'protocol':'', 'performative': 'failure'})
def test__sample_request(self): e = RPCResponseEndpointUnit(interceptors={}) e._get_sflow_manager = Mock(return_value=Mock(spec=SFlowManager)) e._get_sflow_manager.return_value.should_sample = True e._build_sample = Mock(return_value={'test':sentinel.test}) e.channel = Mock() e.channel.get_stats = Mock(return_value=(3, 0)) e.channel._recv_queue.qsize = Mock(return_value=3) e._sample_request(sentinel.status, sentinel.status_descr, sentinel.msg, sentinel.headers, sentinel.response, sentinel.response_headers) e._get_sflow_manager.assert_called_once_with() e._build_sample.assert_called_once_with(ANY, sentinel.status, sentinel.status_descr, sentinel.msg, sentinel.headers, sentinel.response, sentinel.response_headers, 6) e._get_sflow_manager.return_value.transaction.assert_called_once_with(test=sentinel.test)
def test__message_received_interceptor_exception(self): e = RPCResponseEndpointUnit(routing_obj=self) e.send = Mock() e.send.return_value = sentinel.sent e.channel = Mock() with patch('pyon.net.endpoint.ResponseEndpointUnit._message_received', new=Mock(side_effect=exception.IonException)): retval = e._message_received(sentinel.msg, {}) self.assertEquals(retval, sentinel.sent) e.send.assert_called_once_with( None, { 'status_code': -1, 'error_message': '', 'conv-id': '', 'conv-seq': 2, 'protocol': '', 'performative': 'failure' })
def test__message_received_interceptor_exception(self): e = RPCResponseEndpointUnit(routing_obj=self) e.send = Mock() e.send.return_value = sentinel.sent e.channel = Mock() with patch( "pyon.net.endpoint.ResponseEndpointUnit._message_received", new=Mock(side_effect=exception.IonException) ): retval = e._message_received(sentinel.msg, {}) self.assertEquals(retval, sentinel.sent) e.send.assert_called_once_with( None, { "status_code": -1, "error_message": "", "conv-id": "", "conv-seq": 2, "protocol": "", "performative": "failure", }, )
def test__sample_request(self): e = RPCResponseEndpointUnit(interceptors={}) e._get_sflow_manager = Mock(return_value=Mock(spec=SFlowManager)) e._get_sflow_manager.return_value.should_sample = True e._build_sample = Mock(return_value={'test': sentinel.test}) e.channel = Mock() e.channel.get_stats = Mock(return_value=(3, 0)) e.channel._recv_queue.qsize = Mock(return_value=3) e._sample_request(sentinel.status, sentinel.status_descr, sentinel.msg, sentinel.headers, sentinel.response, sentinel.response_headers) e._get_sflow_manager.assert_called_once_with() e._build_sample.assert_called_once_with(ANY, sentinel.status, sentinel.status_descr, sentinel.msg, sentinel.headers, sentinel.response, sentinel.response_headers, 6) e._get_sflow_manager.return_value.transaction.assert_called_once_with( test=sentinel.test)