def test_wrap_session_request_exporter_thread(self):
def wrapped(*args, **kwargs):
result = mock.Mock()
result.status_code = 200
return result
mock_tracer = MockTracer()
patch_tracer = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_tracer',
return_value=mock_tracer)
patch_attr = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_attr',
return_value=['localhost:8080'])
patch_thread = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'is_exporter',
return_value=True)
url = 'http://localhost:8080'
request_method = 'POST'
with patch_tracer, patch_attr, patch_thread:
trace.wrap_session_request(wrapped, 'Session.request',
(request_method, url), {})
self.assertEqual(None, mock_tracer.current_span)
def test_wrap_session_request_blacklist_ok(self):
def wrapped(*args, **kwargs):
result = mock.Mock()
result.status_code = 200
return result
mock_tracer = MockTracer(propagator=mock.Mock(
to_headers=lambda x: {'x-trace': 'some-value'}))
patch_tracer = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_tracer',
return_value=mock_tracer)
patch_attr = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_attr',
return_value=None)
patch_thread = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'is_exporter',
return_value=False)
url = 'http://localhost/'
request_method = 'POST'
with patch_tracer, patch_attr, patch_thread:
trace.wrap_session_request(wrapped, 'Session.request',
(request_method, url), {})
expected_name = '/'
self.assertEqual(expected_name, mock_tracer.current_span.name)
def test_wrap_session_request(self):
wrapped = mock.Mock(return_value=mock.Mock(status_code=200))
mock_tracer = MockTracer(propagator=mock.Mock(
to_headers=lambda x: {'x-trace': 'some-value'}))
patch = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_tracer',
return_value=mock_tracer)
patch_thread = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'is_exporter',
return_value=False)
url = 'http://localhost:8080'
request_method = 'POST'
kwargs = {}
with patch, patch_thread:
trace.wrap_session_request(wrapped, 'Session.request',
(request_method, url), kwargs)
expected_attributes = {'http.url': url, 'http.status_code': '200'}
expected_name = '[requests]POST'
self.assertEqual(span_module.SpanKind.CLIENT,
mock_tracer.current_span.span_kind)
self.assertEqual(expected_attributes,
mock_tracer.current_span.attributes)
self.assertEqual(kwargs['headers']['x-trace'], 'some-value')
self.assertEqual(expected_name, mock_tracer.current_span.name)
def test_tracer_headers_are_overwritten(self):
wrapped = mock.Mock(return_value=mock.Mock(status_code=200))
mock_tracer = MockTracer(
propagator=mock.Mock(
to_headers=lambda x: {'x-trace': 'some-value'}))
patch = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_tracer',
return_value=mock_tracer)
patch_thread = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'is_exporter',
return_value=False)
url = 'http://localhost:8080'
request_method = 'POST'
kwargs = {'headers': {'x-trace': 'original-value'}}
with patch, patch_thread:
trace.wrap_session_request(wrapped, 'Session.request',
(request_method, url), kwargs)
self.assertEqual(kwargs['headers']['x-trace'], 'some-value')
def test_wrap_session_request_exception(self):
wrapped = mock.Mock(return_value=mock.Mock(status_code=200))
wrapped.side_effect = requests.TooManyRedirects
mock_tracer = MockTracer(
propagator=mock.Mock(
to_headers=lambda x: {'x-trace': 'some-value'}))
patch = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_tracer',
return_value=mock_tracer)
patch_thread = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'is_exporter',
return_value=False)
url = 'http://localhost:8080/test'
request_method = 'POST'
kwargs = {}
with patch, patch_thread:
trace.wrap_session_request(wrapped, 'Session.request',
(request_method, url), kwargs)
expected_attributes = {
'http.host': 'localhost:8080',
'http.method': 'POST',
'http.path': '/test',
'http.url': url,
}
expected_name = '/test'
expected_status = status_module.Status(2, '')
self.assertEqual(span_module.SpanKind.CLIENT,
mock_tracer.current_span.span_kind)
self.assertEqual(expected_attributes,
mock_tracer.current_span.attributes)
self.assertEqual(kwargs['headers']['x-trace'], 'some-value')
self.assertEqual(expected_name, mock_tracer.current_span.name)
self.assertEqual(
expected_status.__dict__,
mock_tracer.current_span.status.__dict__
)
def test_header_is_passed_in(self):
wrapped = mock.Mock(return_value=mock.Mock(status_code=200))
mock_tracer = MockTracer(propagator=mock.Mock(
to_headers=lambda x: {'x-trace': 'some-value'}))
patch = mock.patch(
'opencensus.ext.requests.trace.execution_context.'
'get_opencensus_tracer',
return_value=mock_tracer)
url = 'http://localhost:8080'
request_method = 'POST'
kwargs = {}
with patch:
trace.wrap_session_request(wrapped, 'Session.request',
(request_method, url), kwargs)
self.assertEqual(kwargs['headers']['x-trace'], 'some-value')
