def test__get_sflow_manager_with_container(self): Container.instance = None c = Container() # ensure an instance e = RPCResponseEndpointUnit(interceptors={}) self.assertEquals(e._get_sflow_manager(), c.sflow_manager) Container.instance = None
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_no_sample(self, mocklog): e = RPCResponseEndpointUnit(interceptors={}) e._get_sflow_manager = Mock(return_value=Mock(spec=SFlowManager)) e._get_sflow_manager.return_value.should_sample = False e._get_sample_name = Mock() e._sample_request(sentinel.status, sentinel.status_descr, sentinel.msg, sentinel.headers, sentinel.response, sentinel.response_headers) self.assertEquals(mocklog.debug.call_count, 1) self.assertIn("not to sample", mocklog.debug.call_args[0][0])
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__get_sflow_manager(self): Container.instance = None e = RPCResponseEndpointUnit(interceptors={}) self.assertIsNone(e._get_sflow_manager())