/*
* 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.mail;
import junit.framework.TestCase;
import org.apache.axis2.AxisFault;
import org.apache.axis2.Constants;
import org.apache.axis2.addressing.EndpointReference;
import org.apache.axis2.clientapi.AsyncResult;
import org.apache.axis2.clientapi.Callback;
import org.apache.axis2.context.ConfigurationContext;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.context.ServiceContext;
import org.apache.axis2.description.OperationDescription;
import org.apache.axis2.description.ServiceDescription;
import org.apache.axis2.engine.AxisConfiguration;
import org.apache.axis2.engine.Echo;
import org.apache.axis2.engine.MessageReceiver;
import org.apache.axis2.om.OMAbstractFactory;
import org.apache.axis2.om.OMElement;
import org.apache.axis2.om.OMFactory;
import org.apache.axis2.om.OMNamespace;
import org.apache.axis2.soap.SOAPEnvelope;
import org.apache.axis2.util.Utils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLOutputFactory;
import javax.xml.stream.XMLStreamException;
/**
* This will use the In out functionality to test the mailet functionality. This code was
* based on the code on the MailRequestResponceRawXMLTest.java.
*/
public class MailetRequestResponceRawXMLTest extends TestCase {
private EndpointReference targetEPR =
new EndpointReference("axis2-server@127.0.0.1" +
"/axis/services/EchoXMLService/echoOMElement");
private Log log = LogFactory.getLog(getClass());
private QName serviceName = new QName("EchoXMLService");
private QName operationName = new QName("echoOMElement");
private QName transportName = new QName("http://localhost/my",
"NullTransport");
private AxisConfiguration engineRegistry;
private MessageContext mc;
private SOAPEnvelope envelope;
private boolean finish = false;
public MailetRequestResponceRawXMLTest() {
super(MailetRequestResponceRawXMLTest.class.getName());
}
public MailetRequestResponceRawXMLTest(String testName) {
super(testName);
}
protected void setUp() throws Exception {
ConfigurationContext configContext = UtilsMailServer.start();
// configContext.getAxisConfiguration().engageModule(
// new QName(Constants.MODULE_ADDRESSING));
ServiceDescription service =
Utils.createSimpleService(serviceName,
Echo.class.getName(),
operationName);
configContext.getAxisConfiguration().addService(service);
Utils.resolvePhases(configContext.getAxisConfiguration(), service);
}
protected void tearDown() throws Exception {
UtilsMailServer.stop();
}
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.createText(value, "Isaac Assimov, the foundation Sega"));
method.addChild(value);
return method;
}
public void testEchoXMLCompleteASync() throws Exception {
ConfigurationContext configContext = UtilsMailServer.createClientConfigurationContext();
ServiceDescription service = new ServiceDescription(serviceName);
OperationDescription operation = new OperationDescription(
operationName);
operation.setMessageReceiver(new MessageReceiver() {
public void receive(MessageContext messgeCtx) throws AxisFault {
envelope = messgeCtx.getEnvelope();
}
});
service.addOperation(operation);
configContext.getAxisConfiguration().addService(service);
Utils.resolvePhases(configContext.getAxisConfiguration(), service);
ServiceContext serviceContext = Utils.fillContextInformation(operation, service, configContext);
org.apache.axis2.clientapi.Call call = new org.apache.axis2.clientapi.Call(
serviceContext);
// call.engageModule(new QName(Constants.MODULE_ADDRESSING));
call.setTo(targetEPR);
call.setTransportInfo(Constants.TRANSPORT_MAIL,
Constants.TRANSPORT_MAIL,
true);
Callback callback = new Callback() {
public void onComplete(AsyncResult result) {
try {
result.getResponseEnvelope().serialize(
XMLOutputFactory.newInstance()
.createXMLStreamWriter(System.out));
} catch (XMLStreamException e) {
reportError(e);
} finally {
finish = true;
}
}
public void reportError(Exception e) {
log.info(e.getMessage());
finish = true;
}
};
call.invokeNonBlocking(operationName.getLocalPart(),
createEnvelope(),
callback);
int index = 0;
while (!finish) {
Thread.sleep(1000);
index++;
if (index > 10) {
throw new AxisFault(
"Async response is taking too long[10s+]. Server is being shut down.");
}
}
call.close();
}
}