rampart_1_1/modules/integration/test/org/apache/axis2/engine/MessageContextInjectionTest.java

/*
 * Copyright 2004,2005 The Apache Software Foundation.
 *
 * Licensed 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.engine;

import junit.framework.TestCase;
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.axiom.soap.SOAP11Constants;
import org.apache.axis2.AxisFault;
import org.apache.axis2.Constants;
import org.apache.axis2.addressing.EndpointReference;
import org.apache.axis2.client.Options;
import org.apache.axis2.client.ServiceClient;
import org.apache.axis2.context.ConfigurationContext;
import org.apache.axis2.context.ConfigurationContextFactory;
import org.apache.axis2.description.*;
import org.apache.axis2.engine.util.TestConstants;
import org.apache.axis2.receivers.RawXMLINOnlyMessageReceiver;
import org.apache.axis2.receivers.RawXMLINOutMessageReceiver;
import org.apache.axis2.transport.TransportListener;
import org.apache.axis2.transport.local.LocalTransportReceiver;
import org.apache.axis2.transport.local.LocalTransportSender;

import javax.xml.namespace.QName;

public class MessageContextInjectionTest extends TestCase implements TestConstants {
    private TransportOutDescription tOut;

    public MessageContextInjectionTest() {
        super(MessageContextInjectionTest.class.getName());
    }

    public MessageContextInjectionTest(String testName) {
        super(testName);
    }

    protected void setUp() throws Exception {
        AxisConfiguration config = new AxisConfiguration();

        config.addMessageReceiver(
                "http://www.w3.org/2004/08/wsdl/in-only", new RawXMLINOnlyMessageReceiver());
        config.addMessageReceiver(
                "http://www.w3.org/2004/08/wsdl/in-out", new RawXMLINOutMessageReceiver());

        DispatchPhase dispatchPhase = new DispatchPhase();

        dispatchPhase.setName("Dispatch");

        AddressingBasedDispatcher abd = new AddressingBasedDispatcher();

        abd.initDispatcher();

        RequestURIBasedDispatcher rud = new RequestURIBasedDispatcher();

        rud.initDispatcher();

        SOAPActionBasedDispatcher sabd = new SOAPActionBasedDispatcher();

        sabd.initDispatcher();

        SOAPMessageBodyBasedDispatcher smbd = new SOAPMessageBodyBasedDispatcher();

        smbd.initDispatcher();

        InstanceDispatcher id = new InstanceDispatcher();

        id.init(new HandlerDescription("InstanceDispatcher"));
        dispatchPhase.addHandler(abd);
        dispatchPhase.addHandler(rud);
        dispatchPhase.addHandler(sabd);
        dispatchPhase.addHandler(smbd);
        dispatchPhase.addHandler(id);
        config.getGlobalInFlow().add(dispatchPhase);
        TransportInDescription tIn = new TransportInDescription(new QName(Constants.TRANSPORT_LOCAL));
        tIn.setReceiver(new TransportListener() {

            public void init(ConfigurationContext axisConf, TransportInDescription transprtIn) {
                //To change body of implemented methods use File | Settings | File Templates.
            }

            public void start() {
                //To change body of implemented methods use File | Settings | File Templates.
            }

            public void stop() {
                //To change body of implemented methods use File | Settings | File Templates.
            }

            public EndpointReference getEPRForService(String serviceName, String ip) throws AxisFault {
                return null;  //To change body of implemented methods use File | Settings | File Templates.
            }

            public EndpointReference[] getEPRsForService(String serviceName, String ip){
                return null;
            }
        });
        config.addTransportIn(tIn);

        tOut = new TransportOutDescription(new QName(Constants.TRANSPORT_LOCAL));
        tOut.setSender(new LocalTransportSender());
        config.addTransportOut(tOut);

        LocalTransportReceiver.CONFIG_CONTEXT = new ConfigurationContext(
                config);

        AxisService service = new AxisService(serviceName.getLocalPart());
        service.addParameter(
                new Parameter(Constants.SERVICE_CLASS,
                        MessageContextEnabledEcho.class.getName()));
        AxisOperation axisOperation = new InOnlyAxisOperation(
                operationName);
        axisOperation.setMessageReceiver(new RawXMLINOnlyMessageReceiver());
        service.addOperation(axisOperation);
        service.setClassLoader(Thread.currentThread().getContextClassLoader());
        LocalTransportReceiver.CONFIG_CONTEXT.getAxisConfiguration()
                .addService(service);
    }

    protected void tearDown() throws Exception {
    }

    private OMElement createEnvelope() {
        OMFactory fac = OMAbstractFactory.getOMFactory();
        OMNamespace omNs = fac.createOMNamespace("http://localhost/my", "my");
        OMElement method = fac.createOMElement("echoOMElement", omNs);
        OMElement value = fac.createOMElement("myValue", omNs);
        value.addChild(
                fac.createOMText(value, "Isaac Asimov, The Foundation Trilogy"));
        method.addChild(value);

        return method;
    }

    public void testEchoXMLSync() throws Exception {
        OMElement payload = createEnvelope();

        ConfigurationContext configContext =
                ConfigurationContextFactory.createConfigurationContextFromFileSystem("target/test-resources/integrationRepo", null);
        ServiceClient sender = new ServiceClient(configContext, null);
        Options options = new Options();
        sender.setOptions(options);
        options.setTo(targetEPR);
        options.setTransportOut(tOut);
        options.setSoapVersionURI(SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI);
        sender.fireAndForget(payload);

    }

}