baggage propagation

Author: AgraVator

opentelemetry/src/main/java/io/grpc/opentelemetry/OpenTelemetryTracingModule.java

@@ -61,7 +61,6 @@ final class OpenTelemetryTracingModule {
 
   @VisibleForTesting
   final io.grpc.Context.Key<Span> otelSpan = io.grpc.Context.key("opentelemetry-span-key");
-
   @Nullable
   private static final AtomicIntegerFieldUpdater<CallAttemptsTracerFactory> callEndedUpdater;
   @Nullable

opentelemetry/src/test/java/io/grpc/opentelemetry/OpenTelemetryTracingModuleTest.java

@@ -866,6 +866,48 @@ public void testNullBaggageIsHandledGracefully() {
         Baggage.empty(), capturedBaggage.get());
   }
 
+  @Test
+  public void serverSpanPropagationInterceptor_propagatesBaggage() {
+    // 1. Arrange: Setup the module, interceptor, and mocks.
+    OpenTelemetryTracingModule tracingModule = new OpenTelemetryTracingModule(
+        openTelemetryRule.getOpenTelemetry());
+    ServerInterceptor interceptor = tracingModule.getServerSpanPropagationInterceptor();
+
+    @SuppressWarnings("unchecked")
+    ServerCallHandler<String, String> handler = mock(ServerCallHandler.class);
+    ServerCall<String, String> call = new NoopServerCall<>();
+    Metadata metadata = new Metadata();
+    final AtomicReference<Baggage> capturedBaggage = new AtomicReference<>();
+
+    // Mock the handler to capture the Baggage from the current context when it's called.
+    when(handler.startCall(any(), any())).thenAnswer(invocation -> {
+      capturedBaggage.set(io.opentelemetry.api.baggage.Baggage.current());
+      return mockServerCallListener;
+    });
+
+    // Create a test Span and Baggage to be propagated.
+    Span parentSpan = tracerRule.spanBuilder("parent-span").startSpan();
+    io.opentelemetry.api.baggage.Baggage testBaggage =
+        io.opentelemetry.api.baggage.Baggage.builder().put("testKey", "testValue").build();
+
+    // Attach the Span and Baggage to the gRPC context.
+    io.grpc.Context grpcContext = io.grpc.Context.current()
+        .withValue(tracingModule.otelSpan, parentSpan)
+        .withValue(tracingModule.baggageKey, testBaggage);
+
+    io.grpc.Context previous = grpcContext.attach();
+    try {
+      // 2. Act: Call the interceptor.
+      interceptor.interceptCall(call, metadata, handler);
+    } finally {
+      grpcContext.detach(previous);
+    }
+
+    // 3. Assert: Verify the handler was called and the correct Baggage was propagated.
+    verify(handler).startCall(same(call), same(metadata));
+    assertEquals(testBaggage, capturedBaggage.get());
+  }
+
   @Test
   public void generateTraceSpanName() {
     assertEquals(