1 /*
   2  * Copyright (c) 2014, 2016, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 package validation;
  25 
  26 import java.io.File;
  27 
  28 import javax.xml.XMLConstants;
  29 import javax.xml.parsers.DocumentBuilder;
  30 import javax.xml.parsers.DocumentBuilderFactory;
  31 import javax.xml.validation.Schema;
  32 import javax.xml.validation.SchemaFactory;
  33 
  34 import org.testng.Assert;
  35 import org.testng.annotations.Listeners;
  36 import org.testng.annotations.Test;
  37 import org.xml.sax.ErrorHandler;
  38 import org.xml.sax.SAXException;
  39 import org.xml.sax.SAXParseException;
  40 
  41 /*
  42  * @summary Test maxOccurs validation.
  43  */
  44 @Listeners({jaxp.library.FilePolicy.class})
  45 public class IssueTracker30 {
  46 
  47     boolean errorFound;
  48 
  49     @Test
  50     public void testMaxOccursErrorNoOptimization() throws Exception {
  51 
  52         // Parsing this document should result in an error
  53         try {
  54                 if (!testMaxOccurs("IssueTracker30_occurs-error.xml", "IssueTracker30_occurs.xsd"))
  55                         Assert.fail("Expected validation error not reported");
  56         } catch (SAXException ex) {
  57             Assert.fail(ex.getMessage(), ex);
  58         }
  59     }
  60 
  61     @Test
  62     public void testMaxOccursOKNoOptimization() throws Exception {
  63 
  64         // Parsing this document should *not* result in an error
  65         try {
  66                 if (testMaxOccurs("IssueTracker30_occurs-ok.xml", "IssueTracker30_occurs.xsd"))
  67                         Assert.fail("Unexpected validation error reported");
  68         } catch (SAXException ex) {
  69             Assert.fail(ex.getMessage(), ex);
  70         }
  71     }
  72 
  73     @Test
  74     public void testMaxOccursErrorOptimization() throws Exception {
  75 
  76         // Parsing this document should result in an error
  77         try {
  78                 if (!testMaxOccurs("IssueTracker30_occurs-optimize-error.xml", "IssueTracker30_occurs-optimize.xsd"))
  79                         Assert.fail("Expected validation error not reported");
  80         } catch (SAXException ex) {
  81             Assert.fail(ex.getMessage(), ex);
  82         }
  83     }
  84 
  85     @Test
  86     public void testMaxOccursOKOptimization() throws Exception {
  87 
  88         // Parsing this document should *not* result in an error
  89         try {
  90                 if (testMaxOccurs("IssueTracker30_occurs-optimize-ok.xml", "IssueTracker30_occurs-optimize.xsd"))
  91                         Assert.fail("Unexpected validation error reported");
  92         } catch (SAXException ex) {
  93             Assert.fail(ex.getMessage(), ex);
  94         }
  95     }
  96 
  97 
  98     private boolean testMaxOccurs(String xmlFileName, String xsdFileName) throws Exception {
  99         File xmlFile = new File(getClass().getResource(xmlFileName).getFile());
 100 
 101         SchemaFactory factory = SchemaFactory.newInstance(XMLConstants.W3C_XML_SCHEMA_NS_URI);
 102         Schema schema = factory.newSchema(new File(getClass().getResource(xsdFileName).getFile()));
 103 
 104         DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
 105         dbf.setNamespaceAware(true);
 106         dbf.setSchema(schema);
 107 
 108         DocumentBuilder documentBuilder = dbf.newDocumentBuilder();
 109         documentBuilder.setErrorHandler(new ErrorHandler() {
 110             public void error(SAXParseException e) throws SAXException {
 111                 System.out.println("Error: " + e.getMessage());
 112                 errorFound = true;
 113             }
 114 
 115             public void fatalError(SAXParseException e) throws SAXException {
 116                 System.out.println("Fatal error: " + e.getMessage());
 117                 throw e;
 118             }
 119 
 120             public void warning(SAXParseException e) throws SAXException {
 121                 System.out.println("Warning: " + e.getMessage());
 122             }
 123         });
 124 
 125         errorFound = false;
 126         documentBuilder.parse(xmlFile);
 127         return errorFound;
 128     }
 129 
 130 }