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)