def test_document_collection(self):
"""Test the document collection."""
span_processor = AzureMetricsSpanProcessor()
span_processor.is_collecting_documents = True
test_span = Span(
name="test",
kind=SpanKind.SERVER,
context=SpanContext(
trace_id=36873507687745823477771305566750195431,
span_id=12030755672171557338,
is_remote=False,
),
)
test_span.set_status(Status(StatusCanonicalCode.INTERNAL, "test"))
test_span._start_time = 5000000
test_span._end_time = 15000000
span_processor.on_end(test_span)
document = span_processor.documents.pop()
self.assertIsNotNone(document)
self.assertEqual(document.name,
"Microsoft.ApplicationInsights.Request")
def test_failed_request(self):
"""Test the functionality when Server Span is ended."""
span_processor = AzureMetricsSpanProcessor()
test_span = Span(
name="test",
kind=SpanKind.SERVER,
context=SpanContext(
trace_id=36873507687745823477771305566750195431,
span_id=12030755672171557338,
is_remote=False,
),
)
test_span.set_status(Status(StatusCanonicalCode.INTERNAL, "test"))
test_span._start_time = 5000000
test_span._end_time = 15000000
span_processor.on_end(test_span)
self.assertEqual(span_processor.dependency_count, 0)
self.assertEqual(span_processor.dependency_duration, 0)
self.assertEqual(span_processor.failed_dependency_count, 0)
self.assertEqual(span_processor.request_count, 1)
self.assertEqual(span_processor.request_duration, 10)
self.assertEqual(span_processor.failed_request_count, 1)