package org.apache.axiom.om;
import org.apache.axiom.om.impl.builder.StAXOMBuilder;
import org.apache.axiom.om.xpath.AXIOMXPath;
import org.custommonkey.xmlunit.XMLTestCase;
import org.jaxen.JaxenException;
import org.jaxen.SimpleNamespaceContext;
import org.jaxen.XPath;
import javax.xml.namespace.QName;
import javax.xml.stream.*;
import java.io.ByteArrayInputStream;
import java.util.Iterator;
/*
* 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.
*/
public class DefaultNSHandlingTest extends XMLTestCase {
public void testDefaultNamespaceWithSameDefaultNSForAll() {
String testXML = "<html xmlns='http://www.w3.org/TR/REC-html40'>" +
"<head><title>Frobnostication</title></head>" +
"<body><p>Moved to <a href='http://frob.com'>here</a>.</p></body>" +
"</html>";
try {
StAXOMBuilder stAXOMBuilder = new StAXOMBuilder(new ByteArrayInputStream(testXML.getBytes()));
OMElement documentElement = stAXOMBuilder.getDocumentElement();
checkNS(documentElement);
checkNSWithChildren(documentElement);
} catch (XMLStreamException e) {
e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates.
}
}
private void checkNSWithChildren(OMElement documentElement) {
Iterator childElementsIter = documentElement.getChildElements();
while (childElementsIter.hasNext()) {
OMElement omElement = (OMElement) childElementsIter.next();
checkNS(omElement);
checkNSWithChildren(omElement);
}
}
private void checkNS(OMElement element) {
assertTrue("http://www.w3.org/TR/REC-html40".equals(element.getNamespace().getNamespaceURI()));
}
public void testMultipleDefaultNS() {
OMFactory omFactory = OMAbstractFactory.getOMFactory();
OMNamespace defaultNS1 = omFactory.createOMNamespace("http://defaultNS1.org", null);
OMNamespace defaultNS2 = omFactory.createOMNamespace("http://defaultNS2.org", null);
OMElement omElementOne = omFactory.createOMElement("DocumentElement", null);
omElementOne.declareDefaultNamespace("http://defaultNS1.org");
OMElement omElementOneChild = omFactory.createOMElement("ChildOne", null, omElementOne);
OMElement omElementTwo = omFactory.createOMElement("Foo", defaultNS2, omElementOne);
omElementTwo.declareDefaultNamespace("http://defaultNS2.org");
OMElement omElementTwoChild = omFactory.createOMElement("ChildOne", null, omElementTwo);
OMElement omElementThree = omFactory.createOMElement("Bar", defaultNS1, omElementTwo);
omElementThree.declareDefaultNamespace("http://defaultNS1.org");
assertTrue("".equals(omElementOneChild.getNamespace().getNamespaceURI()));
assertTrue("".equals(omElementTwoChild.getNamespace().getNamespaceURI()));
}
public void testChildReDeclaringParentsDefaultNSWithPrefix() {
try {
OMFactory fac = OMAbstractFactory.getOMFactory();
OMElement elem = fac.createOMElement("RequestSecurityToken", null);
elem.declareDefaultNamespace("http://schemas.xmlsoap.org/ws/2005/02/trust");
fac.createOMElement(new QName("TokenType"), elem).setText("test");
fac.createOMElement(new QName("RequestType"), elem).setText("test1");
fac.createOMElement(new QName("http://schemas.xmlsoap.org/ws/2005/02/trust","Entropy", "wst"), elem);
String xml = elem.toString();
XMLStreamReader reader = XMLInputFactory.newInstance().createXMLStreamReader(new ByteArrayInputStream(xml.getBytes()));
StAXOMBuilder builder = new StAXOMBuilder(reader);
builder.getDocumentElement().build();
// The StAX implementation may or may not have a trailing blank in the tag
String assertText1 = "<wst:Entropy xmlns:wst=\"http://schemas.xmlsoap.org/ws/2005/02/trust\" />";
String assertText2 = "<wst:Entropy xmlns:wst=\"http://schemas.xmlsoap.org/ws/2005/02/trust\"/>";
String assertText3 = "<wst:Entropy xmlns:wst=\"http://schemas.xmlsoap.org/ws/2005/02/trust\"></wst:Entropy>";
assertTrue((xml.indexOf(assertText1) != -1) ||
(xml.indexOf(assertText2) != -1) ||
(xml.indexOf(assertText3) != -1));
}catch (Exception e) {
fail(e.getMessage());
}
}
public void testChildReDeclaringGrandParentsDefaultNSWithPrefix() {
try {
OMFactory fac = OMAbstractFactory.getOMFactory();
OMElement elem = fac.createOMElement("RequestSecurityToken", null);
elem.declareDefaultNamespace("http://schemas.xmlsoap.org/ws/2005/02/trust");
fac.createOMElement(new QName("TokenType"), elem).setText("test");
fac.createOMElement(new QName("RequestType"), elem).setText("test1");
OMElement entElem = fac.createOMElement(new QName("http://schemas.xmlsoap.org/ws/2005/02/trust","Entropy", "wst"), elem);
OMElement binSecElem = fac.createOMElement(new QName("http://schemas.xmlsoap.org/ws/2005/02/trust","Binarysecret", "wst"), entElem);
binSecElem.setText("secret value");
String xml = elem.toString();
assertTrue("Binarysecret element should have \'wst\' ns prefix", xml.indexOf("<wst:Binarysecret") != -1);
}catch (Exception e) {
fail(e.getMessage());
}
}
// public void testForIssueWSCOMMONS119() {
//
// try {
// String planXML = "test-resources/xml/defaultNamespace2.xml";
// XMLStreamReader parser = XMLInputFactory.newInstance().
// createXMLStreamReader(new FileInputStream(new File(planXML)));
// StAXOMBuilder staxOMBuilder = new StAXOMBuilder(parser);
// OMElement docEle = staxOMBuilder.getDocumentElement();
// OMElement omElement = getOMElement("//ns:config-property-setting[@name='ConnectionURL']",
// docEle);
// omElement.setText("jdbc:derby:/home/azeez/.tungsten/database/TUNGSTEN_DB");
//
// String serializedXML = docEle.toString();
//
// System.out.println("serializedXML = " + serializedXML);
//
// assertTrue(serializedXML.indexOf("xmlns=\"\"") == -1);
//
// } catch (XMLStreamException e) {
// fail();
// e.printStackTrace();
// } catch (FileNotFoundException e) {
// fail();
// e.printStackTrace();
// } catch (JaxenException e) {
// fail();
// e.printStackTrace();
// }
// }
private OMElement getOMElement(String xpathString,
OMElement parentElement) throws JaxenException {
XPath xpath = getXPath(xpathString);
return (OMElement) xpath.selectSingleNode(parentElement);
}
private XPath getXPath(String xpathString) throws JaxenException {
SimpleNamespaceContext nsCtx = new SimpleNamespaceContext();
nsCtx.addNamespace("ns", "http://geronimo.apache.org/xml/ns/j2ee/connector-1.1");
XPath xpath = new AXIOMXPath(xpathString);
xpath.setNamespaceContext(nsCtx);
return xpath;
}
public static void main(String[] args) {
try {
XMLStreamWriter xmlStreamWriter = XMLOutputFactory.newInstance().createXMLStreamWriter(System.out);
xmlStreamWriter.writeStartElement("Foo");
xmlStreamWriter.writeDefaultNamespace("test.org");
xmlStreamWriter.setDefaultNamespace("test.org");
xmlStreamWriter.writeStartElement("Bar");
xmlStreamWriter.writeEndElement();
xmlStreamWriter.writeEndElement();
xmlStreamWriter.flush();
} catch (XMLStreamException e) {
e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates.
}
}
}