/*
* 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.om;
import org.apache.axis2.om.impl.llom.builder.StAXOMBuilder;
import org.apache.axis2.om.impl.llom.factory.OMLinkedListImplFactory;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLInputFactory;
import java.io.FileReader;
import java.util.Iterator;
public class IteratorTest extends AbstractTestCase {
private OMElement envelope = null;
public IteratorTest(String testName) {
super(testName);
}
protected void setUp() throws Exception {
//lets use a plain OM factory
envelope =
new StAXOMBuilder(new OMLinkedListImplFactory(),
XMLInputFactory.newInstance().createXMLStreamReader(
new FileReader(
getTestResourceFile(
"soap/soapmessage1.xml")))).getDocumentElement();
}
protected void tearDown() throws Exception {
envelope = null;
}
/**
* Test the plain iterator which includes all the
* children (including the texts)
*/
public void testIterator() {
OMElement elt = envelope;
Iterator iter = elt.getChildren();
int counter = 0;
while (iter.hasNext()) {
counter ++;
assertNotNull("Must return not null objects!",iter.next());
}
assertEquals("This element should contain only five children including the text ",5,counter);
}
/**
* Test the element iterator
*/
public void testElementIterator() {
OMElement elt = envelope;
Iterator iter = elt.getChildElements();
int counter = 0;
while (iter.hasNext()) {
counter ++;
Object o = iter.next();
assertNotNull("Must return not null objects!",o);
assertTrue("All these should be elements!",((OMNode)o).getType()==OMNode.ELEMENT_NODE);
}
assertEquals("This element should contain only two elements ",2,counter);
}
/**
* Test the element iterator
*/
public void testElementQNameIterator() {
OMElement elt = envelope;
QName qname = new QName("http://schemas.xmlsoap.org/soap/envelope/","body");
Iterator iter = elt.getChildrenWithName(qname);
int counter = 0;
while (iter.hasNext()) {
counter ++;
Object o = iter.next();
assertNotNull("Must return not null objects!",o);
assertTrue("All these should be elements!",((OMNode)o).getType()==OMNode.ELEMENT_NODE);
}
assertEquals("This element should contain only one element with the given QName ",1,counter);
}
/**
* test the remove exception behavior
*/
public void testIteratorRemove1() {
OMElement elt = envelope;
Iterator iter = elt.getChildren();
//this is supposed to throw an illegal state exception
try {
iter.remove();
fail("remove should throw an exception");
} catch (IllegalStateException e) {
//ok. this is what should happen
}
}
/**
* test the remove exception behavior, consecutive remove calls
*/
public void testIteratorRemove2() {
OMElement elt = envelope;
Iterator iter = elt.getChildren();
if (iter.hasNext()) {
iter.next();
}
iter.remove();
//this call must generate an exception
try {
iter.remove();
fail("calling remove twice without a call to next is prohibited");
} catch (IllegalStateException e) {
//ok if we come here :)
}
}
/**
* Remove all!
*/
public void testIteratorRemove3() {
OMElement elt = envelope;
Iterator iter = elt.getChildren();
while (iter.hasNext()) {
iter.next();
iter.remove();
}
iter = elt.getChildren();
if (iter.hasNext()) {
//we shouldn't reach here!
fail("No children should remain after removing all!");
}
}
/**
* test whether the children count reduces.
*/
public void testIteratorRemove4() {
OMElement elt = envelope;
Iterator iter = elt.getChildren();
int firstChildrenCount = 0;
int secondChildrenCount = 0;
while (iter.hasNext()) {
assertNotNull(iter.next());
firstChildrenCount++;
}
//remove the last node
iter.remove();
//reloop and check the count
//Note- here we should get a fresh iterator since there is no method to
//reset the iterator
iter = elt.getChildren(); //reset the iterator
while (iter.hasNext()) {
assertNotNull(iter.next());
secondChildrenCount++;
}
assertEquals("children count must reduce from 1",
firstChildrenCount - 1,
secondChildrenCount);
}
}