| /* |
| * 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.axis2.addressing; |
| |
| import junit.framework.Test; |
| import junit.framework.TestSuite; |
| import org.apache.axiom.om.OMAbstractFactory; |
| import org.apache.axiom.om.OMElement; |
| import org.apache.axiom.om.OMFactory; |
| import org.apache.axiom.om.OMNamespace; |
| import org.apache.axis2.AxisFault; |
| import org.apache.axis2.Constants; |
| import org.apache.axis2.client.Options; |
| import org.apache.axis2.client.ServiceClient; |
| import org.apache.axis2.context.ConfigurationContext; |
| import org.apache.axis2.context.MessageContext; |
| import org.apache.axis2.context.ServiceContext; |
| import org.apache.axis2.description.AxisOperation; |
| import org.apache.axis2.description.AxisService; |
| import org.apache.axis2.description.InOutAxisOperation; |
| import org.apache.axis2.description.OutInAxisOperation; |
| import org.apache.axis2.description.Parameter; |
| import org.apache.axis2.engine.AxisConfiguration; |
| import org.apache.axis2.engine.util.TestConstants; |
| import org.apache.axis2.integration.UtilServer; |
| import org.apache.axis2.integration.UtilServerBasedTestCase; |
| import org.apache.axis2.receivers.RawXMLINOutMessageReceiver; |
| |
| import javax.xml.namespace.QName; |
| |
| /** |
| * This tests that it is possible to use the more advanced function allowed by the WS-Addressing |
| * specification to redirect responses. This test sets up 2 services, MultiHopRedirectService1 and |
| * MultiHopRedirectService2. MultiHopRedirectService1 echos the XML it receives to |
| * MultiHopRedirectService2, adding the original ReplyTo. MultiHopRedirectService2 then echos the |
| * XML to the orignal ReplyTo, setting the RelatesTo to the messageID of the original request |
| * message to allow the client to correlate the message. |
| * <p/> |
| * The clearly only works if the client is using a separate listener (which the client in the test |
| * does) |
| */ |
| public class MultiHopRedirectServiceTest extends UtilServerBasedTestCase implements TestConstants { |
| |
| EndpointReference targetEPR = new EndpointReference( |
| "http://127.0.0.1:" + (UtilServer.TESTING_PORT) + |
| "/axis2/services/MultiHopRedirectService1/echoRedirect"); |
| |
| protected AxisConfiguration engineRegistry; |
| protected MessageContext mc; |
| protected ServiceContext serviceContext; |
| protected AxisService mhs1; |
| protected AxisService mhs2; |
| |
| public static Test suite() { |
| return getTestSetup(new TestSuite(MultiHopRedirectServiceTest.class)); |
| } |
| |
| protected void setUp() throws Exception { |
| mhs1 = createMultiHopRedirectService1(); |
| mhs2 = createMultiHopRedirectService2(); |
| UtilServer.deployService(mhs1); |
| UtilServer.deployService(mhs2); |
| } |
| |
| AxisService createMultiHopRedirectService1() throws AxisFault { |
| AxisService service = new AxisService("MultiHopRedirectService1"); |
| |
| service.setClassLoader(Thread.currentThread().getContextClassLoader()); |
| service.addParameter( |
| new Parameter(Constants.SERVICE_CLASS, MultiHopRedirectService1.class.getName())); |
| |
| AxisOperation axisOp = new InOutAxisOperation(new QName("echoRedirect")); |
| |
| axisOp.setMessageReceiver(new RawXMLINOutMessageReceiver()); |
| service.addOperation(axisOp); |
| service.mapActionToOperation(Constants.AXIS2_NAMESPACE_URI + "/" + "echoRedirect", axisOp); |
| |
| return service; |
| } |
| |
| AxisService createMultiHopRedirectService2() throws AxisFault { |
| AxisService service = new AxisService("MultiHopRedirectService2"); |
| |
| service.setClassLoader(Thread.currentThread().getContextClassLoader()); |
| service.addParameter( |
| new Parameter(Constants.SERVICE_CLASS, MultiHopRedirectService2.class.getName())); |
| |
| AxisOperation axisOp = new InOutAxisOperation(new QName("echoRedirect")); |
| |
| axisOp.setMessageReceiver(new RawXMLINOutMessageReceiver()); |
| service.addOperation(axisOp); |
| service.mapActionToOperation(Constants.AXIS2_NAMESPACE_URI + "/" + "echoRedirect", axisOp); |
| |
| return service; |
| } |
| |
| protected void tearDown() throws Exception { |
| UtilServer.unDeployService(new QName("MultiHopRedirectService1")); |
| UtilServer.unDeployService(new QName("MultiHopRedirectService2")); |
| UtilServer.unDeployClientService(); |
| } |
| |
| public void testMultiHop() throws Exception { |
| OMElement method = createElement("this message should not cause a fault."); |
| ServiceClient sender = null; |
| |
| try { |
| sender = createServiceClient(); |
| OMElement response = sender.sendReceive(new QName("echoRedirect"), method); |
| assertEquals("this message should not cause a fault.", response.getText()); |
| } finally { |
| if (sender != null) |
| sender.cleanup(); |
| } |
| } |
| |
| private OMElement createElement(String text) { |
| OMFactory fac = OMAbstractFactory.getOMFactory(); |
| |
| OMNamespace omNs = fac.createOMNamespace(Constants.AXIS2_NAMESPACE_URI + "/addressing", |
| "axis2addressing"); |
| OMElement value = fac.createOMElement("element", omNs); |
| value.setText(text); |
| |
| return value; |
| } |
| |
| private ServiceClient createServiceClient() throws AxisFault { |
| AxisService service = new AxisService("MultiHopRedirectService1"); |
| |
| service.setClassLoader(Thread.currentThread().getContextClassLoader()); |
| service.addParameter( |
| new Parameter(Constants.SERVICE_CLASS, MultiHopRedirectService1.class.getName())); |
| |
| AxisOperation axisOp = new OutInAxisOperation(new QName("echoRedirect")); |
| |
| axisOp.setMessageReceiver(new RawXMLINOutMessageReceiver()); |
| axisOp.setSoapAction(Constants.AXIS2_NAMESPACE_URI + "/" + "echoRedirect"); |
| axisOp.setOutputAction(Constants.AXIS2_NAMESPACE_URI + "/" + "echoRedirect"); |
| service.addOperation(axisOp); |
| service.mapActionToOperation(Constants.AXIS2_NAMESPACE_URI + "/" + "echoRedirect", axisOp); |
| |
| ConfigurationContext configcontext = UtilServer.createClientConfigurationContext(); |
| |
| ServiceClient sender; |
| Options options = new Options(); |
| options.setTo(targetEPR); |
| options.setTransportInProtocol(Constants.TRANSPORT_HTTP); |
| options.setUseSeparateListener(true); |
| options.setTimeOutInMilliSeconds(5000); |
| sender = new ServiceClient(configcontext, service); |
| sender.setOptions(options); |
| sender.engageModule("addressing"); |
| return sender; |
| } |
| } |