/*

 * Copyright (c) 2019, 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 validation;

import java.io.File;

import java.io.FileInputStream;

import javax.xml.XMLConstants;

import javax.xml.parsers.SAXParserFactory;

import javax.xml.stream.XMLInputFactory;

import javax.xml.stream.XMLStreamReader;

import javax.xml.stream.events.XMLEvent;

import javax.xml.transform.Source;

import javax.xml.transform.sax.SAXSource;

import javax.xml.transform.stream.StreamSource;

import javax.xml.validation.Schema;

import javax.xml.validation.SchemaFactory;

import javax.xml.validation.Validator;

import org.testng.annotations.DataProvider;

import org.testng.annotations.Listeners;

import org.testng.annotations.Test;

import org.xml.sax.Attributes;

import org.xml.sax.InputSource;

import org.xml.sax.SAXException;

import org.xml.sax.SAXParseException;

import org.xml.sax.XMLFilter;

import org.xml.sax.helpers.XMLFilterImpl;

/*

 * @test

 * @bug 8220818 8176447

 * @library /javax/xml/jaxp/libs /javax/xml/jaxp/unittest

 * @run testng/othervm validation.ValidationTest

 * @summary Runs validations with schemas and sources

 */

@Listeners({jaxp.library.FilePolicy.class})

public class ValidationTest {

    static final String FILE_PATH = "files/";

    /*

     DataProvider: valid xml

     */

    @DataProvider(name = "valid")

    Object[][] getValid() {

        return new Object[][]{

            {"JDK8220818a.xsd", "JDK8220818a_Valid.xml"},

            {"JDK8220818a.xsd", "JDK8220818a_Valid1.xml"},

            {"JDK8220818b.xsd", "JDK8220818b_Valid.xml"},

        };

    }

    /*

     DataProvider: invalid xml

     */

    @DataProvider(name = "invalid")

    Object[][] getInvalid() {

        return new Object[][]{

            {"JDK8220818a.xsd", "JDK8220818a_Invalid.xml"},

            {"JDK8220818b.xsd", "JDK8220818b_Invalid.xml"},

        };

    }

    /*

     DataProvider: uniqueness

     */

    @DataProvider(name = "uniqueness")

    Object[][] getUniqueData() {

        return new Object[][]{

            {"JDK8176447a.xsd", "JDK8176447a.xml"},

            {"JDK8176447b.xsd", "JDK8176447b.xml"},

        };

    }

    @Test(dataProvider = "invalid", expectedExceptions = SAXParseException.class)

    public void testValidateRefType(String xsd, String xml) throws Exception {

        validate(xsd, xml);

    }

    @Test(dataProvider = "valid")

    public void testValidateRefType1(String xsd, String xml) throws Exception {

        validate(xsd, xml);

    }

    /**

     * @bug 8176447

     * Verifies that the uniqueness constraint is checked.

     * @param xsd the XSD

     * @param xml the XML

     * @throws Exception expected when the uniqueness constraint is validated

     * correctly.

     */

    @Test(dataProvider = "uniqueness", expectedExceptions = SAXException.class)

    public void testUnique(String xsd, String xml) throws Exception {

        validate(xsd, xml);

    }

    /**

     * @bug 8068376

     * Verifies that validation performs normally with externally provided string

     * parameters.

     * @throws Exception if the test fails

     */

    @Test

    public void testJDK8068376() throws Exception {

        String xsdFile = getClass().getResource(FILE_PATH + "JDK8068376.xsd").getFile();

        String xmlFile = getClass().getResource(FILE_PATH + "JDK8068376.xml").getFile();

        String targetNamespace = getTargetNamespace(xsdFile);

        XMLFilter namespaceFilter = new XMLFilterImpl(SAXParserFactory.newDefaultNSInstance().newSAXParser().getXMLReader()) {

        @Override

        public void startElement(String uri, String localName, String qName, Attributes atts) throws SAXException {

            uri = targetNamespace; // overwriting the uri with our own choice

            super.startElement(uri, localName, qName, atts);

        }

        };

        Source xmlSource = new SAXSource(namespaceFilter, new InputSource(xmlFile));

        Source schemaSource = new StreamSource(xsdFile);

        validate(schemaSource, xmlSource);

    }

    private static String getTargetNamespace(String xsdFile) throws Exception {

        XMLStreamReader reader = XMLInputFactory.newInstance().createXMLStreamReader(new FileInputStream(xsdFile));

        while (reader.hasNext()) {

            int event = reader.next();

            // Get the root element's "targetNamespace" attribute

            if (event == XMLEvent.START_ELEMENT) {

                // validation fails before patch

                String value = reader.getAttributeValue(null, "targetNamespace"); // fails validation

                // validation passes due to a reference comparison in the original code

                // String value = "mynamespace";

                return value;

            }

        }

        return null;

    }

    private void validate(String xsd, String xml) throws Exception {

        final SchemaFactory schemaFactory = SchemaFactory.newInstance(

                XMLConstants.W3C_XML_SCHEMA_NS_URI);

        final Schema schema = schemaFactory.newSchema(

                new File(getClass().getResource(FILE_PATH + xsd).getFile()));

        final Validator validator = schema.newValidator();

        validator.validate(new StreamSource(

                new File(getClass().getResource(FILE_PATH + xml).getFile())));

    }

    private void validate(Source xsd, Source xml) throws Exception {

        final SchemaFactory schemaFactory = SchemaFactory.newInstance(

                XMLConstants.W3C_XML_SCHEMA_NS_URI);

        final Schema schema = schemaFactory.newSchema(xsd);

        final Validator validator = schema.newValidator();

        validator.validate(xml);

    }

}