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_failed(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.failed(event=mock_event) self.assertEqual(span.attributes["db.mongo.duration_micros"], "duration_micros") self.assertIs( span.status.canonical_code, trace_api.status.StatusCanonicalCode.UNKNOWN, ) self.assertEqual(span.status.description, "failure") self.assertIsNotNone(span.end_time)
def test_failed(self): mock_event = MockEvent({}) command_tracer = CommandTracer(self.tracer) command_tracer.started(event=mock_event) command_tracer.failed(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.UNKNOWN, ) self.assertEqual(span.status.description, "failure") self.assertIsNotNone(span.end_time)