/*
    * 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.customext.attrib;
  
  import junit.framework.TestCase;
  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.w3c.dom.Document;
  import tests.Resources;
  
  import javax.xml.parsers.DocumentBuilderFactory;
  import java.io.ByteArrayInputStream;
  import java.io.ByteArrayOutputStream;
  import java.util.Iterator;
  import java.util.Map;
  
  /**
   * Test class to do a full parsing run with the extensions
   */
  public class CustomExtensionSerializerTest extends TestCase {
  
      public void testSerialization() throws Exception {
          //set the system property for the custom extension registry
          System.setProperty(Constants.SystemConstants.EXTENSION_REGISTRY_KEY,
                  CustomExtensionRegistry.class.getName());
  
          //create a DOM document
          DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
          documentBuilderFactory.setNamespaceAware(true);
  
          // Skip test in JDK1.4 as it uses crimson parser and an old DOM implementation
          if (documentBuilderFactory.getClass().toString().indexOf("crimson") != -1) {
              return;
          }
          Document doc1 = documentBuilderFactory.newDocumentBuilder().
                  parse(Resources.asURI("/external/externalAnnotations.xsd"));
  
          XmlSchemaCollection schemaCol = new XmlSchemaCollection();
          XmlSchema schema = schemaCol.read(doc1,null);
          assertNotNull(schema);
  
          //now serialize it to a byte stream
          //and build a new document out of it
          ByteArrayOutputStream baos = new ByteArrayOutputStream();
          schema.write(baos);
  
  
          Document doc2 = documentBuilderFactory.newDocumentBuilder().
                  parse(new ByteArrayInputStream(baos.toByteArray()));
  
          schemaCol = new XmlSchemaCollection();
          schema = schemaCol.read(doc2,null);
          assertNotNull(schema);
  
          // get the elements and check whether their annotations are properly
          // populated
          Iterator values = schema.getElements().getValues();
          while (values.hasNext()) {
              XmlSchemaElement elt =  (XmlSchemaElement) values.next();
              assertNotNull(elt);
              Map metaInfoMap = elt.getMetaInfoMap();
              assertNotNull(metaInfoMap);
  
              CustomAttribute customAttrib = (CustomAttribute)metaInfoMap.get(CustomAttribute.CUSTOM_ATTRIBUTE_QNAME);
              assertNotNull(customAttrib);
  
          }
  
  
          //remove our system property
          System.getProperties().remove(Constants.SystemConstants.EXTENSION_REGISTRY_KEY);
  
      }
  
  }