jaxb-ri/core/src/test/java/org/glassfish/jaxb/core/unmarshaller/DOMScannerTest.java

/*
 * Copyright (c) 2021, 2022 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Distribution License v. 1.0, which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

package org.glassfish.jaxb.core.unmarshaller;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;

import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.Attributes;
import org.xml.sax.ContentHandler;
import org.xml.sax.Locator;
import org.xml.sax.SAXException;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

public class DOMScannerTest extends TestCase {
	
	public void testParentDefaultNamespace() throws ParserConfigurationException, SAXException, IOException {
		DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
		DocumentBuilder builder = dbf.newDocumentBuilder();
		InputStream is = this.getClass().getClassLoader().getResourceAsStream("parentDefaultNs.xml");
		Document doc = builder.parse(is);
		NodeList testElems = doc.getElementsByTagName("test");
		Node testElem = testElems.item(0);
		
		DOMScanner scanner = new DOMScanner();
		MockContentHandler mockHandler = new MockContentHandler();
		Map<String,String> assertMapping = new HashMap<>();
		assertMapping.put("xml", "http://www.w3.org/XML/1998/namespace");
		assertMapping.put("xsi", "http://www.w3.org/2001/XMLSchema-instance");
		assertMapping.put("", "http://example.com/test/ns");
		mockHandler.setAssertion("value", assertMapping);
		scanner.setContentHandler(mockHandler);
		scanner.scan(testElem);
	}
	
	private static class MockContentHandler implements ContentHandler {
		
		private Map<String,String> prefixMapping = new HashMap<>();
		
		private Map<String,String> assertMapping = new HashMap<>();
		private String assertElement = null;
		
		public void setAssertion(String elementName, Map<String,String> mapping) {
			this.assertElement = elementName;
			this.assertMapping.putAll(mapping);
		}

		@Override
		public void setDocumentLocator(Locator locator) {
			
		}

		@Override
		public void startDocument() {
			
		}

		@Override
		public void endDocument() {
			
		}

		@Override
		public void startPrefixMapping(String prefix, String uri) {
			prefixMapping.put(prefix, uri);
		}

		@Override
		public void endPrefixMapping(String prefix) {
			prefixMapping.remove(prefix);
		}

		@Override
		public void startElement(String uri, String localName, String qName, Attributes atts) {
			if (assertElement != null && assertElement.equals(localName)) {
				assertEquals("Prefix mapping does not match expected mapping", prefixMapping, assertMapping);
			}
		}

		@Override
		public void endElement(String uri, String localName, String qName) {
			
		}

		@Override
		public void characters(char[] ch, int start, int length) {
			
		}

		@Override
		public void ignorableWhitespace(char[] ch, int start, int length) {
			
		}

		@Override
		public void processingInstruction(String target, String data) {
			
		}

		@Override
		public void skippedEntity(String name) {
			
		}
		
	}

}