238 lines
8.6 KiB
Java
238 lines
8.6 KiB
Java
/*
|
|
* Copyright (c) 2018, 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.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
/*
|
|
* @test
|
|
* @summary Ensure that the POST method is retied when the property is set.
|
|
* @run testng/othervm -Djdk.httpclient.enableAllMethodRetry RetryPost
|
|
* @run testng/othervm -Djdk.httpclient.enableAllMethodRetry=true RetryPost
|
|
*/
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import java.io.UncheckedIOException;
|
|
import java.net.InetAddress;
|
|
import java.net.InetSocketAddress;
|
|
import java.net.ServerSocket;
|
|
import java.net.Socket;
|
|
import java.net.URI;
|
|
import java.net.http.HttpClient;
|
|
import java.net.http.HttpRequest;
|
|
import java.net.http.HttpRequest.BodyPublishers;
|
|
import java.net.http.HttpResponse;
|
|
import org.testng.annotations.AfterTest;
|
|
import org.testng.annotations.BeforeTest;
|
|
import org.testng.annotations.DataProvider;
|
|
import org.testng.annotations.Test;
|
|
import static java.lang.System.out;
|
|
import static java.net.http.HttpClient.Builder.NO_PROXY;
|
|
import static java.net.http.HttpResponse.BodyHandlers.ofString;
|
|
import static java.nio.charset.StandardCharsets.US_ASCII;
|
|
import static org.testng.Assert.assertEquals;
|
|
|
|
public class RetryPost {
|
|
|
|
FixedLengthServer fixedLengthServer;
|
|
String httpURIFixLen;
|
|
|
|
static final String RESPONSE_BODY =
|
|
"You use a glass mirror to see your face: you use works of art to see your soul.";
|
|
|
|
@DataProvider(name = "uris")
|
|
public Object[][] variants() {
|
|
return new Object[][] {
|
|
{ httpURIFixLen, true },
|
|
{ httpURIFixLen, false },
|
|
};
|
|
}
|
|
|
|
static final int ITERATION_COUNT = 3;
|
|
|
|
static final String REQUEST_BODY = "Body";
|
|
|
|
@Test(dataProvider = "uris")
|
|
void testSynchronousPOST(String url, boolean sameClient) throws Exception {
|
|
out.print("---\n");
|
|
HttpClient client = null;
|
|
for (int i=0; i< ITERATION_COUNT; i++) {
|
|
if (!sameClient || client == null)
|
|
client = HttpClient.newBuilder().proxy(NO_PROXY).build();
|
|
HttpRequest request = HttpRequest.newBuilder(URI.create(url))
|
|
.POST(BodyPublishers.ofString(REQUEST_BODY))
|
|
.build();
|
|
HttpResponse<String> response = client.send(request, ofString());
|
|
String body = response.body();
|
|
out.println(response + ": " + body);
|
|
assertEquals(response.statusCode(), 200);
|
|
assertEquals(body, RESPONSE_BODY);
|
|
}
|
|
}
|
|
|
|
@Test(dataProvider = "uris")
|
|
void testAsynchronousPOST(String url, boolean sameClient) {
|
|
out.print("---\n");
|
|
HttpClient client = null;
|
|
for (int i=0; i< ITERATION_COUNT; i++) {
|
|
if (!sameClient || client == null)
|
|
client = HttpClient.newBuilder().proxy(NO_PROXY).build();
|
|
HttpRequest request = HttpRequest.newBuilder(URI.create(url))
|
|
.POST(BodyPublishers.ofString(REQUEST_BODY))
|
|
.build();
|
|
client.sendAsync(request, ofString())
|
|
.thenApply(r -> { out.println(r + ": " + r.body()); return r; })
|
|
.thenApply(r -> { assertEquals(r.statusCode(), 200); return r; })
|
|
.thenApply(HttpResponse::body)
|
|
.thenAccept(b -> assertEquals(b, RESPONSE_BODY))
|
|
.join();
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* A server that issues a valid fixed-length reply on even requests, and
|
|
* immediately closes the connection on odd requests ( tick-tock ).
|
|
*/
|
|
static class FixedLengthServer extends Thread implements AutoCloseable {
|
|
|
|
static final String RESPONSE_HEADERS =
|
|
"HTTP/1.1 200 OK\r\n" +
|
|
"Content-Type: text/html; charset=utf-8\r\n" +
|
|
"Content-Length: " + RESPONSE_BODY.length() + "\r\n" +
|
|
"Connection: close\r\n\r\n";
|
|
|
|
static final String RESPONSE = RESPONSE_HEADERS + RESPONSE_BODY;
|
|
|
|
private final ServerSocket ss;
|
|
private volatile boolean closed;
|
|
private int invocationTimes;
|
|
|
|
FixedLengthServer() throws IOException {
|
|
super("FixedLengthServer");
|
|
ss = new ServerSocket();
|
|
ss.bind(new InetSocketAddress(InetAddress.getLoopbackAddress(), 0));
|
|
this.start();
|
|
}
|
|
|
|
public int getPort() { return ss.getLocalPort(); }
|
|
|
|
@Override
|
|
public void run() {
|
|
while (!closed) {
|
|
try (Socket s = ss.accept()) {
|
|
invocationTimes++;
|
|
out.print("FixedLengthServer: got connection ");
|
|
if ((invocationTimes & 0x1) == 0x1) {
|
|
out.println(" closing immediately");
|
|
s.close();
|
|
continue;
|
|
}
|
|
InputStream is = s.getInputStream();
|
|
URI requestMethod = readRequestMethod(is);
|
|
out.print(requestMethod + " ");
|
|
URI uriPath = readRequestPath(is);
|
|
out.println(uriPath);
|
|
readRequestHeaders(is);
|
|
byte[] body = is.readNBytes(4);
|
|
assert body.length == REQUEST_BODY.length() :
|
|
"Unexpected request body " + body.length;
|
|
|
|
OutputStream os = s.getOutputStream();
|
|
out.println("Server: writing response bytes");
|
|
byte[] responseBytes = RESPONSE.getBytes(US_ASCII);
|
|
os.write(responseBytes);
|
|
} catch (IOException e) {
|
|
if (!closed)
|
|
throw new UncheckedIOException("Unexpected", e);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void close() {
|
|
if (closed)
|
|
return;
|
|
closed = true;
|
|
try {
|
|
ss.close();
|
|
} catch (IOException e) {
|
|
throw new UncheckedIOException("Unexpected", e);
|
|
}
|
|
}
|
|
|
|
static final byte[] requestEnd = new byte[] { '\r', '\n', '\r', '\n' };
|
|
|
|
// Read the request method
|
|
static URI readRequestMethod(InputStream is) throws IOException {
|
|
StringBuilder sb = new StringBuilder();
|
|
int r;
|
|
while ((r = is.read()) != -1 && r != 0x20) {
|
|
sb.append((char)r);
|
|
}
|
|
return URI.create(sb.toString());
|
|
}
|
|
|
|
// Read the request URI path
|
|
static URI readRequestPath(InputStream is) throws IOException {
|
|
StringBuilder sb = new StringBuilder();
|
|
int r;
|
|
while ((r = is.read()) != -1 && r != 0x20) {
|
|
sb.append((char)r);
|
|
}
|
|
return URI.create(sb.toString());
|
|
}
|
|
|
|
// Read until the end of a HTTP request headers
|
|
static void readRequestHeaders(InputStream is) throws IOException {
|
|
int requestEndCount = 0, r;
|
|
while ((r = is.read()) != -1) {
|
|
if (r == requestEnd[requestEndCount]) {
|
|
requestEndCount++;
|
|
if (requestEndCount == 4) {
|
|
break;
|
|
}
|
|
} else {
|
|
requestEndCount = 0;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static String serverAuthority(FixedLengthServer server) {
|
|
return InetAddress.getLoopbackAddress().getHostName() + ":"
|
|
+ server.getPort();
|
|
}
|
|
|
|
@BeforeTest
|
|
public void setup() throws Exception {
|
|
fixedLengthServer = new FixedLengthServer();
|
|
httpURIFixLen = "http://" + serverAuthority(fixedLengthServer)
|
|
+ "/http1/fixed/baz";
|
|
}
|
|
|
|
@AfterTest
|
|
public void teardown() throws Exception {
|
|
fixedLengthServer.close();
|
|
}
|
|
}
|