modules/jaxws/test/org/apache/axis2/jaxws/message/SAAJConverterTests.java - axis-axis2-java-core - Git at Google

 /*
  * 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.jaxws.message;

 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.impl.builder.StAXSOAPModelBuilder;
 import org.apache.axis2.jaxws.message.factory.SAAJConverterFactory;
 import org.apache.axis2.jaxws.message.util.SAAJConverter;
 import org.apache.axis2.jaxws.registry.FactoryRegistry;
 import org.w3c.dom.Attr;
 import org.w3c.dom.Element;
 import org.w3c.dom.NamedNodeMap;
 import org.w3c.dom.TypeInfo;

 import javax.xml.soap.MessageFactory;
 import javax.xml.soap.SOAPBody;
 import javax.xml.soap.SOAPBodyElement;
 import javax.xml.soap.SOAPElement;
 import javax.xml.soap.SOAPEnvelope;
 import javax.xml.soap.SOAPFactory;
 import javax.xml.soap.SOAPMessage;
 import javax.xml.stream.XMLInputFactory;
 import javax.xml.stream.XMLStreamReader;
 import java.io.StringReader;
 import java.util.ArrayList;
 import java.util.List;

 /**
  * SAAJConverterTests
  *
  * Test the basic functionality of the SAAJConverter.
  * You can also use these tests to as sample code on how to use
  * the converter.
  *
  */
 public class SAAJConverterTests extends TestCase {

 	private static final String sampleText =
 		"<pre:a xmlns:pre=\"urn://sample\">" +
 		"<b>Hello</b>" +
 		"<c>World</c>" +
 		"</pre:a>";

 	private static final String sampleEnvelope =
 		"<soapenv:Envelope xmlns:soapenv=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
 		"<soapenv:Header /><soapenv:Body>" +
 		sampleText +
 		"</soapenv:Body></soapenv:Envelope>";

 	private static final String sampleText1 =
 		"<pre:a xmlns:pre=\"urn://sample\">" +
 		"<b id=\"100\">Hello</b>" +
 		"<c>World</c>" +
 		"</pre:a>";

 	private static final String sampleEnvelope1 =
 		"<soapenv:Envelope xmlns:soapenv=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
 		"<soapenv:Header /><soapenv:Body>" +
 		sampleText1 +
 		"</soapenv:Body></soapenv:Envelope>";

 	private static XMLInputFactory inputFactory = XMLInputFactory.newInstance();

 	public SAAJConverterTests() {
 		super();
 	}

 	public SAAJConverterTests(String arg0) {
 		super(arg0);
 	}

 	/**
 	 * @testStrategy Tests conversions between SAAJ and OM SOAPEnvelopes
 	 */
 	public void test1() throws Exception {

 		// Bootstrap: Create an OM SOAPEnvelope from the sample text
 		StringReader sr = new StringReader(sampleEnvelope);
 		XMLStreamReader inflow = inputFactory.createXMLStreamReader(sr);
 		StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(inflow, null);
 		org.apache.axiom.soap.SOAPEnvelope omEnvelope = builder.getSOAPEnvelope();

 		// Step 1: Get the SAAJConverter object from the Factory
 		SAAJConverterFactory f = (SAAJConverterFactory)
 			FactoryRegistry.getFactory(SAAJConverterFactory.class);
 		SAAJConverter converter = f.getSAAJConverter();

 		// Step 2: Convert the OM SOAPEnvelope to an SAAJ SOAPEnvelope
 		SOAPEnvelope saajEnvelope = converter.toSAAJ(omEnvelope, true).getSOAPPart().getEnvelope();

 		// Step 2a: Simple assertion check to ensure correctness.
 		String name = saajEnvelope.getBody().getFirstChild().getLocalName();
 		assertTrue("a".equals(name));

 		// Step 3: Convert the SAAJ SOAPEnvelope to an OM SOAPEnvelope
 		omEnvelope = converter.toOM(saajEnvelope);

 		// Step 3a: Simple assertion check to ensure correctness
 		name = omEnvelope.getBody().getFirstElement().getLocalName();
 		assertTrue("a".equals(name));

 		// Step 4: Rinse and repeat
 		saajEnvelope = converter.toSAAJ(omEnvelope, true).getSOAPPart().getEnvelope();
 		name = saajEnvelope.getBody().getFirstChild().getLocalName();
 		assertTrue("a".equals(name));
 		omEnvelope = converter.toOM(saajEnvelope);
 		name = omEnvelope.getBody().getFirstElement().getLocalName();
 		assertTrue("a".equals(name));
 	}

 	/**
 	 * @testStrategy Tests conversions between SAAJ and OM for normal element
 	 */
 	public void test2() throws Exception {

 		// Bootstrap: Create an OM SOAPEnvelope from the sample text.
 		StringReader sr = new StringReader(sampleEnvelope);
 		XMLStreamReader inflow = inputFactory.createXMLStreamReader(sr);
 		StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(inflow, null);
 		org.apache.axiom.soap.SOAPEnvelope omEnvelope = builder.getSOAPEnvelope();

 		// Bootstrap: Get an OMElement from the body
 		OMElement om = omEnvelope.getBody().getFirstElement();

 		// Bootstrap: Get an SAAJ Body to hold the target SOAPElement
 		MessageFactory msgFactory = MessageFactory.newInstance();
 		SOAPMessage message = msgFactory.createMessage();
 		SOAPBody body = message.getSOAPBody();

 		// Step 1: Get the SAAJConverter object from the Factory
 		SAAJConverterFactory f = (SAAJConverterFactory)
 			FactoryRegistry.getFactory(SAAJConverterFactory.class);
 		SAAJConverter converter = f.getSAAJConverter();

 		// Step 2: Convert OM to SAAJ
 		SOAPElement se = converter.toSAAJ(om, body);

 		// Step 2a: Verify
 		assertTrue(se instanceof SOAPBodyElement);
 		assertTrue(se.getLocalName().equals("a"));

 		// Step 3: Convert SAAJ to OM
 		om = converter.toOM(se);

 		// Step 3a: Verify
 		assertTrue(om.getLocalName().equals("a"));

 		// Step 4: Rinse and Repeat
 		se = converter.toSAAJ(om, body);
 		assertTrue(se instanceof SOAPBodyElement);
 		assertTrue(se.getLocalName().equals("a"));
 		om = converter.toOM(se);
 		assertTrue(om.getLocalName().equals("a"));
 	}

 	/**
 	 * @testStrategy: Create an OMElement, without using a builder.  Verification of AXIS2-970
 	 */
     public void test3() throws Exception {

 //    	 Step 1: Get the SAAJConverter object from the Factory
 		SAAJConverterFactory f = (SAAJConverterFactory)
 			FactoryRegistry.getFactory(SAAJConverterFactory.class);
 		SAAJConverter converter = f.getSAAJConverter();

 		// Stept 2: Create OM and parent SOAPElement
         OMFactory fac = OMAbstractFactory.getOMFactory();
         OMNamespace wrapNs = fac.createOMNamespace("namespace", "prefix");
         OMElement ome = fac.createOMElement("localname", wrapNs);
         SOAPFactory sf = SOAPFactory.newInstance();
         SOAPElement se = sf.createElement("name");

         // Step 3: Do the conversion
         converter.toSAAJ(ome, se, sf);
     }

 	/**
 	 * @testStrategy Tests conversions between OM and SAAJ SOAPEnvelopes
 	 * and verify attribute type is correctly stored
 	 */
 	public void test4() throws Exception {
 		// Bootstrap: Create an OM SOAPEnvelope from the sample text
 		StringReader sr = new StringReader(sampleEnvelope1);
 		XMLStreamReader inflow = inputFactory.createXMLStreamReader(sr);
 		StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(inflow, null);
 		org.apache.axiom.soap.SOAPEnvelope omEnvelope = builder.getSOAPEnvelope();

 		// Step 1: Get the SAAJConverter object from the Factory
 		SAAJConverterFactory f = (SAAJConverterFactory)
 			FactoryRegistry.getFactory(SAAJConverterFactory.class);
 		SAAJConverter converter = f.getSAAJConverter();

 		// Step 2: Convert the OM SOAPEnvelope to an SAAJ SOAPEnvelope
 		SOAPEnvelope saajEnvelope = converter.toSAAJ(omEnvelope, true).getSOAPPart().getEnvelope();

 		// Step 3: Verify attribute type is stored after conversion
 		Element testElement = (Element) saajEnvelope.getBody().getFirstChild().getFirstChild();
 		assertTrue("b".equals(testElement.getLocalName()));

 		List attrs = new ArrayList();
         NamedNodeMap map = testElement.getAttributes();

         if (map != null) {
             for (int i = 0; i < map.getLength(); i++) {
                 Attr attr = (Attr)map.item(i);
                 if (attr.getName().equals("xmlns") || attr.getName().startsWith("xmlns:")) {
                     // this is a namespace declaration
                 } else {
                     attrs.add(attr);
                 }
             }
         }

         Attr attr = (Attr)attrs.get(0);
         TypeInfo typeInfo = null;
         String attrType = null;

         try {
         	typeInfo = attr.getSchemaTypeInfo();
             if (typeInfo != null) {
     	    	attrType = typeInfo.getTypeName();
             }
         } catch (Throwable t) {
         	;
         }

         if (attrType == null) {
             attrType = (String) attr.getUserData(SAAJConverter.OM_ATTRIBUTE_KEY);
         }
         assert("CDATA".equals(attrType));
 	}
 }
	/*
	* 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.jaxws.message;

	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.impl.builder.StAXSOAPModelBuilder;
	import org.apache.axis2.jaxws.message.factory.SAAJConverterFactory;
	import org.apache.axis2.jaxws.message.util.SAAJConverter;
	import org.apache.axis2.jaxws.registry.FactoryRegistry;
	import org.w3c.dom.Attr;
	import org.w3c.dom.Element;
	import org.w3c.dom.NamedNodeMap;
	import org.w3c.dom.TypeInfo;

	import javax.xml.soap.MessageFactory;
	import javax.xml.soap.SOAPBody;
	import javax.xml.soap.SOAPBodyElement;
	import javax.xml.soap.SOAPElement;
	import javax.xml.soap.SOAPEnvelope;
	import javax.xml.soap.SOAPFactory;
	import javax.xml.soap.SOAPMessage;
	import javax.xml.stream.XMLInputFactory;
	import javax.xml.stream.XMLStreamReader;
	import java.io.StringReader;
	import java.util.ArrayList;
	import java.util.List;

	/**
	* SAAJConverterTests
	*
	* Test the basic functionality of the SAAJConverter.
	* You can also use these tests to as sample code on how to use
	* the converter.
	*
	*/
	public class SAAJConverterTests extends TestCase {

	private static final String sampleText =
	"<pre:a xmlns:pre=\"urn://sample\">" +
	"<b>Hello</b>" +
	"<c>World</c>" +
	"</pre:a>";

	private static final String sampleEnvelope =
	"<soapenv:Envelope xmlns:soapenv=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
	"<soapenv:Header /><soapenv:Body>" +
	sampleText +
	"</soapenv:Body></soapenv:Envelope>";

	private static final String sampleText1 =
	"<pre:a xmlns:pre=\"urn://sample\">" +
	"<b id=\"100\">Hello</b>" +
	"<c>World</c>" +
	"</pre:a>";

	private static final String sampleEnvelope1 =
	"<soapenv:Envelope xmlns:soapenv=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
	"<soapenv:Header /><soapenv:Body>" +
	sampleText1 +
	"</soapenv:Body></soapenv:Envelope>";

	private static XMLInputFactory inputFactory = XMLInputFactory.newInstance();

	public SAAJConverterTests() {
	super();
	}

	public SAAJConverterTests(String arg0) {
	super(arg0);
	}

	/**
	* @testStrategy Tests conversions between SAAJ and OM SOAPEnvelopes
	*/
	public void test1() throws Exception {

	// Bootstrap: Create an OM SOAPEnvelope from the sample text
	StringReader sr = new StringReader(sampleEnvelope);
	XMLStreamReader inflow = inputFactory.createXMLStreamReader(sr);
	StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(inflow, null);
	org.apache.axiom.soap.SOAPEnvelope omEnvelope = builder.getSOAPEnvelope();

	// Step 1: Get the SAAJConverter object from the Factory
	SAAJConverterFactory f = (SAAJConverterFactory)
	FactoryRegistry.getFactory(SAAJConverterFactory.class);
	SAAJConverter converter = f.getSAAJConverter();

	// Step 2: Convert the OM SOAPEnvelope to an SAAJ SOAPEnvelope
	SOAPEnvelope saajEnvelope = converter.toSAAJ(omEnvelope, true).getSOAPPart().getEnvelope();

	// Step 2a: Simple assertion check to ensure correctness.
	String name = saajEnvelope.getBody().getFirstChild().getLocalName();
	assertTrue("a".equals(name));

	// Step 3: Convert the SAAJ SOAPEnvelope to an OM SOAPEnvelope
	omEnvelope = converter.toOM(saajEnvelope);

	// Step 3a: Simple assertion check to ensure correctness
	name = omEnvelope.getBody().getFirstElement().getLocalName();
	assertTrue("a".equals(name));

	// Step 4: Rinse and repeat
	saajEnvelope = converter.toSAAJ(omEnvelope, true).getSOAPPart().getEnvelope();
	name = saajEnvelope.getBody().getFirstChild().getLocalName();
	assertTrue("a".equals(name));
	omEnvelope = converter.toOM(saajEnvelope);
	name = omEnvelope.getBody().getFirstElement().getLocalName();
	assertTrue("a".equals(name));
	}

	/**
	* @testStrategy Tests conversions between SAAJ and OM for normal element
	*/
	public void test2() throws Exception {

	// Bootstrap: Create an OM SOAPEnvelope from the sample text.
	StringReader sr = new StringReader(sampleEnvelope);
	XMLStreamReader inflow = inputFactory.createXMLStreamReader(sr);
	StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(inflow, null);
	org.apache.axiom.soap.SOAPEnvelope omEnvelope = builder.getSOAPEnvelope();

	// Bootstrap: Get an OMElement from the body
	OMElement om = omEnvelope.getBody().getFirstElement();

	// Bootstrap: Get an SAAJ Body to hold the target SOAPElement
	MessageFactory msgFactory = MessageFactory.newInstance();
	SOAPMessage message = msgFactory.createMessage();
	SOAPBody body = message.getSOAPBody();

	// Step 1: Get the SAAJConverter object from the Factory
	SAAJConverterFactory f = (SAAJConverterFactory)
	FactoryRegistry.getFactory(SAAJConverterFactory.class);
	SAAJConverter converter = f.getSAAJConverter();

	// Step 2: Convert OM to SAAJ
	SOAPElement se = converter.toSAAJ(om, body);

	// Step 2a: Verify
	assertTrue(se instanceof SOAPBodyElement);
	assertTrue(se.getLocalName().equals("a"));

	// Step 3: Convert SAAJ to OM
	om = converter.toOM(se);

	// Step 3a: Verify
	assertTrue(om.getLocalName().equals("a"));

	// Step 4: Rinse and Repeat
	se = converter.toSAAJ(om, body);
	assertTrue(se instanceof SOAPBodyElement);
	assertTrue(se.getLocalName().equals("a"));
	om = converter.toOM(se);
	assertTrue(om.getLocalName().equals("a"));
	}

	/**
	* @testStrategy: Create an OMElement, without using a builder. Verification of AXIS2-970
	*/
	public void test3() throws Exception {

	// Step 1: Get the SAAJConverter object from the Factory
	SAAJConverterFactory f = (SAAJConverterFactory)
	FactoryRegistry.getFactory(SAAJConverterFactory.class);
	SAAJConverter converter = f.getSAAJConverter();

	// Stept 2: Create OM and parent SOAPElement
	OMFactory fac = OMAbstractFactory.getOMFactory();
	OMNamespace wrapNs = fac.createOMNamespace("namespace", "prefix");
	OMElement ome = fac.createOMElement("localname", wrapNs);
	SOAPFactory sf = SOAPFactory.newInstance();
	SOAPElement se = sf.createElement("name");

	// Step 3: Do the conversion
	converter.toSAAJ(ome, se, sf);
	}

	/**
	* @testStrategy Tests conversions between OM and SAAJ SOAPEnvelopes
	* and verify attribute type is correctly stored
	*/
	public void test4() throws Exception {
	// Bootstrap: Create an OM SOAPEnvelope from the sample text
	StringReader sr = new StringReader(sampleEnvelope1);
	XMLStreamReader inflow = inputFactory.createXMLStreamReader(sr);
	StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(inflow, null);
	org.apache.axiom.soap.SOAPEnvelope omEnvelope = builder.getSOAPEnvelope();

	// Step 1: Get the SAAJConverter object from the Factory
	SAAJConverterFactory f = (SAAJConverterFactory)
	FactoryRegistry.getFactory(SAAJConverterFactory.class);
	SAAJConverter converter = f.getSAAJConverter();

	// Step 2: Convert the OM SOAPEnvelope to an SAAJ SOAPEnvelope
	SOAPEnvelope saajEnvelope = converter.toSAAJ(omEnvelope, true).getSOAPPart().getEnvelope();

	// Step 3: Verify attribute type is stored after conversion
	Element testElement = (Element) saajEnvelope.getBody().getFirstChild().getFirstChild();
	assertTrue("b".equals(testElement.getLocalName()));

	List attrs = new ArrayList();
	NamedNodeMap map = testElement.getAttributes();

	if (map != null) {
	for (int i = 0; i < map.getLength(); i++) {
	Attr attr = (Attr)map.item(i);
	if (attr.getName().equals("xmlns") \|\| attr.getName().startsWith("xmlns:")) {
	// this is a namespace declaration
	} else {
	attrs.add(attr);
	}
	}
	}

	Attr attr = (Attr)attrs.get(0);
	TypeInfo typeInfo = null;
	String attrType = null;

	try {
	typeInfo = attr.getSchemaTypeInfo();
	if (typeInfo != null) {
	attrType = typeInfo.getTypeName();
	}
	} catch (Throwable t) {
	;
	}

	if (attrType == null) {
	attrType = (String) attr.getUserData(SAAJConverter.OM_ATTRIBUTE_KEY);
	}
	assert("CDATA".equals(attrType));
	}
	}