1818import static io .opentelemetry .semconv .incubating .MessagingIncubatingAttributes .MESSAGING_SYSTEM ;
1919import static org .assertj .core .api .Assertions .assertThat ;
2020
21+ import io .opentelemetry .api .common .AttributeKey ;
2122import io .opentelemetry .api .trace .SpanContext ;
2223import io .opentelemetry .api .trace .SpanKind ;
2324import io .opentelemetry .sdk .trace .data .LinkData ;
2425import java .nio .charset .StandardCharsets ;
26+ import java .util .Arrays ;
2527import java .util .concurrent .atomic .AtomicReference ;
2628import org .assertj .core .api .AbstractLongAssert ;
2729import org .assertj .core .api .AbstractStringAssert ;
@@ -42,6 +44,11 @@ void assertTraces() {
4244 .hasKind (SpanKind .PRODUCER )
4345 .hasParent (trace .getSpan (0 ))
4446 .hasAttributesSatisfyingExactly (
47+ equalTo (
48+ AttributeKey .stringArrayKey ("messaging.header.baggage" ),
49+ Arrays .asList (
50+ "test-baggage-key-1=test-baggage-value-1" ,
51+ "test-baggage-key-2=test-baggage-value-2" )),
4552 equalTo (MESSAGING_SYSTEM , "kafka" ),
4653 equalTo (MESSAGING_DESTINATION_NAME , SHARED_TOPIC ),
4754 equalTo (MESSAGING_OPERATION , "publish" ),
@@ -64,6 +71,11 @@ void assertTraces() {
6471 .hasNoParent ()
6572 .hasLinksSatisfying (links -> assertThat (links ).isEmpty ())
6673 .hasAttributesSatisfyingExactly (
74+ equalTo (
75+ AttributeKey .stringArrayKey ("messaging.header.baggage" ),
76+ Arrays .asList (
77+ "test-baggage-key-1=test-baggage-value-1" ,
78+ "test-baggage-key-2=test-baggage-value-2" )),
6779 equalTo (MESSAGING_SYSTEM , "kafka" ),
6880 equalTo (MESSAGING_DESTINATION_NAME , SHARED_TOPIC ),
6981 equalTo (MESSAGING_OPERATION , "receive" ),
@@ -78,6 +90,11 @@ void assertTraces() {
7890 .hasParent (trace .getSpan (0 ))
7991 .hasLinks (LinkData .create (producerSpanContext .get ()))
8092 .hasAttributesSatisfyingExactly (
93+ equalTo (
94+ AttributeKey .stringArrayKey ("messaging.header.baggage" ),
95+ Arrays .asList (
96+ "test-baggage-key-1=test-baggage-value-1" ,
97+ "test-baggage-key-2=test-baggage-value-2" )),
8198 equalTo (MESSAGING_SYSTEM , "kafka" ),
8299 equalTo (MESSAGING_DESTINATION_NAME , SHARED_TOPIC ),
83100 equalTo (MESSAGING_OPERATION , "process" ),
0 commit comments