/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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.apache.cxf.systest.jaxrs;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import javax.ws.rs.NotFoundException;
import javax.ws.rs.ProcessingException;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.client.Entity;
import javax.ws.rs.client.InvocationCallback;
import javax.ws.rs.client.ResponseProcessingException;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.Response;
import javax.ws.rs.ext.MessageBodyReader;
import javax.ws.rs.ext.MessageBodyWriter;
import javax.xml.ws.Holder;
import org.apache.cxf.jaxrs.client.WebClient;
import org.apache.cxf.jaxrs.model.AbstractResourceInfo;
import org.apache.cxf.testutil.common.AbstractBusClientServerTestBase;
import org.junit.BeforeClass;
import org.junit.Test;
public class JAXRSAsyncClientTest extends AbstractBusClientServerTestBase {
public static final String PORT = BookServerAsyncClient.PORT;
@BeforeClass
public static void startServers() throws Exception {
AbstractResourceInfo.clearAllMaps();
assertTrue("server did not launch correctly",
launchServer(BookServerAsyncClient.class, true));
createStaticBus();
}
@Test
public void testRetrieveBookCustomMethodAsyncSync() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/retrieve";
WebClient wc = WebClient.create(address);
wc.type("application/xml").accept("application/xml");
WebClient.getConfig(wc).getRequestContext().put("use.async.http.conduit", true);
Book book = wc.invoke("RETRIEVE", new Book("Retrieve", 123L), Book.class);
assertEquals("Retrieve", book.getName());
}
@Test
public void testDeleteWithBody() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/deletebody";
WebClient wc = WebClient.create(address);
wc.type("application/xml").accept("application/xml");
WebClient.getConfig(wc).getRequestContext().put("use.async.http.conduit", true);
Book book = wc.invoke("DELETE", new Book("Delete", 123L), Book.class);
assertEquals("Delete", book.getName());
}
@Test
public void testRetrieveBookCustomMethodAsync() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/retrieve";
WebClient wc = WebClient.create(address);
wc.accept("application/xml");
Future<Book> book = wc.async().method("RETRIEVE", Entity.xml(new Book("Retrieve", 123L)),
Book.class);
assertEquals("Retrieve", book.get().getName());
}
@Test
public void testGetBookAsyncResponse404() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/bookheaders/404";
WebClient wc = createWebClient(address);
Future<Response> future = wc.async().get(Response.class);
assertEquals(404, future.get().getStatus());
}
@Test
public void testGetBookAsync404() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/bookheaders/404";
WebClient wc = createWebClient(address);
Future<Book> future = wc.async().get(Book.class);
try {
future.get();
fail("Exception expected");
} catch (ExecutionException ex) {
assertTrue(ex.getCause() instanceof NotFoundException);
}
}
@Test
public void testPostBookProcessingException() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/";
List<Object> providers = new ArrayList<Object>();
providers.add(new FaultyBookWriter());
WebClient wc = WebClient.create(address, providers);
Future<Book> future = wc.async().post(Entity.xml(new Book()), Book.class);
try {
future.get();
fail("Exception expected");
} catch (ExecutionException ex) {
assertTrue(ex.getCause() instanceof ProcessingException);
}
}
@Test
public void testGetBookResponseProcessingException() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/books/123";
List<Object> providers = new ArrayList<Object>();
providers.add(new FaultyBookReader());
WebClient wc = WebClient.create(address, providers);
Future<Book> future = wc.async().get(Book.class);
try {
future.get();
fail("Exception expected");
} catch (ExecutionException ex) {
assertTrue(ex.getCause() instanceof ResponseProcessingException);
}
}
@Test
public void testGetBookAsync404Callback() throws Exception {
String address = "http://localhost:" + PORT + "/bookstore/bookheaders/404";
WebClient wc = createWebClient(address);
final Holder<Object> holder = new Holder<Object>();
InvocationCallback<Object> callback = createCallback(holder);
try {
wc.async().get(callback).get();
fail("Exception expected");
} catch (ExecutionException ex) {
assertTrue(ex.getCause() instanceof NotFoundException);
assertTrue(ex.getCause() == holder.value);
}
}
private WebClient createWebClient(String address) {
List<Object> providers = new ArrayList<Object>();
WebClient wc = WebClient.create(address, providers);
WebClient.getConfig(wc).getHttpConduit().getClient().setReceiveTimeout(1000000L);
return wc;
}
private InvocationCallback<Object> createCallback(final Holder<Object> holder) {
return new InvocationCallback<Object>() {
public void completed(Object response) {
holder.value = response;
}
public void failed(Throwable error) {
holder.value = error;
}
};
}
private static class FaultyBookWriter implements MessageBodyWriter<Book> {
@Override
public long getSize(Book arg0, Class<?> arg1, Type arg2, Annotation[] arg3, MediaType arg4) {
// TODO Auto-generated method stub
return 0;
}
@Override
public boolean isWriteable(Class<?> arg0, Type arg1, Annotation[] arg2, MediaType arg3) {
return true;
}
@Override
public void writeTo(Book arg0, Class<?> arg1, Type arg2, Annotation[] arg3, MediaType arg4,
MultivaluedMap<String, Object> arg5, OutputStream arg6) throws IOException,
WebApplicationException {
throw new RuntimeException();
}
}
private static class FaultyBookReader implements MessageBodyReader<Book> {
@Override
public boolean isReadable(Class<?> arg0, Type arg1, Annotation[] arg2, MediaType arg3) {
return true;
}
@Override
public Book readFrom(Class<Book> arg0, Type arg1, Annotation[] arg2, MediaType arg3,
MultivaluedMap<String, String> arg4, InputStream arg5) throws IOException,
WebApplicationException {
throw new RuntimeException();
}
}
}