| /* |
| * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| package stream.XMLStreamFilterTest; |
| |
| import java.io.InputStream; |
| |
| import javax.xml.namespace.NamespaceContext; |
| import javax.xml.namespace.QName; |
| import javax.xml.stream.EventFilter; |
| import javax.xml.stream.StreamFilter; |
| import javax.xml.stream.XMLInputFactory; |
| import javax.xml.stream.XMLStreamConstants; |
| import javax.xml.stream.XMLStreamReader; |
| import javax.xml.stream.events.XMLEvent; |
| |
| import org.testng.Assert; |
| import org.testng.annotations.Test; |
| |
| /* |
| * @bug 6481678 |
| * @summary Test Filtered XMLStreamReader parses namespace correctly. |
| */ |
| public class Bug6481678 { |
| |
| String rootElement = "fruits"; |
| String childElement = "varieties"; |
| String prefixApple = "a"; |
| String namespaceURIApple = "apple.fruit"; |
| String prefixOrange = "o"; |
| String namespaceURIOrange = "orange.fruit"; |
| String namespaceURIBanana = "banana.fruit"; |
| |
| TypeFilter filter; |
| XMLInputFactory factory; |
| InputStream is; |
| |
| /** Creates a new instance of NamespaceTest */ |
| public Bug6481678(java.lang.String testName) { |
| init(); |
| } |
| |
| private void init() { |
| factory = XMLInputFactory.newInstance(); |
| factory.setProperty(XMLInputFactory.IS_NAMESPACE_AWARE, Boolean.TRUE); |
| filter = createFilter(); |
| } |
| |
| String getXML() { |
| StringBuffer sbuffer = new StringBuffer(); |
| sbuffer.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>"); |
| sbuffer.append("<" + rootElement + " state=\"WA\""); |
| sbuffer.append(" xmlns:" + prefixApple + "=\"" + namespaceURIApple + "\""); |
| sbuffer.append(" xmlns:" + prefixOrange + "=\"" + namespaceURIOrange + "\""); |
| sbuffer.append(" xmlns=\"" + namespaceURIBanana + "\">"); |
| sbuffer.append("<" + prefixApple + ":" + childElement + ">"); |
| sbuffer.append("<" + prefixApple + ":fuji/>"); |
| sbuffer.append("<" + prefixApple + ":gala/>"); |
| sbuffer.append("</" + prefixApple + ":" + childElement + ">"); |
| sbuffer.append("</" + rootElement + ">"); |
| // System.out.println("XML = " + sbuffer.toString()) ; |
| return sbuffer.toString(); |
| } |
| |
| public TypeFilter createFilter() { |
| |
| TypeFilter f = new TypeFilter(); |
| |
| f.addType(XMLEvent.START_ELEMENT); |
| f.addType(XMLEvent.END_ELEMENT); |
| f.addType(XMLEvent.PROCESSING_INSTRUCTION); |
| f.addType(XMLEvent.CHARACTERS); |
| f.addType(XMLEvent.COMMENT); |
| f.addType(XMLEvent.SPACE); |
| f.addType(XMLEvent.START_DOCUMENT); |
| f.addType(XMLEvent.END_DOCUMENT); |
| return f; |
| } |
| |
| /* |
| * testcase for cr6481678 in our current impl (using cache), the reader |
| * would read from cache when getters are called before next() is. refter to |
| * testRootElementNamespace. |
| */ |
| @Test |
| public void testReadingNamespace() { |
| is = new java.io.ByteArrayInputStream(getXML().getBytes()); |
| try { |
| XMLStreamReader sr = factory.createFilteredReader(factory.createXMLStreamReader(is), (StreamFilter) filter); |
| |
| while (sr.hasNext()) { |
| int eventType = sr.getEventType(); |
| if (eventType == XMLStreamConstants.START_ELEMENT) { |
| if (sr.getLocalName().equals(rootElement)) { |
| Assert.assertTrue(sr.getNamespacePrefix(0).equals(prefixApple) && sr.getNamespaceURI(0).equals(namespaceURIApple)); |
| } |
| } |
| eventType = sr.next(); |
| } |
| } catch (Exception ex) { |
| Assert.fail("Exception: " + ex.getMessage()); |
| } |
| } |
| |
| @Test |
| public void testRootElementNamespace() { |
| is = new java.io.ByteArrayInputStream(getXML().getBytes()); |
| try { |
| XMLStreamReader sr = factory.createFilteredReader(factory.createXMLStreamReader(is), (StreamFilter) filter); |
| |
| while (sr.hasNext()) { |
| int eventType = sr.next(); |
| if (eventType == XMLStreamConstants.START_ELEMENT) { |
| if (sr.getLocalName().equals(rootElement)) { |
| Assert.assertTrue(sr.getNamespacePrefix(0).equals(prefixApple) && sr.getNamespaceURI(0).equals(namespaceURIApple)); |
| } |
| } |
| } |
| } catch (Exception ex) { |
| Assert.fail("Exception: " + ex.getMessage()); |
| } |
| } |
| |
| @Test |
| public void testChildElementNamespace() { |
| is = new java.io.ByteArrayInputStream(getXML().getBytes()); |
| try { |
| XMLStreamReader sr = factory.createFilteredReader(factory.createXMLStreamReader(is), (StreamFilter) filter); |
| while (sr.hasNext()) { |
| int eventType = sr.next(); |
| if (eventType == XMLStreamConstants.START_ELEMENT) { |
| if (sr.getLocalName().equals(childElement)) { |
| QName qname = sr.getName(); |
| Assert.assertTrue(qname.getPrefix().equals(prefixApple) && qname.getNamespaceURI().equals(namespaceURIApple) |
| && qname.getLocalPart().equals(childElement)); |
| } |
| } |
| } |
| } catch (Exception ex) { |
| Assert.fail("Exception: " + ex.getMessage()); |
| } |
| } |
| |
| @Test |
| public void testNamespaceContext() { |
| is = new java.io.ByteArrayInputStream(getXML().getBytes()); |
| try { |
| XMLStreamReader sr = factory.createFilteredReader(factory.createXMLStreamReader(is), (StreamFilter) filter); |
| while (sr.hasNext()) { |
| int eventType = sr.next(); |
| if (eventType == XMLStreamConstants.START_ELEMENT) { |
| if (sr.getLocalName().equals(childElement)) { |
| NamespaceContext context = sr.getNamespaceContext(); |
| Assert.assertTrue(context.getPrefix(namespaceURIApple).equals(prefixApple)); |
| } |
| } |
| } |
| } catch (Exception ex) { |
| Assert.fail("Exception: " + ex.getMessage()); |
| } |
| } |
| |
| @Test |
| public void testNamespaceCount() { |
| is = new java.io.ByteArrayInputStream(getXML().getBytes()); |
| try { |
| XMLStreamReader sr = factory.createFilteredReader(factory.createXMLStreamReader(is), (StreamFilter) filter); |
| while (sr.hasNext()) { |
| int eventType = sr.next(); |
| if (eventType == XMLStreamConstants.START_ELEMENT) { |
| if (sr.getLocalName().equals(rootElement)) { |
| int count = sr.getNamespaceCount(); |
| Assert.assertTrue(count == 3); |
| } |
| } |
| } |
| } catch (Exception ex) { |
| Assert.fail("Exception: " + ex.getMessage()); |
| } |
| } |
| |
| class TypeFilter implements EventFilter, StreamFilter { |
| |
| protected boolean[] types = new boolean[20]; |
| |
| public TypeFilter() { |
| } |
| |
| public void addType(int type) { |
| types[type] = true; |
| } |
| |
| public boolean accept(XMLEvent e) { |
| return types[e.getEventType()]; |
| } |
| |
| public boolean accept(XMLStreamReader r) { |
| return types[r.getEventType()]; |
| } |
| } |
| } |