< prev index next >
test/jdk/java/net/httpclient/CancelledResponse.java
Print this page
@@ -1,7 +1,7 @@
/*
- * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
+ * Copyright (c) 2015, 2017, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
@@ -19,108 +19,323 @@
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
-
-import java.io.IOException;
import jdk.incubator.http.HttpClient;
+import jdk.incubator.http.HttpClient.Version;
+import jdk.incubator.http.HttpHeaders;
import jdk.incubator.http.HttpRequest;
import jdk.incubator.http.HttpResponse;
+import jdk.jshell.spi.ExecutionControl;
+import jdk.testlibrary.SimpleSSLContext;
+
+import javax.net.ServerSocketFactory;
+import javax.net.ssl.SSLContext;
+import javax.net.ssl.SSLException;
+import javax.net.ssl.SSLServerSocketFactory;
+import java.io.IOException;
+import java.net.SocketException;
import java.net.URI;
+import java.nio.ByteBuffer;
+import java.util.List;
import java.util.concurrent.CompletableFuture;
-import java.util.concurrent.Executor;
-import java.util.concurrent.ExecutorService;
-import static jdk.incubator.http.HttpResponse.BodyHandler.asString;
+import java.util.concurrent.CompletionException;
+import java.util.concurrent.CompletionStage;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Flow;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import jdk.incubator.http.HttpResponse.BodyHandler;
+import jdk.incubator.http.HttpResponse.BodySubscriber;
+
+import static java.lang.String.format;
+import static java.lang.System.out;
/**
* @test
* @bug 8087112
- * @key intermittent
- * @build Server
- * @run main/othervm -Djava.net.HttpClient.log=all SplitResponse
+ * @library /lib/testlibrary
+ * @build jdk.testlibrary.SimpleSSLContext
+ * @build MockServer
+ * @run main/othervm CancelledResponse
+ * @run main/othervm CancelledResponse SSL
*/
/**
- * Similar test to QuickResponses except that each byte of the response
- * is sent in a separate packet, which tests the stability of the implementation
- * for receiving unusual packet sizes.
+ * Similar test to SplitResponse except that the client will cancel the response
+ * before receiving it fully.
*/
-public class SplitResponse {
+public class CancelledResponse {
- static Server server;
-
- static String response(String body) {
- return "HTTP/1.1 200 OK\r\nConnection: Close\r\nContent-length: "
- + Integer.toString(body.length())
- + "\r\n\r\n" + body;
+ static String response(String body, boolean serverKeepalive) {
+ StringBuilder sb = new StringBuilder();
+ sb.append("HTTP/1.1 200 OK\r\n");
+ if (!serverKeepalive)
+ sb.append("Connection: Close\r\n");
+
+ sb.append("Content-length: ").append(body.length()).append("\r\n");
+ sb.append("\r\n");
+ sb.append(body);
+ return sb.toString();
}
static final String responses[] = {
- "Lorem ipsum",
- "dolor sit amet",
- "consectetur adipiscing elit, sed do eiusmod tempor",
- "quis nostrud exercitation ullamco",
- "laboris nisi",
- "ut",
- "aliquip ex ea commodo consequat." +
- "Duis aute irure dolor in reprehenderit in voluptate velit esse" +
- "cillum dolore eu fugiat nulla pariatur.",
- "Excepteur sint occaecat cupidatat non proident."
+ "Lorem ipsum dolor sit amet consectetur adipiscing elit,",
+ "sed do eiusmod tempor quis nostrud exercitation ullamco laboris nisi ut",
+ "aliquip ex ea commodo consequat."
};
+ final ServerSocketFactory factory;
+ final SSLContext context;
+ final boolean useSSL;
+ CancelledResponse(boolean useSSL) throws IOException {
+ this.useSSL = useSSL;
+ context = new SimpleSSLContext().get();
+ SSLContext.setDefault(context);
+ factory = useSSL ? SSLServerSocketFactory.getDefault()
+ : ServerSocketFactory.getDefault();
+ }
+
+ public HttpClient newHttpClient() {
+ HttpClient client;
+ if (useSSL) {
+ client = HttpClient.newBuilder()
+ .sslContext(context)
+ .build();
+ } else {
+ client = HttpClient.newHttpClient();
+ }
+ return client;
+ }
+
public static void main(String[] args) throws Exception {
- server = new Server(0);
+ boolean useSSL = false;
+ if (args != null && args.length == 1) {
+ useSSL = "SSL".equals(args[0]);
+ }
+ CancelledResponse sp = new CancelledResponse(useSSL);
+
+ for (Version version : Version.values()) {
+ for (boolean serverKeepalive : new boolean[]{ true, false }) {
+ // Note: the mock server doesn't support Keep-Alive, but
+ // pretending that it might exercises code paths in and out of
+ // the connection pool, and retry logic
+ for (boolean async : new boolean[]{ true, false }) {
+ sp.test(version, serverKeepalive, async);
+ }
+ }
+ }
+ }
+
+ static class CancelException extends IOException {
+ }
+
+ // @Test
+ void test(Version version, boolean serverKeepalive, boolean async)
+ throws Exception
+ {
+ out.println(format("*** version %s, serverKeepAlive: %s, async: %s ***",
+ version, serverKeepalive, async));
+ MockServer server = new MockServer(0, factory);
URI uri = new URI(server.getURL());
+ out.println("server is: " + uri);
server.start();
- HttpClient client = HttpClient.newHttpClient();
- HttpRequest request = HttpRequest.newBuilder(uri).build();
- HttpResponse<String> r;
+ HttpClient client = newHttpClient();
+ HttpRequest request = HttpRequest.newBuilder(uri).version(version).build();
+ try {
+ for (int i = 0; i < responses.length; i++) {
+ HttpResponse<String> r = null;
CompletableFuture<HttpResponse<String>> cf1;
+ CancelException expected = null;
+ AtomicBoolean cancelled = new AtomicBoolean();
- try {
- for (int i=0; i<responses.length; i++) {
- cf1 = client.sendAsync(request, asString());
+ out.println("----- iteration " + i + " -----");
String body = responses[i];
+ Thread t = sendSplitResponse(response(body, serverKeepalive), server, cancelled);
- Server.Connection c = server.activity();
- sendSplitResponse(response(body), c);
+ try {
+ if (async) {
+ out.println("send async: " + request);
+ cf1 = client.sendAsync(request, asString(body, cancelled));
r = cf1.get();
- if (r.statusCode()!= 200)
+ } else { // sync
+ out.println("send sync: " + request);
+ r = client.send(request, asString(body, cancelled));
+ }
+ } catch (CancelException c1) {
+ System.out.println("Got expected exception: " + c1);
+ expected = c1;
+ } catch (IOException | ExecutionException | CompletionException c2) {
+ Throwable c = c2;
+ while (c != null && !(c instanceof CancelException)) {
+ c = c.getCause();
+ }
+ if (c instanceof CancelException) {
+ System.out.println("Got expected exception: " + c);
+ expected = (CancelException)c;
+ } else throw c2;
+ }
+ if (r != null) {
+ if (r.statusCode() != 200)
throw new RuntimeException("Failed");
String rxbody = r.body();
- System.out.println("received " + rxbody);
+ out.println("received " + rxbody);
if (!rxbody.equals(body))
- throw new RuntimeException("Failed");
- c.close();
+ throw new RuntimeException(format("Expected:%s, got:%s", body, rxbody));
}
- } finally {
- Executor def = client.executor();
- if (def instanceof ExecutorService) {
- ((ExecutorService)def).shutdownNow();
+ t.join();
+ conn.close();
+ if (expected == null) {
+ throw new RuntimeException("Expected exception not raised for "
+ + i + " cancelled=" + cancelled.get());
}
}
+ } finally {
+ server.close();
+ }
System.out.println("OK");
}
- // send the response one byte at a time with a small delay between bytes
- // to ensure that each byte is read in a separate read
- static void sendSplitResponse(String s, Server.Connection conn) {
+ static class CancellingSubscriber implements BodySubscriber<String> {
+ private final String expected;
+ private final CompletableFuture<String> result;
+ private Flow.Subscription subscription;
+ final AtomicInteger index = new AtomicInteger();
+ final AtomicBoolean cancelled;
+ CancellingSubscriber(String expected, AtomicBoolean cancelled) {
+ this.cancelled = cancelled;
+ this.expected = expected;
+ result = new CompletableFuture<>();
+ }
+
+ @Override
+ public CompletionStage<String> getBody() {
+ return result;
+ }
+
+ @Override
+ public void onSubscribe(Flow.Subscription subscription) {
+ this.subscription = subscription;
+ subscription.request(1);
+ }
+
+ @Override
+ public void onNext(List<ByteBuffer> item) {
+ //if (result.isDone())
+ for (ByteBuffer b : item) {
+ while (b.hasRemaining() && !result.isDone()) {
+ int i = index.getAndIncrement();
+ char at = expected.charAt(i);
+ byte[] data = new byte[b.remaining()];
+ b.get(data); // we know that the server writes 1 char
+ String s = new String(data);
+ char c = s.charAt(0);
+ if (c != at) {
+ Throwable x = new IllegalStateException("char at "
+ + i + " is '" + c + "' expected '"
+ + at + "' for \"" + expected +"\"");
+ out.println("unexpected char received, cancelling");
+ subscription.cancel();
+ result.completeExceptionally(x);
+ return;
+ }
+ }
+ }
+ if (index.get() > 0 && !result.isDone()) {
+ // we should complete the result here, but let's
+ // see if we get something back...
+ out.println("Cancelling subscription after reading " + index.get());
+ cancelled.set(true);
+ subscription.cancel();
+ result.completeExceptionally(new CancelException());
+ return;
+ }
+ if (!result.isDone()) {
+ out.println("requesting 1 more");
+ subscription.request(1);
+ }
+ }
+
+ @Override
+ public void onError(Throwable throwable) {
+ result.completeExceptionally(throwable);
+ }
+
+ @Override
+ public void onComplete() {
+ int len = index.get();
+ if (len == expected.length()) {
+ result.complete(expected);
+ } else {
+ Throwable x = new IllegalStateException("received only "
+ + len + " chars, expected " + expected.length()
+ + " for \"" + expected +"\"");
+ result.completeExceptionally(x);
+ }
+ }
+ }
+
+ static class CancellingHandler implements BodyHandler<String> {
+ final String expected;
+ final AtomicBoolean cancelled;
+ CancellingHandler(String expected, AtomicBoolean cancelled) {
+ this.expected = expected;
+ this.cancelled = cancelled;
+ }
+ @Override
+ public BodySubscriber<String> apply(int statusCode, HttpHeaders responseHeaders) {
+ assert !cancelled.get();
+ return new CancellingSubscriber(expected, cancelled);
+ }
+ }
+
+ BodyHandler<String> asString(String expected, AtomicBoolean cancelled) {
+ return new CancellingHandler(expected, cancelled);
+ }
+
+ // required for cleanup
+ volatile MockServer.Connection conn;
+
+ // Sends the response, mostly, one byte at a time with a small delay
+ // between bytes, to encourage that each byte is read in a separate read
+ Thread sendSplitResponse(String s, MockServer server, AtomicBoolean cancelled) {
System.out.println("Sending: ");
Thread t = new Thread(() -> {
+ System.out.println("Waiting for server to receive headers");
+ conn = server.activity();
+ System.out.println("Start sending response");
+ int sent = 0;
try {
int len = s.length();
+ out.println("sending " + s);
for (int i = 0; i < len; i++) {
String onechar = s.substring(i, i + 1);
conn.send(onechar);
- Thread.sleep(30);
+ sent++;
+ Thread.sleep(10);
+ }
+ out.println("sent " + s);
+ } catch (SSLException | SocketException x) {
+ // if SSL then we might get a "Broken Pipe", otherwise
+ // a "Socket closed".
+ boolean expected = cancelled.get();
+ if (sent > 0 && expected) {
+ System.out.println("Connection closed by peer as expected: " + x);
+ return;
+ } else {
+ System.out.println("Unexpected exception (sent="
+ + sent + ", cancelled=" + expected + "): " + x);
+ throw new RuntimeException(x);
}
- System.out.println("sent");
} catch (IOException | InterruptedException e) {
+ throw new RuntimeException(e);
}
});
t.setDaemon(true);
t.start();
+ return t;
}
}
< prev index next >