parent
417bce8be5
commit
d8191b4808
|
|
@ -1,5 +1,5 @@
|
||||||
/*
|
/*
|
||||||
* Copyright 2002-2019 the original author or authors.
|
* Copyright 2002-2021 the original author or authors.
|
||||||
*
|
*
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
* you may not use this file except in compliance with the License.
|
* you may not use this file except in compliance with the License.
|
||||||
|
|
@ -42,7 +42,7 @@ class MessageBodyClientHttpResponseWrapper implements ClientHttpResponse {
|
||||||
private PushbackInputStream pushbackInputStream;
|
private PushbackInputStream pushbackInputStream;
|
||||||
|
|
||||||
|
|
||||||
public MessageBodyClientHttpResponseWrapper(ClientHttpResponse response) throws IOException {
|
public MessageBodyClientHttpResponseWrapper(ClientHttpResponse response) {
|
||||||
this.response = response;
|
this.response = response;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -13,43 +13,44 @@
|
||||||
* See the License for the specific language governing permissions and
|
* See the License for the specific language governing permissions and
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
package org.springframework.web.client;
|
package org.springframework.web.client;
|
||||||
|
|
||||||
|
import java.io.ByteArrayInputStream;
|
||||||
|
import java.io.InputStream;
|
||||||
|
|
||||||
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.Test;
|
||||||
import org.springframework.http.HttpHeaders;
|
|
||||||
import org.springframework.http.HttpStatus;
|
|
||||||
import org.springframework.http.client.ClientHttpResponse;
|
import org.springframework.http.client.ClientHttpResponse;
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
import static org.mockito.BDDMockito.given;
|
import static org.mockito.BDDMockito.given;
|
||||||
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.mock;
|
||||||
|
|
||||||
import java.io.ByteArrayInputStream;
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.io.InputStream;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Unit tests for {MessageBodyClientHttpResponseWrapper}.
|
* Unit tests for {@link MessageBodyClientHttpResponseWrapper}.
|
||||||
*
|
*
|
||||||
|
* @since 5.3.10
|
||||||
* @author Yin-Jui Liao
|
* @author Yin-Jui Liao
|
||||||
*/
|
*/
|
||||||
class MessageBodyClientHttpResponseWrapperTests {
|
class MessageBodyClientHttpResponseWrapperTests {
|
||||||
|
|
||||||
private final ClientHttpResponse response = mock(ClientHttpResponse.class);
|
private final ClientHttpResponse response = mock(ClientHttpResponse.class);
|
||||||
|
|
||||||
|
private final MessageBodyClientHttpResponseWrapper responseWrapper = new MessageBodyClientHttpResponseWrapper(response);
|
||||||
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
void testMessageBodyNotExist() throws IOException {
|
void messageBodyDoesNotExist() throws Exception {
|
||||||
given(response.getBody()).willReturn(null);
|
given(response.getBody()).willReturn(null);
|
||||||
MessageBodyClientHttpResponseWrapper responseWrapper = new MessageBodyClientHttpResponseWrapper(response);
|
|
||||||
assertThat(responseWrapper.hasEmptyMessageBody()).isTrue();
|
assertThat(responseWrapper.hasEmptyMessageBody()).isTrue();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
void testMessageBodyExist() throws IOException {
|
void messageBodyExists() throws Exception {
|
||||||
String body = "Accepted request";
|
InputStream stream = new ByteArrayInputStream("content".getBytes());
|
||||||
InputStream stream = new ByteArrayInputStream(body.getBytes());
|
|
||||||
given(response.getBody()).willReturn(stream);
|
given(response.getBody()).willReturn(stream);
|
||||||
MessageBodyClientHttpResponseWrapper responseWrapper = new MessageBodyClientHttpResponseWrapper(response);
|
|
||||||
assertThat(responseWrapper.hasEmptyMessageBody()).isFalse();
|
assertThat(responseWrapper.hasEmptyMessageBody()).isFalse();
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue