/*******************************************************************************
 * Copyright (c) 1998, 2009 Oracle. All rights reserved.
 * This program and the accompanying materials are made available under the 
 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 
 * which accompanies this distribution. 
 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 * and the Eclipse Distribution License is available at 
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * Contributors:
 *     Oracle - initial API and implementation from Oracle TopLink
 ******************************************************************************/  
package org.eclipse.persistence.testing.jaxb.jaxbelement;

import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.StringReader;
import java.io.StringWriter;

import javax.xml.bind.JAXBElement;
import javax.xml.stream.XMLStreamReader;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;

import org.w3c.dom.Document;
import org.xml.sax.InputSource;

import org.eclipse.persistence.platform.xml.SAXDocumentBuilder;
import org.eclipse.persistence.testing.jaxb.JAXBTestCases;
import org.eclipse.persistence.testing.oxm.xmlroot.Person;

/**
 * This class should not be used directly - only it's subclasses should
 * be setup and used for testing.
 */
public class JAXBElementTestCases extends JAXBTestCases {
	protected final static String CONTROL_PERSON_NAME = "Joe Smith";
	protected final static String CONTROL_NAMESPACE_URI = "test";	
	protected String resourceName;
	protected Class target;

    public JAXBElementTestCases(String name) throws Exception {
        super(name);
        Class[] classes = new Class[1];
        classes[0] = Person.class;
        setClasses(classes);
    }

    /**
     * Satisfy the abstract method declaration in JAXBTestCases
     */
    protected Object getControlObject() {
		return null;
	}
    
    protected void setControlDocument(String xmlResource) throws Exception {
    	super.setControlDocument(xmlResource);
    	resourceName = xmlResource;  
    }
    
    protected void setTargetClass(Class targetCls) {
    	target = targetCls;
    }
    
    public void testXMLToObjectFromDocument() throws Exception {
        InputStream instream = ClassLoader.getSystemResourceAsStream(resourceName);
        InputSource inputSource = new InputSource(instream);
        Document testDocument = parser.parse(inputSource);
        Object testObject = getJAXBUnmarshaller().unmarshal(testDocument, target);
        instream.close();
        xmlToObjectTest(testObject);
    }
    
    public void testXMLToObjectFromSource() throws Exception {
        InputStream instream = ClassLoader.getSystemResourceAsStream(resourceName);
        StreamSource source = new StreamSource(instream);
        Object testObject = getJAXBUnmarshaller().unmarshal(source, target);
        instream.close();
        xmlToObjectTest(testObject);
    }

    public void testXMLToObjectFromXMLStreamReader() throws Exception {
        if(null != XML_INPUT_FACTORY) {
            InputStream instream = ClassLoader.getSystemResourceAsStream(resourceName);
            XMLStreamReader xmlStreamReader = XML_INPUT_FACTORY.createXMLStreamReader(instream);
            Object testObject = jaxbUnmarshaller.unmarshal(xmlStreamReader, target);
            instream.close();
            xmlToObjectTest(testObject);
        }
    }

    public void xmlToObjectTest(Object testObject) throws Exception {
    	JAXBElement testObj = (JAXBElement ) testObject;
        JAXBElement controlObj = (JAXBElement) getReadControlObject();

        log("\n**testXMLDocumentToObject**");
        log("Expected:");
        log("uri=" + controlObj.getName().getNamespaceURI() +  
        	", prefix=" + controlObj.getName().getPrefix() + 
            ", localname=" + controlObj.getName().getLocalPart() + 
            ", object=" + controlObj.getValue());
        log("Actual:");
        log("uri=" + testObj.getName().getNamespaceURI() +  
            ", prefix=" + testObj.getName().getPrefix() + 
        	", localname=" + testObj.getName().getLocalPart() + 
        	", object=" + testObj.getValue());

        
        this.assertEquals(controlObj.getName().getNamespaceURI(), testObj.getName().getNamespaceURI());
        this.assertEquals(controlObj.getName().getPrefix(), testObj.getName().getPrefix());
        this.assertEquals(controlObj.getName().getLocalPart(), testObj.getName().getLocalPart());
        this.assertEquals(controlObj.getValue(), testObj.getValue());
    }

    public void testObjectToStringWriter() throws Exception {
        StringWriter writer = new StringWriter();
        getJAXBMarshaller().marshal(getWriteControlObject(), writer);
        StringReader reader = new StringReader(writer.toString());
        InputSource inputSource = new InputSource(reader);
        Document testDocument = parser.parse(inputSource);
        writer.close();
        reader.close();
        objectToXMLDocumentTest(testDocument);
    }
    
    public void testObjectToByteArrayOutputStream() throws Exception {
    	ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
    	getJAXBMarshaller().marshal(getWriteControlObject(), outputStream);
        StringReader reader = new StringReader(outputStream.toString());
        InputSource inputSource = new InputSource(reader);
        Document testDocument = parser.parse(inputSource);
        outputStream.close();
        reader.close();
        objectToXMLDocumentTest(testDocument);
    }
    
    public void testObjectToStreamResult() throws Exception {
        StringWriter writer = new StringWriter();
    	StreamResult result = new StreamResult(writer);
    	getJAXBMarshaller().marshal(getWriteControlObject(), result);
        StringReader reader = new StringReader(writer.toString());
        InputSource inputSource = new InputSource(reader);
        Document testDocument = parser.parse(inputSource);
        writer.close();
        reader.close();
        objectToXMLDocumentTest(testDocument);
    }
    
    public void testObjectToXMLDocument() throws Exception {
    	Document testDocument = new SAXDocumentBuilder().getInitializedDocument();
    	getJAXBMarshaller().marshal(getWriteControlObject(), testDocument);
        objectToXMLDocumentTest(testDocument);
    }
    
    public void testObjectToContentHandler() throws Exception {
        SAXDocumentBuilder builder = new SAXDocumentBuilder();
        getJAXBMarshaller().marshal(getWriteControlObject(), builder);
        Document controlDocument = getWriteControlDocument();
        Document testDocument = builder.getDocument();
        log("**testObjectToXMLDocument**");
        log("Expected:");
        log(controlDocument);
        log("\nActual:");
        log(testDocument);

        assertXMLIdentical(controlDocument, testDocument);
    }

    // TODO:  add support for StAX
    public void testObjectToStreamWriter() throws Exception {}

    public void objectToXMLDocumentTest(Document testDocument) throws Exception {
        log("**testObjectToXMLDocument**");
        log("Expected:");
        log(getWriteControlDocument());
        log("\nActual:");
        log(testDocument);

        assertXMLIdentical(getWriteControlDocument(), testDocument);
    }

    // THESE TESTS DO NOT APPLY
    public void testXMLToObjectFromInputStream() throws Exception {}
    public void testXMLToObjectFromURL() throws Exception {}
    public void testObjectToXMLStringWriter() throws Exception {}
}
