|
18 | 18 |
|
19 | 19 | import java.util.ArrayList;
|
20 | 20 | import java.util.List;
|
| 21 | +import java.util.stream.Stream; |
21 | 22 |
|
22 | 23 | import io.micrometer.observation.Observation;
|
23 | 24 | import io.micrometer.observation.ObservationHandler;
|
24 | 25 | import io.micrometer.observation.ObservationRegistry;
|
25 | 26 | import io.micrometer.observation.contextpropagation.ObservationThreadLocalAccessor;
|
26 | 27 | import org.junit.jupiter.api.Test;
|
| 28 | +import org.junit.jupiter.params.ParameterizedTest; |
| 29 | +import org.junit.jupiter.params.provider.Arguments; |
| 30 | +import org.junit.jupiter.params.provider.MethodSource; |
| 31 | +import org.mockito.ArgumentCaptor; |
27 | 32 | import reactor.core.publisher.Mono;
|
28 | 33 |
|
29 | 34 | import org.springframework.mock.http.server.reactive.MockServerHttpRequest;
|
30 | 35 | import org.springframework.mock.web.server.MockServerWebExchange;
|
| 36 | +import org.springframework.web.server.ServerWebExchange; |
31 | 37 | import org.springframework.web.server.WebFilter;
|
32 | 38 | import org.springframework.web.server.WebFilterChain;
|
33 | 39 |
|
34 | 40 | import static org.assertj.core.api.Assertions.assertThat;
|
35 | 41 | import static org.mockito.ArgumentMatchers.any;
|
36 | 42 | import static org.mockito.BDDMockito.given;
|
37 | 43 | import static org.mockito.Mockito.mock;
|
| 44 | +import static org.mockito.Mockito.times; |
38 | 45 | import static org.mockito.Mockito.verify;
|
39 | 46 | import static org.mockito.Mockito.verifyNoInteractions;
|
40 | 47 |
|
@@ -113,6 +120,51 @@ void decorateWhenCustomAfterFilterThenObserves() {
|
113 | 120 | handler.assertSpanStop(9, "http");
|
114 | 121 | }
|
115 | 122 |
|
| 123 | + @ParameterizedTest |
| 124 | + @MethodSource("decorateFiltersWhenCompletesThenHasSpringSecurityReachedFilterNameTagArguments") |
| 125 | + void decorateFiltersWhenCompletesThenHasSpringSecurityReachedFilterNameTag(WebFilter filter, |
| 126 | + String expectedFilterNameTag) { |
| 127 | + ObservationHandler<Observation.Context> handler = mock(ObservationHandler.class); |
| 128 | + given(handler.supportsContext(any())).willReturn(true); |
| 129 | + ObservationRegistry registry = ObservationRegistry.create(); |
| 130 | + registry.observationConfig().observationHandler(handler); |
| 131 | + ObservationWebFilterChainDecorator decorator = new ObservationWebFilterChainDecorator(registry); |
| 132 | + WebFilterChain chain = mock(WebFilterChain.class); |
| 133 | + given(chain.filter(any())).willReturn(Mono.empty()); |
| 134 | + WebFilterChain decorated = decorator.decorate(chain, List.of(filter)); |
| 135 | + decorated.filter(MockServerWebExchange.from(MockServerHttpRequest.get("/").build())).block(); |
| 136 | + |
| 137 | + ArgumentCaptor<Observation.Context> context = ArgumentCaptor.forClass(Observation.Context.class); |
| 138 | + verify(handler, times(3)).onStop(context.capture()); |
| 139 | + |
| 140 | + assertThat(context.getValue().getLowCardinalityKeyValue("spring.security.reached.filter.name").getValue()) |
| 141 | + .isEqualTo(expectedFilterNameTag); |
| 142 | + } |
| 143 | + |
| 144 | + static Stream<Arguments> decorateFiltersWhenCompletesThenHasSpringSecurityReachedFilterNameTagArguments() { |
| 145 | + WebFilter filterWithName = new BasicAuthenticationFilter(); |
| 146 | + |
| 147 | + // Anonymous class leads to an empty filter-name |
| 148 | + WebFilter filterWithoutName = new WebFilter() { |
| 149 | + @Override |
| 150 | + public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) { |
| 151 | + return chain.filter(exchange); |
| 152 | + } |
| 153 | + }; |
| 154 | + |
| 155 | + return Stream.of(Arguments.of(filterWithName, "BasicAuthenticationFilter"), |
| 156 | + Arguments.of(filterWithoutName, "none")); |
| 157 | + } |
| 158 | + |
| 159 | + static class BasicAuthenticationFilter implements WebFilter { |
| 160 | + |
| 161 | + @Override |
| 162 | + public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) { |
| 163 | + return chain.filter(exchange); |
| 164 | + } |
| 165 | + |
| 166 | + } |
| 167 | + |
116 | 168 | static class AccumulatingObservationHandler implements ObservationHandler<Observation.Context> {
|
117 | 169 |
|
118 | 170 | List<Event> contexts = new ArrayList<>();
|
|
0 commit comments