/*
* 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.saaj;
import junit.framework.TestCase;
import javax.xml.soap.MessageFactory;
import javax.xml.soap.MimeHeaders;
import javax.xml.soap.Name;
import javax.xml.soap.SOAPBody;
import javax.xml.soap.SOAPElement;
import javax.xml.soap.SOAPEnvelope;
import javax.xml.soap.SOAPHeader;
import javax.xml.soap.SOAPMessage;
import javax.xml.soap.SOAPPart;
import javax.xml.soap.Text;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.util.Iterator;
public class PrefixesTest extends TestCase {
public PrefixesTest(String name) {
super(name);
}
public void testAddingPrefixesForChildElements() throws Exception {
MessageFactory factory = MessageFactory.newInstance();
SOAPMessage msg = factory.createMessage();
SOAPPart sp = msg.getSOAPPart();
SOAPEnvelope se = sp.getEnvelope();
SOAPBody sb = se.getBody();
SOAPElement el1 = sb.addBodyElement(se.createName("element1",
"prefix1",
"http://www.sun.com"));
el1.addChildElement(se.createName("element2",
"prefix2",
"http://www.apache.org"));
ByteArrayOutputStream baos = new ByteArrayOutputStream();
msg.writeTo(baos);
String xml = new String(baos.toByteArray());
System.out.println("########## xml = " + xml);
assertTrue(xml.indexOf("prefix1") != -1);
assertTrue(xml.indexOf("prefix2") != -1);
assertTrue(xml.indexOf("http://www.sun.com") != -1);
assertTrue(xml.indexOf("http://www.apache.org") != -1);
}
public void testAttribute() throws Exception {
String soappacket =
"<soapenv:Envelope xmlns:soapenv =\"http://schemas.xmlsoap.org/soap/envelope/\"\n" +
" xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\"\n" +
" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">\n" +
" <soapenv:Body>\n" +
" <t:helloworld t:name=\"test\" xmlns:t='http://test.org/Test'>Hello</t:helloworld>\n" +
" </soapenv:Body>\n" +
"</soapenv:Envelope>";
SOAPMessage msg =
MessageFactory.newInstance().createMessage(new MimeHeaders(),
new ByteArrayInputStream(soappacket.getBytes()));
SOAPBody body = msg.getSOAPPart().getEnvelope().getBody();
msg.writeTo(System.out);
validateBody(body.getChildElements());
}
private void validateBody(Iterator iter) {
while (iter.hasNext()) {
final Object obj = iter.next();
if (obj instanceof Text) {
System.out.println("\n- Text Ignored.");
} else {
final SOAPElement soapElement = (SOAPElement) obj;
final Iterator attIter = soapElement.getAllAttributes();
while (attIter.hasNext()) {
final Name name = (Name) attIter.next();
assertEquals("test", soapElement.getAttributeValue(name));
assertEquals("t", name.getPrefix());
assertEquals("t:name", name.getQualifiedName());
assertEquals("name", name.getLocalName());
assertEquals("http://test.org/Test", name.getURI());
}
final Iterator childElementIter = soapElement.getChildElements();
if (childElementIter == null) return;
validateBody(childElementIter);
}
}
}
public void testAttrPrifix() {
try {
MessageFactory fac = MessageFactory.newInstance();
SOAPMessage msg = fac.createMessage();
SOAPEnvelope env = msg.getSOAPPart().getEnvelope();
SOAPHeader header = msg.getSOAPHeader();
Name name = env.createName("Local","pre1", "http://test1");
SOAPElement local = header.addChildElement(name);
Name name2 = env.createName("Local1","pre1", "http://test1");
SOAPElement local2 = local.addChildElement(name2);
Name aName = env.createName("attrib","pre1", "http://test1");
local2.addAttribute(aName, "value");
ByteArrayOutputStream baos = new ByteArrayOutputStream();
msg.writeTo(baos);
String xml = new String(baos.toByteArray());
assertTrue(xml.indexOf("xmlns:http://test1") == -1);
assertTrue(xml.indexOf("pre1:attrib=\"value\"") > 0);
} catch (Exception e) {
e.printStackTrace();
fail(e.getMessage());
}
}
}