/*
* Copyright 2010 Ning, Inc.
*
* Ning licenses this file to you under the Apache License, version 2.0
* (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package org.asynchttpclient.async;
import static org.asynchttpclient.async.util.TestUtils.SIMPLE_TEXT_FILE;
import static org.asynchttpclient.async.util.TestUtils.SIMPLE_TEXT_FILE_STRING;
import static org.asynchttpclient.async.util.TestUtils.createSSLContext;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertTrue;
import org.asynchttpclient.AsyncHttpClient;
import org.asynchttpclient.AsyncHttpClientConfig.Builder;
import org.asynchttpclient.Response;
import org.testng.Assert;
import org.testng.annotations.Test;
import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.SSLHandshakeException;
import javax.net.ssl.SSLSession;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.ConnectException;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
public abstract class BasicHttpsTest extends AbstractBasicHttpsTest {
protected String getTargetUrl() {
return String.format("https://127.0.0.1:%d/foo/test", port1);
}
@Test(groups = { "standalone", "default_provider" })
public void zeroCopyPostTest() throws Exception {
final AsyncHttpClient client = getAsyncHttpClient(new Builder().setSSLContext(createSSLContext(new AtomicBoolean(true))).build());
try {
Response resp = client.preparePost(getTargetUrl()).setBody(SIMPLE_TEXT_FILE).setHeader("Content-Type", "text/html").execute().get();
assertNotNull(resp);
assertEquals(resp.getStatusCode(), HttpServletResponse.SC_OK);
assertEquals(resp.getResponseBody(), SIMPLE_TEXT_FILE_STRING);
} finally {
client.close();
}
}
@Test(groups = { "standalone", "default_provider" })
public void multipleSSLRequestsTest() throws Exception {
final AsyncHttpClient c = getAsyncHttpClient(new Builder().setSSLContext(createSSLContext(new AtomicBoolean(true))).build());
try {
String body = "hello there";
// once
Response response = c.preparePost(getTargetUrl()).setBody(body).setHeader("Content-Type", "text/html").execute().get(TIMEOUT, TimeUnit.SECONDS);
assertEquals(response.getResponseBody(), body);
// twice
response = c.preparePost(getTargetUrl()).setBody(body).setHeader("Content-Type", "text/html").execute().get(TIMEOUT, TimeUnit.SECONDS);
assertEquals(response.getResponseBody(), body);
} finally {
c.close();
}
}
@Test(groups = { "standalone", "default_provider" })
public void multipleSSLWithoutCacheTest() throws Exception {
AsyncHttpClient c = getAsyncHttpClient(new Builder().setSSLContext(createSSLContext(new AtomicBoolean(true))).setAllowPoolingSslConnections(false).build());
try {
String body = "hello there";
c.preparePost(getTargetUrl()).setBody(body).setHeader("Content-Type", "text/html").execute();
c.preparePost(getTargetUrl()).setBody(body).setHeader("Content-Type", "text/html").execute();
Response response = c.preparePost(getTargetUrl()).setBody(body).setHeader("Content-Type", "text/html").execute().get();
assertEquals(response.getResponseBody(), body);
} finally {
c.close();
}
}
@Test(groups = { "standalone", "default_provider" })
public void reconnectsAfterFailedCertificationPath() throws Exception {
AtomicBoolean trusted = new AtomicBoolean(false);
AsyncHttpClient c = getAsyncHttpClient(new Builder().setSSLContext(createSSLContext(trusted)).build());
try {
String body = "hello there";
// first request fails because server certificate is rejected
Throwable cause = null;
try {
c.preparePost(getTargetUrl()).setBody(body).setHeader("Content-Type", "text/html").execute().get(TIMEOUT, TimeUnit.SECONDS);
} catch (final ExecutionException e) {
cause = e.getCause();
if (cause instanceof ConnectException) {
//assertNotNull(cause.getCause());
assertTrue(cause.getCause() instanceof SSLHandshakeException, "Expected an SSLHandshakeException, got a " + cause.getCause());
} else {
assertTrue(cause instanceof IOException, "Expected an IOException, got a " + cause);
}
}
assertNotNull(cause);
trusted.set(true);
// second request should succeed
Response response = c.preparePost(getTargetUrl()).setBody(body).setHeader("Content-Type", "text/html").execute().get(TIMEOUT, TimeUnit.SECONDS);
assertEquals(response.getResponseBody(), body);
} finally {
c.close();
}
}
@Test(timeOut = 5000)
public void failInstantlyIfHostNamesDiffer() throws Exception {
AsyncHttpClient client = null;
try {
final Builder builder = new Builder().setHostnameVerifier(new HostnameVerifier() {
public boolean verify(String arg0, SSLSession arg1) {
return false;
}
}).setRequestTimeout(20000);
client = getAsyncHttpClient(builder.build());
try {
client.prepareGet("https://github.com/AsyncHttpClient/async-http-client/issues/355").execute().get(TIMEOUT, TimeUnit.SECONDS);
Assert.assertTrue(false, "Shouldn't be here: should get an Exception");
} catch (ExecutionException e) {
Assert.assertTrue(e.getCause() instanceof ConnectException, "Cause should be a ConnectException");
} catch (Exception e) {
Assert.assertTrue(false, "Shouldn't be here: should get a ConnectException wrapping a ConnectException");
}
} finally {
client.close();
}
}
}