Java tutorial
/** * 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.camel.component.cxf.jaxrs.simplebinding; import java.io.File; import java.io.InputStream; import java.io.StringReader; import java.io.StringWriter; import java.nio.charset.Charset; import java.util.ArrayList; import java.util.List; import javax.activation.DataHandler; import javax.xml.bind.JAXBContext; import org.apache.camel.Exchange; import org.apache.camel.Processor; import org.apache.camel.builder.RouteBuilder; import org.apache.camel.component.cxf.CXFTestSupport; import org.apache.camel.component.cxf.jaxrs.simplebinding.testbean.Customer; import org.apache.camel.component.cxf.jaxrs.simplebinding.testbean.CustomerList; import org.apache.camel.component.cxf.jaxrs.simplebinding.testbean.Order; import org.apache.camel.component.cxf.jaxrs.simplebinding.testbean.Product; import org.apache.camel.test.junit4.CamelTestSupport; import org.apache.cxf.message.MessageContentsList; import org.apache.http.HttpResponse; import org.apache.http.client.HttpClient; import org.apache.http.client.methods.HttpDelete; import org.apache.http.client.methods.HttpGet; import org.apache.http.client.methods.HttpPost; import org.apache.http.client.methods.HttpPut; import org.apache.http.entity.InputStreamEntity; import org.apache.http.entity.StringEntity; import org.apache.http.entity.mime.HttpMultipartMode; import org.apache.http.entity.mime.MultipartEntity; import org.apache.http.entity.mime.content.FileBody; import org.apache.http.entity.mime.content.StringBody; import org.apache.http.impl.client.DefaultHttpClient; import org.apache.http.util.EntityUtils; import org.junit.Test; /** * Tests for the Simple Binding style of CXF JAX-RS consumers. */ public class CxfRsConsumerSimpleBindingTest extends CamelTestSupport { private static final String PORT_PATH = CXFTestSupport.getPort1() + "/CxfRsConsumerTest"; private static final String CXF_RS_ENDPOINT_URI = "cxfrs://http://localhost:" + PORT_PATH + "/rest?resourceClasses=org.apache.camel.component.cxf.jaxrs.simplebinding.testbean.CustomerServiceResource&bindingStyle=SimpleConsumer"; private JAXBContext jaxb; private HttpClient httpclient; public void setUp() throws Exception { super.setUp(); httpclient = new DefaultHttpClient(); jaxb = JAXBContext.newInstance(CustomerList.class, Customer.class, Order.class, Product.class); } public void tearDown() throws Exception { super.tearDown(); httpclient.getConnectionManager().shutdown(); } protected RouteBuilder createRouteBuilder() throws Exception { return new RouteBuilder() { public void configure() { from(CXF_RS_ENDPOINT_URI).recipientList(simple("direct:${header.operationName}")); from("direct:getCustomer").process(new Processor() { public void process(Exchange exchange) throws Exception { assertNotNull(exchange.getIn().getHeader("id")); long id = exchange.getIn().getHeader("id", Long.class); if (id == 123) { assertEquals("123", exchange.getIn().getHeader("id")); assertEquals(MessageContentsList.class, exchange.getIn().getBody().getClass()); exchange.getOut().setBody(new Customer(123, "Raul")); exchange.getOut().setHeader(Exchange.HTTP_RESPONSE_CODE, 200); } else if (id == 456) { exchange.getOut().setHeader(Exchange.HTTP_RESPONSE_CODE, 404); } else { fail(); } } }); from("direct:updateCustomer").process(new Processor() { public void process(Exchange exchange) throws Exception { assertEquals("123", exchange.getIn().getHeader("id")); Customer c = exchange.getIn().getBody(Customer.class); assertEquals(123, c.getId()); assertNotNull(c); } }); from("direct:newCustomer").process(new Processor() { public void process(Exchange exchange) throws Exception { Customer c = exchange.getIn().getBody(Customer.class); assertNotNull(c); assertEquals(123, c.getId()); } }); from("direct:listVipCustomers").process(new Processor() { public void process(Exchange exchange) throws Exception { assertEquals("gold", exchange.getIn().getHeader("status", String.class)); assertEquals(MessageContentsList.class, exchange.getIn().getBody().getClass()); assertEquals(0, exchange.getIn().getBody(MessageContentsList.class).size()); CustomerList response = new CustomerList(); List<Customer> list = new ArrayList<Customer>(2); list.add(new Customer(123, "Raul")); list.add(new Customer(456, "Raul2")); response.setCustomers(list); exchange.getOut().setBody(response); } }); from("direct:updateVipCustomer").process(new Processor() { public void process(Exchange exchange) throws Exception { assertEquals("gold", exchange.getIn().getHeader("status", String.class)); assertEquals("123", exchange.getIn().getHeader("id")); Customer c = exchange.getIn().getBody(Customer.class); assertEquals(123, c.getId()); assertNotNull(c); } }); from("direct:deleteVipCustomer").process(new Processor() { public void process(Exchange exchange) throws Exception { assertEquals("gold", exchange.getIn().getHeader("status", String.class)); assertEquals("123", exchange.getIn().getHeader("id")); } }); from("direct:uploadImageInputStream").process(new Processor() { public void process(Exchange exchange) throws Exception { assertEquals("123", exchange.getIn().getHeader("id")); assertEquals("image/jpeg", exchange.getIn().getHeader("Content-Type")); assertTrue(InputStream.class.isAssignableFrom(exchange.getIn().getBody().getClass())); InputStream is = exchange.getIn().getBody(InputStream.class); is.close(); exchange.getOut().setBody(null); } }); from("direct:uploadImageDataHandler").process(new Processor() { public void process(Exchange exchange) throws Exception { assertEquals("123", exchange.getIn().getHeader("id")); assertEquals("image/jpeg", exchange.getIn().getHeader("Content-Type")); assertTrue(DataHandler.class.isAssignableFrom(exchange.getIn().getBody().getClass())); DataHandler dh = exchange.getIn().getBody(DataHandler.class); assertEquals("image/jpeg", dh.getContentType()); dh.getInputStream().close(); exchange.getOut().setBody(null); } }); from("direct:multipartPostWithParametersAndPayload").process(new Processor() { public void process(Exchange exchange) throws Exception { assertEquals("abcd", exchange.getIn().getHeader("query")); assertEquals("123", exchange.getIn().getHeader("id")); assertNotNull(exchange.getIn().getAttachment("part1")); assertNotNull(exchange.getIn().getAttachment("part2")); assertNull(exchange.getIn().getHeader("part1")); assertNull(exchange.getIn().getHeader("part2")); assertEquals(Customer.class, exchange.getIn().getHeader("body").getClass()); exchange.getOut().setBody(null); } }); from("direct:multipartPostWithoutParameters").process(new Processor() { public void process(Exchange exchange) throws Exception { assertNotNull(exchange.getIn().getAttachment("part1")); assertNotNull(exchange.getIn().getAttachment("part2")); assertNull(exchange.getIn().getHeader("part1")); assertNull(exchange.getIn().getHeader("part2")); assertEquals(Customer.class, exchange.getIn().getHeader("body").getClass()); exchange.getOut().setBody(null); } }); } }; } @Test public void testGetCustomerOnlyHeaders() throws Exception { HttpGet get = new HttpGet("http://localhost:" + PORT_PATH + "/rest/customerservice/customers/123"); get.addHeader("Accept", "text/xml"); HttpResponse response = httpclient.execute(get); assertEquals(200, response.getStatusLine().getStatusCode()); Customer entity = (Customer) jaxb.createUnmarshaller().unmarshal(response.getEntity().getContent()); assertEquals(123, entity.getId()); } @Test public void testGetCustomerHttp404CustomStatus() throws Exception { HttpGet get = new HttpGet("http://localhost:" + PORT_PATH + "/rest/customerservice/customers/456"); get.addHeader("Accept", "text/xml"); HttpResponse response = httpclient.execute(get); assertEquals(404, response.getStatusLine().getStatusCode()); } @Test public void testUpdateCustomerBodyAndHeaders() throws Exception { HttpPut put = new HttpPut("http://localhost:" + PORT_PATH + "/rest/customerservice/customers/123"); StringWriter sw = new StringWriter(); jaxb.createMarshaller().marshal(new Customer(123, "Raul"), sw); put.setEntity(new StringEntity(sw.toString())); put.addHeader("Content-Type", "text/xml"); put.addHeader("Accept", "text/xml"); HttpResponse response = httpclient.execute(put); assertEquals(200, response.getStatusLine().getStatusCode()); } @Test public void testNewCustomerOnlyBody() throws Exception { HttpPost post = new HttpPost("http://localhost:" + PORT_PATH + "/rest/customerservice/customers"); StringWriter sw = new StringWriter(); jaxb.createMarshaller().marshal(new Customer(123, "Raul"), sw); post.setEntity(new StringEntity(sw.toString())); post.addHeader("Content-Type", "text/xml"); post.addHeader("Accept", "text/xml"); HttpResponse response = httpclient.execute(post); assertEquals(200, response.getStatusLine().getStatusCode()); } @Test public void testListVipCustomers() throws Exception { HttpGet get = new HttpGet("http://localhost:" + PORT_PATH + "/rest/customerservice/customers/vip/gold"); get.addHeader("Content-Type", "text/xml"); get.addHeader("Accept", "text/xml"); HttpResponse response = httpclient.execute(get); assertEquals(200, response.getStatusLine().getStatusCode()); CustomerList cl = (CustomerList) jaxb.createUnmarshaller() .unmarshal(new StringReader(EntityUtils.toString(response.getEntity()))); List<Customer> vips = cl.getCustomers(); assertEquals(2, vips.size()); assertEquals(123, vips.get(0).getId()); assertEquals(456, vips.get(1).getId()); } @Test public void testUpdateVipCustomer() throws Exception { HttpPut put = new HttpPut("http://localhost:" + PORT_PATH + "/rest/customerservice/customers/vip/gold/123"); StringWriter sw = new StringWriter(); jaxb.createMarshaller().marshal(new Customer(123, "Raul2"), sw); put.setEntity(new StringEntity(sw.toString())); put.addHeader("Content-Type", "text/xml"); put.addHeader("Accept", "text/xml"); HttpResponse response = httpclient.execute(put); assertEquals(200, response.getStatusLine().getStatusCode()); } @Test public void testDeleteVipCustomer() throws Exception { HttpDelete delete = new HttpDelete( "http://localhost:" + PORT_PATH + "/rest/customerservice/customers/vip/gold/123"); delete.addHeader("Accept", "text/xml"); HttpResponse response = httpclient.execute(delete); assertEquals(200, response.getStatusLine().getStatusCode()); } @Test public void testUploadInputStream() throws Exception { HttpPost post = new HttpPost( "http://localhost:" + PORT_PATH + "/rest/customerservice/customers/123/image_inputstream"); post.addHeader("Content-Type", "image/jpeg"); post.addHeader("Accept", "text/xml"); post.setEntity( new InputStreamEntity(this.getClass().getClassLoader().getResourceAsStream("java.jpg"), 100)); HttpResponse response = httpclient.execute(post); assertEquals(200, response.getStatusLine().getStatusCode()); } @Test public void testUploadDataHandler() throws Exception { HttpPost post = new HttpPost( "http://localhost:" + PORT_PATH + "/rest/customerservice/customers/123/image_datahandler"); post.addHeader("Content-Type", "image/jpeg"); post.addHeader("Accept", "text/xml"); post.setEntity( new InputStreamEntity(this.getClass().getClassLoader().getResourceAsStream("java.jpg"), 100)); HttpResponse response = httpclient.execute(post); assertEquals(200, response.getStatusLine().getStatusCode()); } @Test public void testMultipartPostWithParametersAndPayload() throws Exception { HttpPost post = new HttpPost( "http://localhost:" + PORT_PATH + "/rest/customerservice/customers/multipart/123?query=abcd"); MultipartEntity multipart = new MultipartEntity(HttpMultipartMode.STRICT); multipart.addPart("part1", new FileBody( new File(this.getClass().getClassLoader().getResource("java.jpg").toURI()), "java.jpg")); multipart.addPart("part2", new FileBody( new File(this.getClass().getClassLoader().getResource("java.jpg").toURI()), "java.jpg")); StringWriter sw = new StringWriter(); jaxb.createMarshaller().marshal(new Customer(123, "Raul"), sw); multipart.addPart("body", new StringBody(sw.toString(), "text/xml", Charset.forName("UTF-8"))); post.setEntity(multipart); HttpResponse response = httpclient.execute(post); assertEquals(200, response.getStatusLine().getStatusCode()); } @Test public void testMultipartPostWithoutParameters() throws Exception { HttpPost post = new HttpPost("http://localhost:" + PORT_PATH + "/rest/customerservice/customers/multipart"); MultipartEntity multipart = new MultipartEntity(HttpMultipartMode.STRICT); multipart.addPart("part1", new FileBody( new File(this.getClass().getClassLoader().getResource("java.jpg").toURI()), "java.jpg")); multipart.addPart("part2", new FileBody( new File(this.getClass().getClassLoader().getResource("java.jpg").toURI()), "java.jpg")); StringWriter sw = new StringWriter(); jaxb.createMarshaller().marshal(new Customer(123, "Raul"), sw); multipart.addPart("body", new StringBody(sw.toString(), "text/xml", Charset.forName("UTF-8"))); post.setEntity(multipart); HttpResponse response = httpclient.execute(post); assertEquals(200, response.getStatusLine().getStatusCode()); } }