| /** |
| * 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; |
| |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| |
| import javax.servlet.ServletRequest; |
| |
| import org.apache.camel.Exchange; |
| import org.apache.camel.Message; |
| import org.apache.camel.Processor; |
| import org.apache.camel.builder.RouteBuilder; |
| import org.apache.camel.component.cxf.common.message.CxfConstants; |
| import org.apache.camel.test.junit4.CamelTestSupport; |
| import org.apache.cxf.BusFactory; |
| import org.apache.cxf.endpoint.Client; |
| import org.apache.cxf.frontend.ClientFactoryBean; |
| import org.apache.cxf.frontend.ClientProxyFactoryBean; |
| import org.junit.Test; |
| |
| public class CxfConsumerResponseTest extends CamelTestSupport { |
| |
| private static final String ECHO_OPERATION = "echo"; |
| private static final String ECHO_BOOLEAN_OPERATION = "echoBoolean"; |
| private static final String PING_OPERATION = "ping"; |
| private static final String TEST_MESSAGE = "Hello World!"; |
| private static int pingCounter; |
| |
| protected final String simpleEndpointAddress = "http://localhost:" |
| + CXFTestSupport.getPort1() + "/" + getClass().getSimpleName() + "/test"; |
| |
| protected final String simpleEndpointURI = "cxf://" + simpleEndpointAddress |
| + "?serviceClass=org.apache.camel.component.cxf.HelloService" |
| + "&publishedEndpointUrl=http://www.simple.com/services/test"; |
| |
| |
| @Override |
| public boolean isCreateCamelContextPerClass() { |
| return true; |
| } |
| |
| // START SNIPPET: example |
| protected RouteBuilder createRouteBuilder() { |
| return new RouteBuilder() { |
| public void configure() { |
| from(simpleEndpointURI).inOnly("log:test") |
| .choice().when(header(CxfConstants.OPERATION_NAME).isEqualTo(ECHO_OPERATION)).process(new Processor() { |
| public void process(final Exchange exchange) { |
| assertEquals(DataFormat.POJO, exchange.getProperty(CxfConstants.DATA_FORMAT_PROPERTY, DataFormat.class)); |
| Message in = exchange.getIn(); |
| // check the remote IP from the cxfMessage |
| org.apache.cxf.message.Message cxfMessage = in.getHeader(CxfConstants.CAMEL_CXF_MESSAGE, org.apache.cxf.message.Message.class); |
| assertNotNull("Should get the cxfMessage instance from message header", cxfMessage); |
| ServletRequest request = (ServletRequest)cxfMessage.get("HTTP.REQUEST"); |
| assertNotNull("Should get the ServletRequest", request); |
| assertNotNull("Should get the RemoteAddress" + request.getRemoteAddr()); |
| // Get the parameter list |
| List<?> parameter = in.getBody(List.class); |
| // Get the operation name |
| String operation = (String)in.getHeader(CxfConstants.OPERATION_NAME); |
| Object result = operation + " " + (String)parameter.get(0); |
| // Put the result back |
| exchange.getIn().setBody(result); |
| // set up the response context which force start document |
| Map<String, Object> map = new HashMap<String, Object>(); |
| map.put("org.apache.cxf.stax.force-start-document", Boolean.TRUE); |
| exchange.getIn().setHeader(Client.RESPONSE_CONTEXT, map); |
| } |
| }) |
| .when(header(CxfConstants.OPERATION_NAME).isEqualTo(ECHO_BOOLEAN_OPERATION)).process(new Processor() { |
| public void process(final Exchange exchange) { |
| Message in = exchange.getIn(); |
| // Get the parameter list |
| List<?> parameter = in.getBody(List.class); |
| // Put the result back |
| exchange.getOut().setBody(parameter.get(0)); |
| } |
| }) |
| .when(header(CxfConstants.OPERATION_NAME).isEqualTo(PING_OPERATION)).process(new Processor() { |
| public void process(final Exchange exchange) { |
| pingCounter++; |
| } |
| |
| }); |
| |
| } |
| }; |
| } |
| // END SNIPPET: example |
| |
| @Test |
| public void testInvokingServiceFromCXFClient() throws Exception { |
| ClientProxyFactoryBean proxyFactory = new ClientProxyFactoryBean(); |
| ClientFactoryBean clientBean = proxyFactory.getClientFactoryBean(); |
| clientBean.setAddress(simpleEndpointAddress); |
| clientBean.setServiceClass(HelloService.class); |
| clientBean.setBus(BusFactory.getDefaultBus()); |
| |
| HelloService client = (HelloService) proxyFactory.create(); |
| assertNotNull(client); |
| |
| String result = client.echo(TEST_MESSAGE); |
| assertEquals("We should get the echo string result from router", result, "echo " + TEST_MESSAGE); |
| |
| Boolean bool = client.echoBoolean(Boolean.TRUE); |
| assertNotNull("The result should not be null", bool); |
| assertEquals("We should get the echo boolean result from router ", bool.toString(), "true"); |
| |
| int beforeCallingPing = pingCounter; |
| client.ping(); |
| int afterCallingPing = pingCounter; |
| assertTrue("The ping operation doesn't be called", afterCallingPing - beforeCallingPing == 1); |
| } |
| |
| |
| } |