|
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;
|
@@ -328,6 +330,29 @@ public void toEntityListTypeReference() {
|
328 | 330 | assertThat(result.getHeaders().getContentType()).isEqualTo(MediaType.TEXT_PLAIN);
|
329 | 331 | }
|
330 | 332 |
|
| 333 | + @Test |
| 334 | + public void createException() { |
| 335 | + byte[] bytes = "foo".getBytes(StandardCharsets.UTF_8); |
| 336 | + DefaultDataBuffer dataBuffer = DefaultDataBufferFactory.sharedInstance.wrap(ByteBuffer.wrap(bytes)); |
| 337 | + Flux<DataBuffer> body = Flux.just(dataBuffer); |
| 338 | + httpHeaders.setContentType(MediaType.TEXT_PLAIN); |
| 339 | + given(mockResponse.getStatusCode()).willReturn(HttpStatus.NOT_FOUND); |
| 340 | + given(mockResponse.getRawStatusCode()).willReturn(HttpStatus.NOT_FOUND.value()); |
| 341 | + given(mockResponse.getBody()).willReturn(body); |
| 342 | + |
| 343 | + List<HttpMessageReader<?>> messageReaders = Collections.singletonList( |
| 344 | + new DecoderHttpMessageReader<>(new ByteArrayDecoder())); |
| 345 | + given(mockExchangeStrategies.messageReaders()).willReturn(messageReaders); |
| 346 | + |
| 347 | + Mono<WebClientResponseException> resultMono = defaultClientResponse.createException(); |
| 348 | + WebClientResponseException exception = resultMono.block(); |
| 349 | + assertThat(exception.getStatusCode()).isEqualTo(HttpStatus.NOT_FOUND); |
| 350 | + assertThat(exception.getMessage()).isEqualTo("404 Not Found"); |
| 351 | + assertThat(exception.getHeaders()).containsExactly(entry("Content-Type", |
| 352 | + Collections.singletonList("text/plain"))); |
| 353 | + assertThat(exception.getResponseBodyAsByteArray()).isEqualTo(bytes); |
| 354 | + } |
| 355 | + |
331 | 356 |
|
332 | 357 | private void mockTextPlainResponse(Flux<DataBuffer> body) {
|
333 | 358 | httpHeaders.setContentType(MediaType.TEXT_PLAIN);
|
|
0 commit comments