/*
* 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 java.io.StringReader;
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 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;
/**
* 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 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);
// 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);
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);
}
}