mirror of
https://github.com/spring-projects/spring-boot.git
synced 2024-07-05 00:56:58 +08:00
Polish BaggagePropagationIntegrationTests
This commit is contained in:
parent
d2f74426f7
commit
2605f86731
|
@ -62,23 +62,23 @@ class BaggagePropagationIntegrationTests {
|
||||||
autoConfig.get().run((context) -> {
|
autoConfig.get().run((context) -> {
|
||||||
Tracer tracer = tracer(context);
|
Tracer tracer = tracer(context);
|
||||||
Span span = createSpan(tracer);
|
Span span = createSpan(tracer);
|
||||||
|
BaggageManager baggageManager = baggageManager(context);
|
||||||
assertThatTracingContextIsInitialized(autoConfig);
|
assertThatTracingContextIsInitialized(autoConfig);
|
||||||
try (Tracer.SpanInScope scope = tracer.withSpan(span.start())) {
|
try (Tracer.SpanInScope scope = tracer.withSpan(span.start())) {
|
||||||
BaggageManager baggageManager = baggageManager(context);
|
assertMdcValue("traceId", span.context().traceId());
|
||||||
try (BaggageInScope fo = baggageManager.createBaggageInScope(span.context(), COUNTRY_CODE, "FO");
|
try (BaggageInScope fo = baggageManager.createBaggageInScope(span.context(), COUNTRY_CODE, "FO");
|
||||||
BaggageInScope alm = baggageManager.createBaggageInScope(span.context(), BUSINESS_PROCESS,
|
BaggageInScope alm = baggageManager.createBaggageInScope(span.context(), BUSINESS_PROCESS,
|
||||||
"ALM")) {
|
"ALM")) {
|
||||||
assertThat(MDC.get("traceId")).isEqualTo(span.context().traceId());
|
assertMdcValue(COUNTRY_CODE, "FO");
|
||||||
assertThat(MDC.get(COUNTRY_CODE)).isEqualTo("FO");
|
assertMdcValue(BUSINESS_PROCESS, "ALM");
|
||||||
assertThat(MDC.get(BUSINESS_PROCESS)).isEqualTo("ALM");
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
finally {
|
finally {
|
||||||
span.end();
|
span.end();
|
||||||
}
|
}
|
||||||
assertThatMdcContainsUnsetTraceId(autoConfig);
|
assertThatMdcContainsUnsetTraceId(autoConfig);
|
||||||
assertThat(MDC.get(COUNTRY_CODE)).isNull();
|
assertUnsetMdc(COUNTRY_CODE);
|
||||||
assertThat(MDC.get(BUSINESS_PROCESS)).isNull();
|
assertUnsetMdc(BUSINESS_PROCESS);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -88,25 +88,25 @@ class BaggagePropagationIntegrationTests {
|
||||||
autoConfig.get().run((context) -> {
|
autoConfig.get().run((context) -> {
|
||||||
Tracer tracer = tracer(context);
|
Tracer tracer = tracer(context);
|
||||||
Span span = createSpan(tracer);
|
Span span = createSpan(tracer);
|
||||||
|
BaggageManager baggageManager = baggageManager(context);
|
||||||
assertThatTracingContextIsInitialized(autoConfig);
|
assertThatTracingContextIsInitialized(autoConfig);
|
||||||
try (Tracer.SpanInScope scope = tracer.withSpan(span.start())) {
|
try (Tracer.SpanInScope scope = tracer.withSpan(span.start())) {
|
||||||
try (BaggageInScope fo = baggageManager(context).createBaggageInScope(span.context(), COUNTRY_CODE,
|
assertMdcValue("traceId", span.context().traceId());
|
||||||
"FO")) {
|
try (BaggageInScope fo = baggageManager.createBaggageInScope(span.context(), COUNTRY_CODE, "FO")) {
|
||||||
assertThat(MDC.get("traceId")).isEqualTo(span.context().traceId());
|
assertMdcValue(COUNTRY_CODE, "FO");
|
||||||
assertThat(MDC.get(COUNTRY_CODE)).isEqualTo("FO");
|
|
||||||
try (Tracer.SpanInScope scope2 = tracer.withSpan(null)) {
|
try (Tracer.SpanInScope scope2 = tracer.withSpan(null)) {
|
||||||
assertThatMdcContainsUnsetTraceId(autoConfig);
|
assertThatMdcContainsUnsetTraceId(autoConfig);
|
||||||
assertThat(MDC.get(COUNTRY_CODE)).isNull();
|
assertUnsetMdc(COUNTRY_CODE);
|
||||||
}
|
}
|
||||||
assertThat(MDC.get("traceId")).isEqualTo(span.context().traceId());
|
assertMdcValue("traceId", span.context().traceId());
|
||||||
assertThat(MDC.get(COUNTRY_CODE)).isEqualTo("FO");
|
assertMdcValue(COUNTRY_CODE, "FO");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
finally {
|
finally {
|
||||||
span.end();
|
span.end();
|
||||||
}
|
}
|
||||||
assertThatMdcContainsUnsetTraceId(autoConfig);
|
assertThatMdcContainsUnsetTraceId(autoConfig);
|
||||||
assertThat(MDC.get(COUNTRY_CODE)).isNull();
|
assertUnsetMdc(COUNTRY_CODE);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -142,6 +142,14 @@ class BaggagePropagationIntegrationTests {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private void assertUnsetMdc(String key) {
|
||||||
|
assertThat(MDC.get(key)).as("MDC[%s]", key).isNull();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void assertMdcValue(String key, String expected) {
|
||||||
|
assertThat(MDC.get(key)).as("MDC[%s]", key).isEqualTo(expected);
|
||||||
|
}
|
||||||
|
|
||||||
enum AutoConfig implements Supplier<ApplicationContextRunner> {
|
enum AutoConfig implements Supplier<ApplicationContextRunner> {
|
||||||
|
|
||||||
BRAVE_DEFAULT {
|
BRAVE_DEFAULT {
|
||||||
|
|
Loading…
Reference in New Issue
Block a user