moxy/org.eclipse.persistence.moxy/src/test/java/org/eclipse/persistence/testing/jaxb/jaxbcontext/JaxbTypeToSchemaTypeTestCases.java

/*
 * Copyright (c) 1998, 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:
//     Denise Smith - 2009-12-09
package org.eclipse.persistence.testing.jaxb.jaxbcontext;

import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.List;
import java.util.Map;
import java.util.Set;

import jakarta.activation.DataHandler;
import jakarta.xml.bind.annotation.XmlAttachmentRef;
import javax.xml.namespace.QName;

import org.eclipse.persistence.jaxb.JAXBContext;
import org.eclipse.persistence.jaxb.JAXBContextFactory;
import org.eclipse.persistence.jaxb.TypeMappingInfo;
import org.eclipse.persistence.jaxb.TypeMappingInfo.ElementScope;
import org.eclipse.persistence.oxm.XMLConstants;
import org.eclipse.persistence.testing.oxm.OXTestCase;

public class JaxbTypeToSchemaTypeTestCases extends OXTestCase{

    @XmlAttachmentRef
    public DataHandler attachmentRefField;

    public String stringField;

    public List<String[]> listOfStringArrayField;
    public Set<String[]> setOfStringArrayField;

    // collections
    public List<String> listOfStringsField;

    // to test nested collections
    public Set<List<String>> setOfListsField;
    public List<List<String>> listOfListsField;

    public Set<List<List<String>>> setOfListOfListOfStringsField;

    public JaxbTypeToSchemaTypeTestCases(String name) {
        super(name);
    }

    public void testClassArray() throws Exception{
        Class<?>[] classes = new Class<?>[]{String.class};
        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(classes, null);
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(1, typeMap.size());

        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(0, tmiMap.size());
    }

    public void testTypeArray() throws Exception{
        Type[] types = new Type[]{String.class};
        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(types, null, Thread.currentThread().getContextClassLoader());
        // Additionally, test refreshMetadata when bootstrapping from Type[]
        ctx.refreshMetadata();
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(1, typeMap.size());
        assertNotNull(typeMap.get(String.class));

        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(0, tmiMap.size());
    }

    public void testTypeMappingInfoArray() throws Exception{
        TypeMappingInfo tmi = new TypeMappingInfo();
        tmi.setType(String.class);

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());

        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(1, tmiMap.size());
        assertNotNull(tmiMap.get(tmi));
    }

    public void testDatahandlerTypes() throws Exception{
        TypeMappingInfo tmi = new TypeMappingInfo();
        tmi.setType(DataHandler.class);
        tmi.setElementScope(TypeMappingInfo.ElementScope.Local);
        tmi.setXmlTagName(new QName("uri1", "tmi1"));
        Annotation[] annotations = getClass().getField("attachmentRefField").getAnnotations();
        tmi.setAnnotations(annotations);

        TypeMappingInfo tmi2 = new TypeMappingInfo();
        tmi2.setElementScope(TypeMappingInfo.ElementScope.Local);
        tmi2.setXmlTagName(new QName("uri1", "tmi2"));
        tmi2.setType(DataHandler.class);

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi, tmi2};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());

        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(2, tmiMap.size());
        assertNotNull(tmiMap.get(tmi2));
        assertEquals(XMLConstants.BASE_64_BINARY_QNAME, tmiMap.get(tmi2));
        assertNotNull(tmiMap.get(tmi));
        assertEquals(XMLConstants.SWA_REF_QNAME, tmiMap.get(tmi));
    }

    public void testObject() throws Exception{
        TypeMappingInfo tmi = new TypeMappingInfo();
        tmi.setType(Object.class);

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(1, tmiMap.size());
        assertNotNull(tmiMap.get(tmi));
    }

    public void testByteArray() throws Exception{
        TypeMappingInfo tmi = new TypeMappingInfo();
        tmi.setType(byte[].class);

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(1, tmiMap.size());
        assertNotNull(tmiMap.get(tmi));
    }

    public void testStringType() throws Exception{
        TypeMappingInfo tmi = mappingInfo("stringField");

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(1, tmiMap.size());
        assertNotNull(tmiMap.get(tmi));
    }

    public void testListofStringType() throws Exception{
        TypeMappingInfo tmi = mappingInfo("listOfStringsField");

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(1, tmiMap.size());
        assertNotNull(tmiMap.get(tmi));
    }

    public void testListOfStringArrayField() throws Exception{
        TypeMappingInfo tmi1 = mappingInfo("listOfStringArrayField");
        TypeMappingInfo tmi2 = mappingInfo("setOfStringArrayField");

        TypeMappingInfo[] tmis = new TypeMappingInfo[] { tmi1, tmi2 };

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(tmis.length, tmiMap.size());
        assertNotNull(tmiMap.get(tmi1));
    }

    public void testNestedCollections() throws Exception{
        TypeMappingInfo tmi1 = mappingInfo("setOfListsField");
        TypeMappingInfo tmi2 = mappingInfo("listOfListsField");
        TypeMappingInfo[] tmis = new TypeMappingInfo[]{
                tmi1,
                tmi2,
        };

        ClassLoader tccl = Thread.currentThread().getContextClassLoader();
        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, tccl);
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(tmis.length, tmiMap.size());
        assertNotNull(tmiMap.get(tmis[0]));
    }

    private TypeMappingInfo mappingInfo(String name) throws NoSuchFieldException {
        TypeMappingInfo tmi = new TypeMappingInfo();
        tmi.setType(getClass().getField(name).getGenericType());
        return tmi;
    }

    public void testIntegerArray() throws Exception{
        TypeMappingInfo tmi = new TypeMappingInfo();
        tmi.setType(Integer[].class);

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(1, tmiMap.size());
        assertNotNull(tmiMap.get(tmi));
    }

    public void testInteger2DArray() throws Exception{
        TypeMappingInfo tmi = new TypeMappingInfo();
        tmi.setType(Integer[][].class);

        TypeMappingInfo[] tmis = new TypeMappingInfo[]{tmi};

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(tmis, null, Thread.currentThread().getContextClassLoader());
        Map<Type, QName> typeMap = ctx.getTypeToSchemaType();
        assertEquals(0, typeMap.size());
        Map<TypeMappingInfo, QName> tmiMap = ctx.getTypeMappingInfoToSchemaType();
        assertEquals(1, tmiMap.size());
        assertNotNull(tmiMap.get(tmi));
    }

    public void testSimpleTypes() throws Exception{
        TypeMappingInfo[] typesToBeBound = new TypeMappingInfo[5];

        TypeMappingInfo tmi1 = new TypeMappingInfo();
        tmi1.setType(String.class);

        TypeMappingInfo tmi2 = new TypeMappingInfo();
        tmi2.setType(int.class);

        TypeMappingInfo tmi3 = new TypeMappingInfo();
        tmi3.setType(Integer.class);

        TypeMappingInfo tmi4 = new TypeMappingInfo();
        tmi4.setType(char.class);

        TypeMappingInfo tmi5 = new TypeMappingInfo();
        tmi5.setType(Object.class);

        typesToBeBound[0] = tmi1;
        typesToBeBound[1] = tmi2;
        typesToBeBound[2] = tmi3;
        typesToBeBound[3] = tmi4;
        typesToBeBound[4] = tmi5;

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(typesToBeBound, null, Thread.currentThread().getContextClassLoader());
        Map<TypeMappingInfo, QName> types = ctx.getTypeMappingInfoToSchemaType();

        assertEquals(5, types.size());
        assertNotNull(types.get(tmi1));
        assertNotNull(types.get(tmi2));
        assertNotNull(types.get(tmi3));
        assertNotNull(types.get(tmi4));
        assertNotNull(types.get(tmi5));
    }

    public void testInnerClass() throws Exception{
        TypeMappingInfo tmi1 = new TypeMappingInfo();
        tmi1.setXmlTagName(new QName("test", "theRoot"));
        tmi1.setElementScope(ElementScope.Global);
        tmi1.setType(MyInnerClass.class);

        TypeMappingInfo[] typesToBeBound = new TypeMappingInfo[1];
        typesToBeBound[0] = tmi1;

        JAXBContext ctx = (JAXBContext) JAXBContextFactory.createContext(typesToBeBound, null, Thread.currentThread().getContextClassLoader());
        Map<TypeMappingInfo, QName> types = ctx.getTypeMappingInfoToSchemaType();

        assertEquals(1, types.size());
        assertNotNull(types.get(tmi1));

    }

    public static class MyInnerClass{
        public String someString;
    }
}