zentol commented on a change in pull request #6763: [FLINK-10415] Fail response
future if connection closes in RestClient
URL: https://github.com/apache/flink/pull/6763#discussion_r220638812
##########
File path:
flink-runtime/src/test/java/org/apache/flink/runtime/rest/RestClientTest.java
##########
@@ -89,7 +91,78 @@ public void testInvalidVersionRejection() throws Exception {
} catch (IllegalArgumentException e) {
// expected
}
+ }
+ /**
+ * Tests that we fail the operation if the remote connection closes.
+ */
+ @Test
+ public void testConnectionClosedHandling() throws Exception {
+ final ServerSocket serverSocket = new ServerSocket(0);
+ final String targetAddress = "localhost";
+ final int targetPort = serverSocket.getLocalPort();
+
+ try (final RestClient restClient = new
RestClient(RestClientConfiguration.fromConfiguration(new Configuration()),
TestingUtils.defaultExecutor())) {
+ final CompletableFuture<EmptyResponseBody>
responseFuture = restClient.sendRequest(
+ targetAddress,
+ targetPort,
+ new TestMessageHeaders(),
+ EmptyMessageParameters.getInstance(),
+ EmptyRequestBody.getInstance(),
+ Collections.emptyList());
+
+ // establish connection
+ final Socket connectionSocket = serverSocket.accept();
+
+ // close connection
+ connectionSocket.close();
+
+ try {
+ responseFuture.get();
+ } catch (ExecutionException ee) {
+ assertThat(ExceptionUtils.findThrowable(ee,
ConnectionClosedException.class).isPresent(), is(true));
+ }
+ } finally {
+ serverSocket.close();
+ }
+ }
+
+ /**
+ * Tests that we fail the operation if the client closes.
+ */
+ @Test
+ public void testRestClientClosedHandling() throws Exception {
+ final ServerSocket serverSocket = new ServerSocket(0);
+ final String targetAddress = "localhost";
+ final int targetPort = serverSocket.getLocalPort();
+ Socket connectionSocket = null;
+
+ try (final RestClient restClient = new
RestClient(RestClientConfiguration.fromConfiguration(new Configuration()),
TestingUtils.defaultExecutor())) {
+ final CompletableFuture<EmptyResponseBody>
responseFuture = restClient.sendRequest(
Review comment:
but it does fail with a different exception, no? although from a
user-perspective the same issue happened.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
