|
30 | 30 | import reactor.core.publisher.Mono;
|
31 | 31 |
|
32 | 32 | import org.springframework.core.ParameterizedTypeReference;
|
| 33 | +import org.springframework.core.codec.ByteArrayDecoder; |
33 | 34 | import org.springframework.core.codec.StringDecoder;
|
34 | 35 | import org.springframework.core.io.buffer.DataBuffer;
|
35 | 36 | import org.springframework.core.io.buffer.DefaultDataBuffer;
|
|
48 | 49 |
|
49 | 50 | import static org.assertj.core.api.Assertions.assertThat;
|
50 | 51 | import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
|
| 52 | +import static org.assertj.core.api.Assertions.entry; |
51 | 53 | import static org.mockito.BDDMockito.given;
|
52 | 54 | import static org.mockito.Mockito.mock;
|
53 | 55 | import static org.springframework.web.reactive.function.BodyExtractors.toMono;
|
@@ -340,6 +342,33 @@ public void toEntityListTypeReference() {
|
340 | 342 | assertThat(result.getHeaders().getContentType()).isEqualTo(MediaType.TEXT_PLAIN);
|
341 | 343 | }
|
342 | 344 |
|
| 345 | + @Test |
| 346 | + public void createException() { |
| 347 | + DefaultDataBufferFactory factory = new DefaultDataBufferFactory(); |
| 348 | + byte[] bytes = "foo".getBytes(StandardCharsets.UTF_8); |
| 349 | + DefaultDataBuffer dataBuffer = factory.wrap(ByteBuffer.wrap(bytes)); |
| 350 | + Flux<DataBuffer> body = Flux.just(dataBuffer); |
| 351 | + |
| 352 | + HttpHeaders httpHeaders = new HttpHeaders(); |
| 353 | + httpHeaders.setContentType(MediaType.TEXT_PLAIN); |
| 354 | + given(mockResponse.getHeaders()).willReturn(httpHeaders); |
| 355 | + given(mockResponse.getStatusCode()).willReturn(HttpStatus.NOT_FOUND); |
| 356 | + given(mockResponse.getRawStatusCode()).willReturn(HttpStatus.NOT_FOUND.value()); |
| 357 | + given(mockResponse.getBody()).willReturn(body); |
| 358 | + |
| 359 | + List<HttpMessageReader<?>> messageReaders = Collections.singletonList( |
| 360 | + new DecoderHttpMessageReader<>(new ByteArrayDecoder())); |
| 361 | + given(mockExchangeStrategies.messageReaders()).willReturn(messageReaders); |
| 362 | + |
| 363 | + Mono<WebClientResponseException> resultMono = defaultClientResponse.createException(); |
| 364 | + WebClientResponseException exception = resultMono.block(); |
| 365 | + assertThat(exception.getStatusCode()).isEqualTo(HttpStatus.NOT_FOUND); |
| 366 | + assertThat(exception.getMessage()).isEqualTo("404 Not Found"); |
| 367 | + assertThat(exception.getHeaders()).containsExactly(entry("Content-Type", |
| 368 | + Collections.singletonList("text/plain"))); |
| 369 | + assertThat(exception.getResponseBodyAsByteArray()).isEqualTo(bytes); |
| 370 | + } |
| 371 | + |
343 | 372 |
|
344 | 373 | private void mockTextPlainResponse(Flux<DataBuffer> body) {
|
345 | 374 | HttpHeaders httpHeaders = new HttpHeaders();
|
|
0 commit comments