axiom/modules/axiom-tests/src/test/java/org/apache/axiom/om/DefaultNSHandlingTest.java - ws-axiom - Git at Google

 package org.apache.axiom.om;

 import junit.framework.TestCase;
 import org.apache.axiom.om.impl.builder.StAXOMBuilder;

 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 TestCase {

     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 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.
         }
     }
 }
	package org.apache.axiom.om;

	import junit.framework.TestCase;
	import org.apache.axiom.om.impl.builder.StAXOMBuilder;

	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 TestCase {

	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 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.
	}
	}
	}