import static org.junit.Assert.assertEquals;
import org.junit.Before;
import org.junit.Test;
-import org.junit.runner.RunWith;
-import org.junit.runners.Parameterized;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-import org.springframework.http.client.reactive.ClientHttpConnector;
+import org.springframework.http.HttpStatus;
import org.springframework.http.client.reactive.JettyClientHttpConnector;
-import org.springframework.http.client.reactive.ReactorClientHttpConnector;
+import org.springframework.web.reactive.function.client.ClientRequest;
+import org.springframework.web.reactive.function.client.ClientResponse;
+import org.springframework.web.reactive.function.client.ExchangeFunction;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;
import reactor.test.StepVerifier;
*
* @author Kai Moritz
*/
-@RunWith(Parameterized.class)
public class GraphApiExchangeFilterFunctionIntegrationTest
{
private static final Logger LOG =
private MockWebServer server;
private WebClient webClient;
- @Parameterized.Parameter(0)
- public ClientHttpConnector connector;
-
- @Parameterized.Parameters(name = "webClient [{0}]")
- public static Object[][] arguments()
- {
- return new Object[][]
- {
- { new JettyClientHttpConnector() },
- { new ReactorClientHttpConnector() }
- };
- }
-
@Before
public void setup()
webClient =
WebClient
.builder()
- .clientConnector(this.connector)
+ .clientConnector(new JettyClientHttpConnector())
.baseUrl(server.url("/").toString())
- .filter(GraphApiExchangeFilterFunction.INSTANCE)
+ .filter(this::errorHandlingFilter)
.build();
}
@Test
public void testValidError()
{
- LOG.info("testValidError");
-
-
server
.enqueue(new MockResponse()
- .setStatus("400")
+ .setResponseCode(HttpStatus.BAD_REQUEST.value())
.setHeader("Content-Type", "application/json")
.setBody(
"{\n" +
})
.verify(Duration.ofSeconds(3));
}
+
+ @Test
+ public void testTest()
+ {
+ server
+ .enqueue(new MockResponse()
+ .setResponseCode(400)
+ .setHeader("Content-Type", "text/plain")
+ .setBody("Hello Spring!"));
+
+ Mono<String> result;
+
+ result = webClient.get()
+ .uri("/greeting?name=Spring")
+ .retrieve()
+ .bodyToMono(String.class);
+
+ StepVerifier.create(result)
+ .expectError(Exception.class)
+ .verify(Duration.ofSeconds(3));
+
+ server
+ .enqueue(new MockResponse()
+ .setResponseCode(200)
+ .setHeader("Content-Type", "text/plain")
+ .setHeader("Foo", "Bar")
+ .setBody("Hello Spring!"));
+
+ result = webClient.get()
+ .uri("/greeting?name=Spring")
+ .retrieve()
+ .bodyToMono(String.class);
+
+ StepVerifier.create(result)
+ .expectNext("Hello Spring!")
+ .expectComplete()
+ .verify(Duration.ofSeconds(3));
+ }
+
+ public Mono<ClientResponse> errorHandlingFilter(ClientRequest request, ExchangeFunction next)
+ {
+ return
+ next
+ .exchange(request)
+ .flatMap(response ->
+ {
+ return
+ HttpStatus.Series.CLIENT_ERROR.equals(response.statusCode().series())
+ ? response
+ .bodyToMono(String.class)
+ .flatMap(errorBody -> Mono.error(GraphApiException.create(
+ response.statusCode(),
+ response.headers().asHttpHeaders(),
+ errorBody.getBytes())))
+ : Mono.just(response);
+ });
+ }
}