|
29 | 29 |
|
30 | 30 | import org.springframework.boot.actuate.trace.http.HttpTrace.Request;
|
31 | 31 | import org.springframework.http.HttpHeaders;
|
| 32 | +import org.springframework.http.MediaType; |
32 | 33 | import org.springframework.util.LinkedMultiValueMap;
|
33 | 34 | import org.springframework.util.MultiValueMap;
|
34 | 35 |
|
@@ -270,6 +271,29 @@ void timeTakenCanBeIncluded() {
|
270 | 271 | assertThat(trace.getTimeTaken()).isNotNull();
|
271 | 272 | }
|
272 | 273 |
|
| 274 | + @Test |
| 275 | + void defaultIncludes() { |
| 276 | + HttpHeaders requestHeaders = new HttpHeaders(); |
| 277 | + requestHeaders.setAccept(Arrays.asList(MediaType.APPLICATION_JSON)); |
| 278 | + requestHeaders.set(HttpHeaders.COOKIE, "value"); |
| 279 | + requestHeaders.set(HttpHeaders.AUTHORIZATION, "secret"); |
| 280 | + HttpExchangeTracer tracer = new HttpExchangeTracer(Include.defaultIncludes()); |
| 281 | + HttpTrace trace = tracer.receivedRequest(createRequest(requestHeaders)); |
| 282 | + HttpHeaders responseHeaders = new HttpHeaders(); |
| 283 | + responseHeaders.set(HttpHeaders.SET_COOKIE, "test=test"); |
| 284 | + responseHeaders.setContentLength(0); |
| 285 | + tracer.sendingResponse(trace, createResponse(responseHeaders), this::createPrincipal, () -> "sessionId"); |
| 286 | + assertThat(trace.getTimeTaken()).isNotNull(); |
| 287 | + assertThat(trace.getPrincipal()).isNull(); |
| 288 | + assertThat(trace.getSession()).isNull(); |
| 289 | + assertThat(trace.getTimestamp()).isNotNull(); |
| 290 | + assertThat(trace.getRequest().getMethod()).isEqualTo("GET"); |
| 291 | + assertThat(trace.getRequest().getRemoteAddress()).isNull(); |
| 292 | + assertThat(trace.getResponse().getStatus()).isEqualTo(204); |
| 293 | + assertThat(trace.getRequest().getHeaders()).containsOnlyKeys(HttpHeaders.ACCEPT); |
| 294 | + assertThat(trace.getResponse().getHeaders()).containsOnlyKeys(HttpHeaders.CONTENT_LENGTH); |
| 295 | + } |
| 296 | + |
273 | 297 | private TraceableRequest createRequest() {
|
274 | 298 | return createRequest(Collections.singletonMap(HttpHeaders.ACCEPT, Arrays.asList("application/json")));
|
275 | 299 | }
|
|
0 commit comments