def test__build_sample(self): e = RPCResponseEndpointUnit(interceptors={}) heads = { 'conv-id': sentinel.conv_id, 'ts': '1', 'op': 'remove_femur', 'sender': 'sender', 'receiver': 'getter' } resp_heads = {'sender-service': 'theservice'} samp = e._build_sample(sentinel.name, 200, "Ok", "msg", heads, "response", resp_heads, sentinel.qlen) self.assertEquals( samp, { 'app_name': get_sys_name(), 'op': 'remove_femur', 'attrs': { 'ql': sentinel.qlen, 'pid': sentinel.name }, 'status_descr': "Ok", 'status': '0', 'req_bytes': len('msg'), 'resp_bytes': len('response'), 'uS': 999000, # it's in microseconds! 'initiator': 'sender', 'target': 'theservice' })
def test__sample_request_exception(self, mocklog): 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(side_effect=TestError) e._sample_request(sentinel.status, sentinel.status_descr, sentinel.msg, sentinel.headers, sentinel.response, sentinel.response_headers) mocklog.exception.assert_called_once_with("Could not sample, ignoring")
def test__sample_request_exception(self, mocklog): 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(side_effect=TestError) e._sample_request(sentinel.status, sentinel.status_descr, sentinel.msg, sentinel.headers, sentinel.response, sentinel.response_headers) mocklog.exception.assert_called_once_with("Could not sample, ignoring")
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__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__build_sample(self): e = RPCResponseEndpointUnit(interceptors={}) heads = {'conv-id': sentinel.conv_id, 'ts': '1', 'op': 'remove_femur', 'sender': 'sender', 'receiver': 'getter'} resp_heads = {'sender-service': 'theservice'} samp = e._build_sample("app_name", 200, "Ok", "msg", heads, "response", resp_heads, sentinel.qlen) self.assertEquals(samp, { 'app_name' : "app_name", 'op' : 'remove_femur', 'attrs' : {'ql':sentinel.qlen}, 'status_descr' : "Ok", 'status' : '0', 'req_bytes' : len('msg'), 'resp_bytes': len('response'), 'uS' : 999000, # it's in microseconds! 'initiator' : 'sender', 'target' : 'theservice' })