def test_start_trace(self):
tracer = context_tracer.ContextTracer()
cur_trace = mock.Mock()
tracer.cur_trace = cur_trace
tracer.start_trace()
self.assertTrue(cur_trace.start.called)
def test_end_span_empty_span_stack(self):
tracer = context_tracer.ContextTracer()
span = mock.Mock()
tracer._span_stack.append(span)
tracer.end_span()
self.assertEqual(tracer.span_context.span_id, tracer.root_span_id)
self.assertTrue(span.finish.called)
def test_trace(self):
from opencensus.trace import trace
tracer = context_tracer.ContextTracer()
cur_trace = tracer.trace()
self.assertEqual(cur_trace.trace_id, tracer.trace_id)
assert isinstance(cur_trace, trace.Trace)
def test_constructor_defaults(self):
from opencensus.trace import span_context
from opencensus.trace import trace
tracer = context_tracer.ContextTracer()
assert isinstance(tracer.span_context, span_context.SpanContext)
assert isinstance(tracer.cur_trace, trace.Trace)
self.assertEqual(tracer._span_stack, [])
self.assertEqual(tracer.root_span_id, tracer.span_context.span_id)
def test_list_collected_spans(self):
tracer = context_tracer.ContextTracer()
span1 = mock.Mock()
span2 = mock.Mock()
tracer.cur_trace.spans.append(span1)
tracer.cur_trace.spans.append(span2)
spans = tracer.list_collected_spans()
self.assertEqual(spans, [span1, span2])
def test_end_trace_without_spans(self):
spans = []
trace_id = '6e0c63257de34c92bf9efcd03927272e'
cur_trace = mock.Mock()
tracer = context_tracer.ContextTracer()
tracer.cur_trace = cur_trace
tracer.cur_trace.spans = spans
tracer.trace_id = trace_id
trace = tracer.end_trace()
self.assertIsNone(trace)
def test_constructor_explicit(self):
from opencensus.trace import span_context
from opencensus.trace import trace
span_context = span_context.SpanContext()
tracer = context_tracer.ContextTracer(span_context=span_context)
self.assertIs(tracer.span_context, span_context)
self.assertEqual(tracer.trace_id, span_context.trace_id)
self.assertEqual(tracer._span_stack, [])
self.assertEqual(tracer.root_span_id, span_context.span_id)
assert isinstance(tracer.cur_trace, trace.Trace)
def test_end_span_span_stack_not_empty(self):
tracer = context_tracer.ContextTracer()
span1 = mock.Mock()
span2 = mock.Mock()
tracer._span_stack.append(span1)
tracer._span_stack.append(span2)
tracer.end_span()
self.assertEqual(tracer.span_context.span_id,
tracer._span_stack[-1].span_id)
self.assertEqual(tracer.span_context.span_id, span1.span_id)
self.assertTrue(span2.finish.called)
def test_add_label_to_current_span(self):
from opencensus.trace.trace_span import TraceSpan
tracer = context_tracer.ContextTracer()
span1 = mock.Mock(spec=TraceSpan)
span1.labels = {}
tracer._span_stack.append(span1)
label_key = 'key'
label_value = 'value'
tracer.add_label_to_current_span(label_key, label_value)
span1.add_label.assert_called_once_with(label_key, label_value)
def test_span(self):
from opencensus.trace import span_context
span_id = 1234
span_name = 'test_span'
span_context = span_context.SpanContext(span_id=span_id)
tracer = context_tracer.ContextTracer(span_context=span_context)
span = tracer.span(name=span_name)
self.assertEqual(span.parent_span_id, span_id)
self.assertEqual(span.name, span_name)
self.assertEqual(len(tracer.cur_trace.spans), 1)
self.assertEqual(len(tracer._span_stack), 1)
self.assertEqual(span_context.span_id, span.span_id)
def test_add_label_to_spans(self):
from opencensus.trace.trace_span import TraceSpan
tracer = context_tracer.ContextTracer()
cur_trace = mock.Mock()
span1 = mock.Mock(spec=TraceSpan)
span2 = mock.Mock(spec=TraceSpan)
span1.labels = {}
span2.labels = {}
cur_trace.spans = [span1, span2]
tracer.cur_trace = cur_trace
label_key = 'key'
label_value = 'value'
tracer.add_label_to_spans(label_key, label_value)
span1.add_label.assert_called_once_with(label_key, label_value)
span2.add_label.assert_called_once_with(label_key, label_value)
def test_end_trace_with_spans(self):
from opencensus.trace.enums import Enum
from opencensus.trace.trace_span import TraceSpan
cur_trace = mock.Mock()
tracer = context_tracer.ContextTracer()
tracer.cur_trace = cur_trace
trace_id = '6e0c63257de34c92bf9efcd03927272e'
cur_trace.trace_id = trace_id
child_span_name = 'child_span'
root_span_name = 'root_span'
child_span_id = 123
root_span_id = 456
kind = Enum.SpanKind.SPAN_KIND_UNSPECIFIED
start_time = '2017-06-25'
end_time = '2017-06-26'
labels = {
'/http/status_code': '200',
'/component': 'HTTP load balancer',
}
child_span = mock.Mock(spec=TraceSpan)
child_span.name = child_span_name
child_span.kind = kind
child_span.parent_span_id = root_span_id
child_span.span_id = child_span_id
child_span.start_time = start_time
child_span.end_time = end_time
child_span.labels = labels
child_span.children = []
child_span.__iter__ = mock.Mock(return_value=iter([child_span]))
root_span = mock.Mock(spec=TraceSpan)
root_span.name = root_span_name
root_span.kind = kind
root_span.parent_span_id = None
root_span.span_id = root_span_id
root_span.start_time = start_time
root_span.end_time = end_time
root_span.labels = None
root_span.children = []
root_span.__iter__ = mock.Mock(
return_value=iter([root_span, child_span]))
child_span_json = {
'name': child_span.name,
'kind': kind,
'parentSpanId': root_span_id,
'spanId': child_span_id,
'startTime': start_time,
'endTime': end_time,
'labels': labels,
}
root_span_json = {
'name': root_span.name,
'kind': kind,
'spanId': root_span_id,
'startTime': start_time,
'endTime': end_time,
}
tracer.cur_trace.spans = [root_span]
trace = {
'traceId': trace_id,
'spans': [root_span_json, child_span_json]
}
trace_json = tracer.end_trace()
self.assertEqual(trace_json, trace)
self.assertTrue(cur_trace.finish.called)
def test_end_span_index_error(self):
tracer = context_tracer.ContextTracer()
with self.assertRaises(IndexError):
tracer.end_span()
def test_start_span(self):
tracer = context_tracer.ContextTracer()
tracer.start_span()
self.assertEqual(len(tracer._span_stack), 1)
self.assertEqual(len(tracer.cur_trace.spans), 1)