/**
* 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.cxf.systest.aegis.mtom;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
import javax.activation.DataHandler;
import javax.xml.namespace.QName;
import org.w3c.dom.Attr;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.apache.cxf.Bus;
import org.apache.cxf.aegis.databinding.AegisDatabinding;
import org.apache.cxf.aegis.type.mtom.AbstractXOPType;
import org.apache.cxf.endpoint.Server;
import org.apache.cxf.frontend.ClientProxyFactoryBean;
import org.apache.cxf.helpers.IOUtils;
import org.apache.cxf.jaxws.JaxWsProxyFactoryBean;
import org.apache.cxf.systest.aegis.mtom.fortest.DataHandlerBean;
import org.apache.cxf.systest.aegis.mtom.fortest.MtomTestImpl;
import org.apache.cxf.systest.aegis.mtom.fortest.MtomTestService;
import org.apache.cxf.test.TestUtilities;
import org.apache.cxf.testutil.common.TestUtil;
import org.apache.ws.commons.schema.constants.Constants;
import org.junit.Assert;
import org.junit.Test;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests;
/**
*
*/
@ContextConfiguration(locations = { "classpath:mtomTestBeans.xml" })
public class MtomTest extends AbstractJUnit4SpringContextTests {
static final String PORT = TestUtil.getPortNumber(MtomTest.class);
private MtomTestImpl impl;
private MtomTestService client;
private MtomTestService jaxwsClient;
private TestUtilities testUtilities;
public MtomTest() {
testUtilities = new TestUtilities(getClass());
}
private void setupForTest(boolean enableClientMTOM) throws Exception {
AegisDatabinding aegisBinding = new AegisDatabinding();
aegisBinding.setMtomEnabled(enableClientMTOM);
ClientProxyFactoryBean proxyFac = new ClientProxyFactoryBean();
proxyFac.setDataBinding(aegisBinding);
proxyFac.setAddress("http://localhost:" + PORT + "/mtom");
JaxWsProxyFactoryBean jaxwsFac = new JaxWsProxyFactoryBean();
jaxwsFac.setDataBinding(new AegisDatabinding());
jaxwsFac.setAddress("http://localhost:" + PORT + "/jaxWsMtom");
Map<String, Object> props = new HashMap<>();
if (enableClientMTOM) {
props.put("mtom-enabled", Boolean.TRUE);
}
proxyFac.setProperties(props);
client = proxyFac.create(MtomTestService.class);
jaxwsClient = jaxwsFac.create(MtomTestService.class);
impl = (MtomTestImpl)applicationContext.getBean("mtomImpl");
}
@Test
public void testMtomReply() throws Exception {
setupForTest(true);
DataHandlerBean dhBean = client.produceDataHandlerBean();
Assert.assertNotNull(dhBean);
String result = IOUtils.toString(dhBean.getDataHandler().getInputStream(), "utf-8");
Assert.assertEquals(MtomTestImpl.STRING_DATA, result);
}
//TODO: how do we see if MTOM actually happened?
@Test
public void testJaxWsMtomReply() throws Exception {
setupForTest(true);
DataHandlerBean dhBean = jaxwsClient.produceDataHandlerBean();
Assert.assertNotNull(dhBean);
String result = IOUtils.toString(dhBean.getDataHandler().getInputStream(), "utf-8");
Assert.assertEquals(MtomTestImpl.STRING_DATA, result);
}
@Test
public void testAcceptDataHandler() throws Exception {
setupForTest(true);
DataHandlerBean dhBean = new DataHandlerBean();
dhBean.setName("some name");
// some day, we might need this to be higher than some threshold.
String someData = "This is the cereal shot from guns.";
DataHandler dataHandler = new DataHandler(someData, "text/plain;charset=utf-8");
dhBean.setDataHandler(dataHandler);
client.acceptDataHandler(dhBean);
DataHandlerBean accepted = impl.getLastDhBean();
Assert.assertNotNull(accepted);
Object o = accepted.getDataHandler().getContent();
String data = null;
if (o instanceof String) {
data = (String)o;
} else if (o instanceof InputStream) {
data = IOUtils.toString((InputStream)o);
}
Assert.assertNotNull(data);
Assert.assertEquals("This is the cereal shot from guns.", data);
}
@Test
public void testAcceptDataHandlerNoMTOM() throws Exception {
setupForTest(false);
DataHandlerBean dhBean = new DataHandlerBean();
dhBean.setName("some name");
// some day, we might need this to be longer than some threshold.
String someData = "This is the cereal shot from guns.";
DataHandler dataHandler = new DataHandler(someData, "text/plain;charset=utf-8");
dhBean.setDataHandler(dataHandler);
client.acceptDataHandler(dhBean);
DataHandlerBean accepted = impl.getLastDhBean();
Assert.assertNotNull(accepted);
InputStream data = accepted.getDataHandler().getInputStream();
Assert.assertNotNull(data);
String dataString = org.apache.commons.io.IOUtils.toString(data, "utf-8");
Assert.assertEquals("This is the cereal shot from guns.", dataString);
}
@Test
public void testMtomSchema() throws Exception {
testUtilities.setBus((Bus)applicationContext.getBean("cxf"));
testUtilities.addDefaultNamespaces();
testUtilities.addNamespace("xmime", "http://www.w3.org/2005/05/xmlmime");
Server s = testUtilities.
getServerForService(new QName("http://fortest.mtom.aegis.systest.cxf.apache.org/",
"MtomTestService"));
Document wsdl = testUtilities.getWSDLDocument(s);
Assert.assertNotNull(wsdl);
NodeList typeAttrList =
testUtilities.assertValid("//xsd:complexType[@name='inputDhBean']/xsd:sequence/"
+ "xsd:element[@name='dataHandler']/"
+ "@type",
wsdl);
Attr typeAttr = (Attr)typeAttrList.item(0);
String typeAttrValue = typeAttr.getValue();
// now, this thing is a qname with a :, and we have to work out if it's correct.
String[] pieces = typeAttrValue.split(":");
Assert.assertEquals("base64Binary", pieces[1]);
Node elementNode = typeAttr.getOwnerElement();
String url = testUtilities.resolveNamespacePrefix(pieces[0], elementNode);
Assert.assertEquals(Constants.URI_2001_SCHEMA_XSD, url);
s = testUtilities.getServerForAddress("http://localhost:" + PORT + "/mtomXmime");
wsdl = testUtilities.getWSDLDocument(s);
Assert.assertNotNull(wsdl);
typeAttrList =
testUtilities.assertValid("//xsd:complexType[@name='inputDhBean']/xsd:sequence/"
+ "xsd:element[@name='dataHandler']/"
+ "@type",
wsdl);
typeAttr = (Attr)typeAttrList.item(0);
typeAttrValue = typeAttr.getValue();
// now, this thing is a qname with a :, and we have to work out if it's correct.
pieces = typeAttrValue.split(":");
Assert.assertEquals("base64Binary", pieces[1]);
elementNode = typeAttr.getOwnerElement();
url = testUtilities.resolveNamespacePrefix(pieces[0], elementNode);
Assert.assertEquals(AbstractXOPType.XML_MIME_NS, url);
/* when I add a test for a custom mapping.
testUtilities.assertValid("//xsd:complexType[@name='inputDhBean']/xsd:sequence/"
+ "xsd:element[@name='dataHandler']/"
+ "@xmime:expectedContentType/text()",
wsdl);
*/
}
}