| /* | |
| * Copyright 2004,2005 The Apache Software Foundation. | |
| * Copyright 2006 International Business Machines Corp. | |
| * | |
| * 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.jaxws.message; | |
| import java.io.StringReader; | |
| import java.io.StringWriter; | |
| import javax.xml.bind.JAXBContext; | |
| import javax.xml.bind.JAXBIntrospector; | |
| import javax.xml.namespace.QName; | |
| import javax.xml.stream.XMLInputFactory; | |
| import javax.xml.stream.XMLOutputFactory; | |
| import javax.xml.stream.XMLStreamReader; | |
| import javax.xml.stream.XMLStreamWriter; | |
| import org.apache.axis2.jaxws.message.factory.JAXBBlockFactory; | |
| import org.apache.axis2.jaxws.message.factory.XMLStringBlockFactory; | |
| import org.apache.axis2.jaxws.message.util.Reader2Writer; | |
| import org.apache.axis2.jaxws.message.util.XMLStreamReaderSplitter; | |
| import org.apache.axis2.jaxws.registry.FactoryRegistry; | |
| import test.EchoString; | |
| import test.ObjectFactory; | |
| import junit.framework.TestCase; | |
| /** | |
| * Tests XMLStreamReaderSplitter | |
| */ | |
| public class XMLStreamReaderSplitterTests extends TestCase { | |
| // String test variables | |
| private static final String sampleText = | |
| "<body>" + | |
| "spurious text" + | |
| "<pre:a1 xmlns:pre=\"urn://sample\">" + | |
| "<b1>Hello</b1>" + | |
| "<c1>World</c1>" + | |
| "</pre:a1>" + | |
| "<!-- Spurious Comment -->" + | |
| "<pre:a2 xmlns:pre=\"urn://sample2\">" + | |
| "<b2>Hello</b2>" + | |
| "<c2>World</c2>" + | |
| "</pre:a2>" + | |
| "</body>"; | |
| private static XMLInputFactory inputFactory = XMLInputFactory.newInstance(); | |
| private static XMLOutputFactory outputFactory = XMLOutputFactory.newInstance(); | |
| public XMLStreamReaderSplitterTests() { | |
| super(); | |
| } | |
| public XMLStreamReaderSplitterTests(String arg0) { | |
| super(arg0); | |
| } | |
| /** | |
| * Test XMLStreamReaderSplitter | |
| * @throws Exception | |
| */ | |
| public void test() throws Exception { | |
| // Create a full XMLStreamReader for the message | |
| StringReader sr = new StringReader(sampleText); | |
| XMLStreamReader fullReader = inputFactory.createXMLStreamReader(sr); | |
| // Advance past the first element (body) | |
| fullReader.next(); | |
| fullReader.next(); | |
| // Create a Splitter | |
| XMLStreamReaderSplitter splitter = new XMLStreamReaderSplitter(fullReader); | |
| // Pipe the splitter to the writer. This should generated only the | |
| // first element tree. | |
| StringWriter sw = new StringWriter(); | |
| XMLStreamWriter writer = outputFactory.createXMLStreamWriter(sw); | |
| Reader2Writer r2w = new Reader2Writer(splitter); | |
| r2w.outputTo(writer); | |
| writer.flush(); | |
| sw.flush(); | |
| String tree1 = sw.toString(); | |
| // Now get the next Stream | |
| XMLStreamReader stream2 = splitter.getNextReader(); | |
| r2w = new Reader2Writer(stream2); | |
| sw = new StringWriter(); | |
| writer = outputFactory.createXMLStreamWriter(sw); | |
| r2w.outputTo(writer); | |
| writer.flush(); | |
| sw.flush(); | |
| String tree2 = sw.toString(); | |
| // Do assertion checks | |
| assertTrue(!tree1.contains("text")); | |
| assertTrue(!tree1.contains("Comment")); | |
| assertTrue( tree1.contains("a1")); | |
| assertTrue(!tree1.contains("a2")); | |
| assertTrue(!tree2.contains("text")); | |
| assertTrue(!tree2.contains("Comment")); | |
| assertTrue(!tree2.contains("a1")); | |
| assertTrue( tree2.contains("a2")); | |
| } | |
| } |