/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you 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 tests.ext;

import java.util.Map;

import javax.xml.parsers.DocumentBuilderFactory;

import org.w3c.dom.Attr;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.NodeList;

import org.apache.ws.commons.schema.XmlSchema;
import org.apache.ws.commons.schema.XmlSchemaCollection;
import org.apache.ws.commons.schema.XmlSchemaElement;
import org.apache.ws.commons.schema.constants.Constants;

import org.junit.Assert;
import org.junit.Test;

import tests.Resources;

/**
 * Test the custom extension deserialization without any specialized hooks
 */
public class PlainExtensionDeserializerTest extends Assert {

    @Test
    public void testDeserialization() throws Exception {

        // create a DOM document
        DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
        documentBuilderFactory.setNamespaceAware(true);
        Document doc = documentBuilderFactory.newDocumentBuilder()
            .parse(Resources.asURI("/external/externalAnnotations.xsd"));

        XmlSchemaCollection schemaCol = new XmlSchemaCollection();
        XmlSchema schema = schemaCol.read(doc, null);
        assertNotNull(schema);

        // get the elements and check whether their annotations are properly
        // populated
        for (XmlSchemaElement elt : schema.getElements().values()) {
            Map<Object, Object> metaInfoMap = elt.getMetaInfoMap();
            assertNotNull(metaInfoMap);

        }
    }

    @Test
    public void testDeserialization1() throws Exception {

        // create a DOM document
        DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
        documentBuilderFactory.setNamespaceAware(true);
        Document doc = documentBuilderFactory.newDocumentBuilder()
            .parse(Resources.asURI("/external/externalElementAnnotations.xsd"));

        XmlSchemaCollection schemaCol = new XmlSchemaCollection();
        XmlSchema schema = schemaCol.read(doc, null);
        assertNotNull(schema);

        // get the elements and check whether their annotations are properly
        // populated
        for (XmlSchemaElement elt : schema.getElements().values()) {
            assertNotNull(elt);
            Map<Object, Object> metaInfoMap = elt.getMetaInfoMap();
            assertNotNull(metaInfoMap);

        }
    }
    
    @Test
    public void testExtensionAttributeNamespace() throws Exception {
        //Test for XMLSCHEMA-23.  Need to deserialize the extension attributes and
        //then serialize it again, but retain the namespace declarations
        
        // create a DOM document
        DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
        documentBuilderFactory.setNamespaceAware(true);
        Document doc = documentBuilderFactory.newDocumentBuilder()
            .parse(Resources.asURI("/XMLSCHEMA-23/test.xsd"));

        XmlSchemaCollection schemaCol = new XmlSchemaCollection();
        XmlSchema schema = schemaCol.read(doc, null);
        assertNotNull(schema);

        
        doc = schema.getSchemaDocument();
        NodeList nl = doc.getDocumentElement()
                .getElementsByTagNameNS("http://www.w3.org/2001/XMLSchema", "element");
        assertEquals(1, nl.getLength());
        Element del = (Element)nl.item(0);
        NamedNodeMap mp = del.getAttributes();
        for (int x = 0; x < mp.getLength(); x++) {
            Attr attr = (Attr)mp.item(x);
            if (attr.getNamespaceURI() != null 
                && !"".equals(attr.getNamespaceURI())
                && !Constants.XMLNS_ATTRIBUTE_NS_URI.equals(attr.getNamespaceURI())) {
                String ns = del.lookupNamespaceURI(attr.getPrefix());
                String pfx = del.lookupPrefix(attr.getNamespaceURI());
                assertEquals(attr.getPrefix(), pfx);
                assertEquals(attr.getNamespaceURI(), ns);
            }
        }
    }
    
    
}