Add ot-trace header casing tests

open-telemetry · Oct 13, 2023 · bb4c7d2 · bb4c7d2
1 parent ff58f67
commit bb4c7d2
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/...ors/src/test/java/io/opentelemetry/extension/trace/propagation/OtTracePropagatorTest.java b/...ors/src/test/java/io/opentelemetry/extension/trace/propagation/OtTracePropagatorTest.java
@@ -20,6 +20,7 @@
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.LinkedHashMap;
+import java.util.Locale;
 import java.util.Map;
 import javax.annotation.Nullable;
 import org.junit.jupiter.api.Test;
@@ -55,6 +56,18 @@ private static Context withSpanContext(SpanContext spanContext, Context context)
     return context.with(Span.wrap(spanContext));
   }
 
+  private static String capitalizeFirstLetter(String input, String delimiter) {
+    String[] words = input.split(delimiter);
+
+    for (int i = 0; i < words.length; i++) {
+      String firstLetter = words[i].substring(0, 1).toUpperCase(Locale.ROOT);
+      String restOfWord = words[i].substring(1).toLowerCase(Locale.ROOT);
+      words[i] = firstLetter + restOfWord;
+    }
+
+    return String.join(delimiter, words);
+  }
+
   @Test
   void inject_invalidContext() {
     Map<String, String> carrier = new LinkedHashMap<>();
@@ -313,6 +326,23 @@ void extract_Baggage() {
     assertThat(Baggage.fromContext(context)).isEqualTo(expectedBaggage);
   }
 
+  @Test
+  void extract_Baggage_CapitalizedHeaders() {
+    String capitalizedBaggageHeader = capitalizeFirstLetter(
+        OtTracePropagator.PREFIX_BAGGAGE_HEADER + "some-key",
+        "-");
+    Map<String, String> carrier = new LinkedHashMap<>();
+    carrier.put(OtTracePropagator.TRACE_ID_HEADER, TRACE_ID);
+    carrier.put(OtTracePropagator.SPAN_ID_HEADER, SPAN_ID);
+    carrier.put(OtTracePropagator.SAMPLED_HEADER, Common.TRUE_INT);
+    carrier.put(capitalizedBaggageHeader, "value");
+
+    Context context = propagator.extract(Context.current(), carrier, getter);
+
+    Baggage expectedBaggage = Baggage.builder().put("some-key", "value").build();
+    assertThat(Baggage.fromContext(context)).isEqualTo(expectedBaggage);
+  }
+
   @Test
   void extract_Baggage_InvalidContext() {
     Map<String, String> carrier = new LinkedHashMap<>();