def test_span_context(self): """Test construction of `SpanContextShim` objects.""" otel_context = trace.SpanContext(1234, 5678, is_remote=False) context = opentracingshim.SpanContextShim(otel_context) self.assertIsInstance(context, opentracing.SpanContext) self.assertEqual(context.unwrap().trace_id, 1234) self.assertEqual(context.unwrap().span_id, 5678)
def test_inject_binary(self): """Test `inject()` method for Format.BINARY.""" otel_context = trace.SpanContext(trace_id=1220, span_id=7478) context = opentracingshim.SpanContextShim(otel_context) # Verify exception for non supported binary format. with self.assertRaises(opentracing.UnsupportedFormatException): self.shim.inject(context, opentracing.Format.BINARY, bytearray())
def test_inject_http_headers(self): """Test `inject()` method for Format.HTTP_HEADERS.""" otel_context = trace.SpanContext(trace_id=1220, span_id=7478) context = opentracingshim.SpanContextShim(otel_context) headers = {} self.shim.inject(context, opentracing.Format.HTTP_HEADERS, headers) self.assertEqual(headers[MockHTTPTextFormat.TRACE_ID_KEY], str(1220)) self.assertEqual(headers[MockHTTPTextFormat.SPAN_ID_KEY], str(7478))
def test_inject_text_map(self): """Test `inject()` method for Format.TEXT_MAP.""" otel_context = trace.SpanContext(trace_id=1220, span_id=7478) context = opentracingshim.SpanContextShim(otel_context) # Verify Format.TEXT_MAP text_map = {} self.shim.inject(context, opentracing.Format.TEXT_MAP, text_map) self.assertEqual(text_map[MockHTTPTextFormat.TRACE_ID_KEY], str(1220)) self.assertEqual(text_map[MockHTTPTextFormat.SPAN_ID_KEY], str(7478))