def test_multiple_commands(self):
mock_tracer = MockTracer()
first_mock_event = MockEvent({}, ("firstUrl", "123"), "first")
second_mock_event = MockEvent({}, ("secondUrl", "456"), "second")
command_tracer = CommandTracer(mock_tracer)
command_tracer.started(event=first_mock_event)
# pylint: disable=protected-access
first_span = command_tracer._get_span(first_mock_event)
command_tracer.started(event=second_mock_event)
# pylint: disable=protected-access
second_span = command_tracer._get_span(second_mock_event)
command_tracer.succeeded(event=first_mock_event)
command_tracer.failed(event=second_mock_event)
self.assertEqual(first_span.attributes["db.mongo.request_id"], "first")
self.assertIs(
first_span.status.canonical_code,
trace_api.status.StatusCanonicalCode.OK,
)
self.assertEqual(second_span.attributes["db.mongo.request_id"],
"second")
self.assertIs(
second_span.status.canonical_code,
trace_api.status.StatusCanonicalCode.UNKNOWN,
)
def test_multiple_commands(self):
first_mock_event = MockEvent({}, ("firstUrl", "123"), "first")
second_mock_event = MockEvent({}, ("secondUrl", "456"), "second")
command_tracer = CommandTracer(self.tracer)
command_tracer.started(event=first_mock_event)
command_tracer.started(event=second_mock_event)
command_tracer.succeeded(event=first_mock_event)
command_tracer.failed(event=second_mock_event)
spans_list = self.memory_exporter.get_finished_spans()
self.assertEqual(len(spans_list), 2)
first_span = spans_list[0]
second_span = spans_list[1]
self.assertEqual(first_span.attributes["db.mongo.request_id"], "first")
self.assertIs(
first_span.status.canonical_code,
trace_api.status.StatusCanonicalCode.OK,
)
self.assertEqual(second_span.attributes["db.mongo.request_id"],
"second")
self.assertIs(
second_span.status.canonical_code,
trace_api.status.StatusCanonicalCode.UNKNOWN,
)
def test_succeeded(self):
mock_tracer = MockTracer()
mock_event = MockEvent({})
command_tracer = CommandTracer(mock_tracer)
command_tracer.started(event=mock_event)
# pylint: disable=protected-access
span = command_tracer._get_span(mock_event)
command_tracer.succeeded(event=mock_event)
self.assertEqual(span.attributes["db.mongo.duration_micros"],
"duration_micros")
self.assertIs(span.status.canonical_code,
trace_api.status.StatusCanonicalCode.OK)
self.assertEqual(span.status.description, "reply")
self.assertIsNotNone(span.end_time)
def test_succeeded(self):
mock_event = MockEvent({})
command_tracer = CommandTracer(self.tracer)
command_tracer.started(event=mock_event)
command_tracer.succeeded(event=mock_event)
spans_list = self.memory_exporter.get_finished_spans()
self.assertEqual(len(spans_list), 1)
span = spans_list[0]
self.assertEqual(span.attributes["db.mongo.duration_micros"],
"duration_micros")
self.assertIs(span.status.canonical_code,
trace_api.status.StatusCanonicalCode.OK)
self.assertEqual(span.status.description, "reply")
self.assertIsNotNone(span.end_time)