def tearDown(self):
from django.test.utils import teardown_test_environment
from opencensus.trace import execution_context
execution_context.clear()
teardown_test_environment()
def __enter__(self):
self.orig_tracer = OpenCensusSpan.get_current_tracer()
self.orig_current_span = OpenCensusSpan.get_current_span()
execution_context.clear()
if self.tracer_to_use is not None:
settings.tracing_implementation.set_value(self.tracer_to_use)
self.os_env.start()
execution_context.clear()
return self
def test_blacklist_path(self):
from django.test import RequestFactory
from opencensus.trace.ext.django import middleware
from opencensus.trace.tracers import base
from opencensus.trace.tracers import noop_tracer
from opencensus.trace.ext import utils
from opencensus.trace import execution_context
execution_context.clear()
blacklist_paths = [
'test_blacklist_path',
]
params = {
'BLACKLIST_PATHS': [
'test_blacklist_path',
],
'TRANSPORT':
'opencensus.trace.exporters.transports.sync.SyncTransport',
}
patch_params = mock.patch(
'opencensus.trace.ext.django.middleware.settings.params', params)
with patch_params:
middleware_obj = middleware.OpencensusMiddleware()
django_request = RequestFactory().get('/test_blacklist_path')
disabled = utils.disable_tracing_url(django_request.path,
blacklist_paths)
self.assertTrue(disabled)
self.assertEqual(middleware_obj._blacklist_paths, blacklist_paths)
# test process_request
middleware_obj.process_request(django_request)
tracer = middleware._get_current_tracer()
span = tracer.current_span()
# process view
view_func = mock.Mock()
middleware_obj.process_view(django_request, view_func)
tracer = middleware._get_current_tracer()
span = tracer.current_span()
assert isinstance(span, base.NullContextManager)
# process response
django_response = mock.Mock()
django_response.status_code = 200
middleware_obj.process_response(django_request, django_response)
tracer = middleware._get_current_tracer()
span = tracer.current_span()
assert isinstance(span, base.NullContextManager)
def test_excludelist_path(self):
from opencensus.ext.django import middleware
execution_context.clear()
excludelist_paths = ['test_excludelist_path']
settings = type('Test', (object,), {})
settings.OPENCENSUS = {
'TRACE': {
'SAMPLER': 'opencensus.trace.samplers.AlwaysOnSampler()', # noqa
'EXCLUDELIST_PATHS': excludelist_paths,
'EXPORTER': mock.Mock(),
}
}
patch_settings = mock.patch(
'django.conf.settings',
settings)
with patch_settings:
middleware_obj = middleware.OpencensusMiddleware()
django_request = RequestFactory().get('/test_excludelist_path')
disabled = utils.disable_tracing_url(django_request.path,
excludelist_paths)
self.assertTrue(disabled)
self.assertEqual(middleware_obj.excludelist_paths, excludelist_paths)
# test process_request
middleware_obj.process_request(django_request)
tracer = middleware._get_current_tracer()
span = tracer.current_span()
# process view
view_func = mock.Mock()
middleware_obj.process_view(django_request, view_func)
tracer = middleware._get_current_tracer()
span = tracer.current_span()
assert isinstance(span, BlankSpan)
# process response
django_response = mock.Mock()
django_response.status_code = 200
middleware_obj.process_response(django_request, django_response)
tracer = middleware._get_current_tracer()
span = tracer.current_span()
assert isinstance(span, BlankSpan)
def test_get_span_from_thread():
result = []
def get_span_from_thread(output):
current_span = OpenCensusSpan.get_current_span()
output.append(current_span)
tracer = Tracer(sampler=AlwaysOnSampler())
with tracer.span(name="TestSpan") as span:
thread = threading.Thread(
target=get_span_from_thread,
args=(result,)
)
thread.start()
thread.join()
assert span is result[0]
execution_context.clear()
def tearDown(self):
execution_context.clear()
teardown_test_environment()
def setUp(self):
from opencensus.trace import execution_context
execution_context.clear()
def tearDown(self):
from opencensus.trace import execution_context
execution_context.clear()
def tearDown(self):
execution_context.clear()
def setUp(self):
execution_context.clear()
