def test_trace_integration_set_tracer(self):
mock_wrap = mock.Mock()
mock_requests = mock.Mock()
wrap_result = 'wrap result'
mock_wrap.return_value = wrap_result
for func in trace.REQUESTS_WRAP_METHODS:
mock_func = mock.Mock()
mock_func.__name__ = func
setattr(mock_requests, func, mock_func)
patch_wrap = mock.patch('opencensus.ext.requests.trace.wrap_requests',
mock_wrap)
patch_requests = mock.patch('opencensus.ext.requests.trace.requests',
mock_requests)
class TmpTracer(noop_tracer.NoopTracer):
pass
with patch_wrap, patch_requests:
trace.trace_integration(tracer=TmpTracer())
self.assertIsInstance(execution_context.get_opencensus_tracer(),
TmpTracer)
def test_trace_integration(self):
mock_wrap = mock.Mock()
patch_wrapt = mock.patch('wrapt.wrap_function_wrapper', mock_wrap)
with patch_wrapt:
trace.trace_integration()
self.assertIsInstance(execution_context.get_opencensus_tracer(),
noop_tracer.NoopTracer)
mock_wrap.assert_called_once_with(trace.MODULE_NAME,
'Session.request',
trace.wrap_session_request)
