moxy/org.eclipse.persistence.moxy/src/test/java/org/eclipse/persistence/testing/jaxb/xmlmarshaller/HandleListenerExceptionsTestCases.java

/*
 * Copyright (c) 2013, 2021 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0,
 * or the Eclipse Distribution License v. 1.0 which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
 */

// Contributors:
//     Blaise Doughan - 2.4.2 - initial implementation
package org.eclipse.persistence.testing.jaxb.xmlmarshaller;

import java.io.StringReader;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;

import jakarta.xml.bind.JAXBContext;
import jakarta.xml.bind.Marshaller;
import jakarta.xml.bind.UnmarshalException;
import jakarta.xml.bind.Unmarshaller;
import jakarta.xml.bind.ValidationEvent;
import jakarta.xml.bind.ValidationEventHandler;
import jakarta.xml.bind.annotation.XmlRootElement;

import org.eclipse.persistence.exceptions.XMLMarshalException;
import org.eclipse.persistence.jaxb.JAXBContextFactory;
import org.xml.sax.Locator;
import org.xml.sax.SAXParseException;

import junit.framework.TestCase;

public class HandleListenerExceptionsTestCases extends TestCase {

    private static String AFTER_MARSHAL = "afterMarshal";
    private static String AFTER_UNMARSHAL = "afterUnmarshal";
    private static String BEFORE_MARSHAL = "beforeMarshal";
    private static String BEFORE_UNMARSHAL = "beforeUnmarshal";
    private static String XML = "<foo><bar/></foo>";

    private JAXBContext jc;

    @Override
    protected void setUp() throws Exception {
        jc = JAXBContextFactory.createContext(new Class<?>[] {Foo.class}, null);
        //jc = JAXBContext.newInstance(Foo.class);
    }

    public void testUnmarshal() throws Exception {
        Unmarshaller unmarshaller = jc.createUnmarshaller();
        TestValidationEventHandler veh = new TestValidationEventHandler();
        unmarshaller.setEventHandler(veh);

        StringReader reader = new StringReader(XML);
        unmarshaller.unmarshal(reader);

        assertEquals(2, veh.validationEvents.size());

        ValidationEvent veh1 = veh.validationEvents.get(0);
        assertEquals(ValidationEvent.FATAL_ERROR, veh1.getSeverity());
        assertTrue(veh1.getLinkedException() instanceof XMLMarshalException);
        assertEquals(BEFORE_UNMARSHAL, veh1.getLinkedException().getCause().getCause().getMessage());

        ValidationEvent veh2 = veh.validationEvents.get(1);
        assertEquals(ValidationEvent.FATAL_ERROR, veh2.getSeverity());
        assertTrue(veh2.getLinkedException() instanceof XMLMarshalException);
        assertEquals(AFTER_UNMARSHAL, veh2.getLinkedException().getCause().getCause().getMessage());
    }

    public void testMarshal() throws Exception {
        Marshaller marshaller = jc.createMarshaller();
        TestValidationEventHandler veh = new TestValidationEventHandler();
        marshaller.setEventHandler(veh);

        Foo foo = new Foo();
        foo.bar = new Bar();

        marshaller.marshal(foo, new StringWriter());

        assertEquals(2, veh.validationEvents.size());

        ValidationEvent veh1 = veh.validationEvents.get(0);
        assertEquals(ValidationEvent.FATAL_ERROR, veh1.getSeverity());
        assertTrue(veh1.getLinkedException() instanceof XMLMarshalException);
        assertEquals(BEFORE_MARSHAL, veh1.getLinkedException().getCause().getCause().getMessage());

        ValidationEvent veh2 = veh.validationEvents.get(1);
        assertEquals(ValidationEvent.FATAL_ERROR, veh2.getSeverity());
        assertTrue(veh2.getLinkedException() instanceof XMLMarshalException);
        assertEquals(AFTER_MARSHAL, veh2.getLinkedException().getCause().getCause().getMessage());

    }

    @XmlRootElement
    public static class Foo {

        public Bar bar;

    }

    public static class Bar {

        private void afterMarshal(Marshaller marshaller) {
            throw new RuntimeException(AFTER_MARSHAL);
        }

        private void afterUnmarshal(Unmarshaller unmarshaller, Object object) {
            throw new RuntimeException(AFTER_UNMARSHAL);
        }

        private void beforeMarshal(Marshaller marshaller) {
            throw new RuntimeException(BEFORE_MARSHAL);
        }

        private void beforeUnmarshal(Unmarshaller unmarshaller, Object object) {
            throw new RuntimeException(BEFORE_UNMARSHAL);
        }

    }

    public static class TestValidationEventHandler implements ValidationEventHandler {

        List<ValidationEvent> validationEvents = new ArrayList<ValidationEvent>(2);

        public List<ValidationEvent> getValidationEvents() {
            return validationEvents;
        }

        @Override
        public boolean handleEvent(ValidationEvent validationEvent) {
            this.validationEvents.add(validationEvent);
            return true;
        }

    }

}