/*
* 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.mtom;
import junit.framework.TestCase;
import org.apache.axis2.AxisFault;
import org.apache.axis2.Constants;
import org.apache.axis2.addressing.EndpointReference;
import org.apache.axis2.attachments.ByteArrayDataSource;
import org.apache.axis2.client.Call;
import org.apache.axis2.client.Options;
import org.apache.axis2.client.async.AsyncResult;
import org.apache.axis2.client.async.Callback;
import org.apache.axis2.description.AxisService;
import org.apache.axis2.engine.Echo;
import org.apache.axis2.integration.UtilServer;
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.om.OMText;
import org.apache.axis2.om.impl.llom.OMTextImpl;
import org.apache.axis2.soap.SOAP12Constants;
import org.apache.axis2.soap.SOAPEnvelope;
import org.apache.axis2.soap.SOAPFactory;
import org.apache.axis2.util.Utils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import javax.activation.DataHandler;
import javax.xml.namespace.QName;
public class EchoRawMTOMToBase64Test extends TestCase {
private EndpointReference targetEPR = new EndpointReference("http://127.0.0.1:"
+ (UtilServer.TESTING_PORT)
+ "/axis/services/EchoXMLService/echoMTOMtoBase64");
private Log log = LogFactory.getLog(getClass());
private QName serviceName = new QName("EchoXMLService");
private QName operationName = new QName("echoMTOMtoBase64");
private AxisService service;
OMText expectedTextData;
private boolean finish = false;
public EchoRawMTOMToBase64Test() {
super(EchoRawMTOMToBase64Test.class.getName());
}
public EchoRawMTOMToBase64Test(String testName) {
super(testName);
}
protected void setUp() throws Exception {
UtilServer.start();
service = Utils.createSimpleService(serviceName, Echo.class.getName(),
operationName);
UtilServer.deployService(service);
}
protected void tearDown() throws Exception {
UtilServer.unDeployService(serviceName);
UtilServer.stop();
UtilServer.unDeployClientService();
}
private OMElement createPayload() {
OMFactory fac = OMAbstractFactory.getOMFactory();
OMNamespace omNs = fac.createOMNamespace("http://localhost/my", "my");
OMElement rpcWrapEle = fac.createOMElement("echoMTOMtoBase64", omNs);
OMElement data = fac.createOMElement("data", omNs);
byte[] byteArray = new byte[]{13, 56, 65, 32, 12, 12, 7, -3, -2, -1,
98};
DataHandler dataHandler = new DataHandler(new ByteArrayDataSource(byteArray));
expectedTextData = new OMTextImpl(dataHandler, true);
data.addChild(expectedTextData);
rpcWrapEle.addChild(data);
return rpcWrapEle;
}
public void testEchoXMLASync() throws Exception {
OMElement payload = createPayload();
String clientHome = "target/test-resources/intregrationRepo";
Options clientOptions = new Options();
clientOptions.setTo(targetEPR);
clientOptions.setTransportInfo(Constants.TRANSPORT_HTTP, Constants.TRANSPORT_HTTP,
false);
Call call = new Call(clientHome);
call.setClientOptions(clientOptions);
Callback callback = new Callback() {
public void onComplete(AsyncResult result) {
SOAPEnvelope envelope = result.getResponseEnvelope();
OMElement data = (OMElement) envelope.getBody().getFirstElement().getFirstOMChild();
compareWithCreatedOMText(data.getText());
finish = true;
}
public void reportError(Exception e) {
log.info(e.getMessage());
finish = true;
}
};
call.invokeNonBlocking(operationName.getLocalPart(),
payload,
callback);
int index = 0;
while (!finish) {
Thread.sleep(1000);
index++;
if (index > 10) {
throw new AxisFault(
"Server was shutdown as the async response take too long to complete");
}
}
call.close();
}
public void testEchoXMLSync() throws Exception {
for (int i = 0; i < 10; i++) {
SOAPFactory fac = OMAbstractFactory.getSOAP11Factory();
OMElement payload = createPayload();
String clientHome = "target/test-resources/intregrationRepo";
Options clientOptions = new Options();
clientOptions.setTo(targetEPR);
clientOptions.setProperty(Constants.Configuration.ENABLE_MTOM, Constants.VALUE_TRUE);
clientOptions.setTransportInfo(Constants.TRANSPORT_HTTP,
Constants.TRANSPORT_HTTP, false);
clientOptions.setSoapVersionURI(SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI);
Call call =
new Call(clientHome);
call.setClientOptions(clientOptions);
OMElement result = call.invokeBlocking(operationName
.getLocalPart(), payload);
OMElement data = (OMElement) result.getFirstOMChild();
compareWithCreatedOMText(data.getText());
call.close();
log.info("" + i);
UtilServer.unDeployClientService();
}
}
private void compareWithCreatedOMText(String actualText) {
String originalTextValue = expectedTextData.getText();
TestCase.assertEquals(actualText, originalTextValue);
}
}