All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.evosuite.runtime.javaee.injection.Injector Maven / Gradle / Ivy

There is a newer version: 1.0.6
Show newest version
/**
 * Copyright (C) 2010-2017 Gordon Fraser, Andrea Arcuri and EvoSuite
 * contributors
 *
 * This file is part of EvoSuite.
 *
 * EvoSuite is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3.0 of the License, or
 * (at your option) any later version.
 *
 * EvoSuite is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with EvoSuite. If not, see .
 */
package org.evosuite.runtime.javaee.injection;

import org.evosuite.runtime.FalsePositiveException;
import org.evosuite.runtime.PrivateAccess;
import org.evosuite.runtime.annotation.BoundInputVariable;
import org.evosuite.runtime.annotation.Constraints;
import org.evosuite.runtime.annotation.EvoSuiteExclude;
import org.evosuite.runtime.javaee.db.DBManager;
import org.evosuite.runtime.javaee.javax.enterprise.event.EvoEvent;
import org.evosuite.runtime.javaee.javax.transaction.EvoUserTransaction;
import org.evosuite.runtime.util.Inputs;
import org.evosuite.runtime.util.ReflectionUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.annotation.PostConstruct;
import javax.enterprise.event.Event;
import javax.inject.Inject;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.PersistenceContext;
import javax.persistence.PersistenceUnit;
import javax.transaction.UserTransaction;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;


/**
 * Class used to inject fields into tagged JavaEE objects
 *
 * Created by Andrea Arcuri on 30/05/15.
 */
public class Injector {

    private static final Logger logger = LoggerFactory.getLogger(Injector.class);

    /*
        Note: these fields are static because we call the static methods of this class
        directly in the JUnit tests.
        At any rate, they are just caches.
     */

    /**
     * Key -> class name,
     * Value -> @PostConstruct method, or null if none
     */
    private static final Map, Method> postConstructCache = new LinkedHashMap<>();


    private static final InjectionCache entityManagerCache =
            new InjectionCache(EntityManager.class, Inject.class, PersistenceContext.class);

    private static final InjectionCache entityManagerFactoryCache =
            new InjectionCache(EntityManagerFactory.class, Inject.class, PersistenceUnit.class);


    private static final InjectionCache userTransactionCache =
            new InjectionCache(UserTransaction.class, Inject.class);

    private static final InjectionCache eventCache =
            new InjectionCache(Event.class, Inject.class);

    //this should be initialized with all the caches declared above
    private static final GeneralInjection generalInjection =
            new GeneralInjection(entityManagerCache, entityManagerFactoryCache,
                    userTransactionCache, eventCache);


    @EvoSuiteExclude
    public static void reset(){
        generalInjection.reset();
        postConstructCache.clear();
    }

    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public  static  void inject(@BoundInputVariable(initializer = true, atMostOnceWithSameParameters = true) T instance,
                                   Class klass, String fieldName, Object value)
            throws IllegalArgumentException {

        try {
            PrivateAccess.setVariable(klass, instance, fieldName, value, InjectionList.getList());
        } catch (FalsePositiveException e){
            /*
                do not abort the test if failed to add dependency injection.
                This is quite different from private access: a PA does have effect
                on test, otherwise minimization would had removed them.
                But a dependency injection might have no impact whatsoever
             */
        }
    }

    @EvoSuiteExclude
    public static List getAllFieldsToInject(Class klass) {
        List fields = getGeneralFieldsToInject(klass);
        if(hasEntityManager(klass)) {fields.add(entityManagerCache.getField(klass));}
        if(hasEntityManagerFactory(klass)) {fields.add(entityManagerFactoryCache.getField(klass));}
        if(hasUserTransaction(klass)) {fields.add(userTransactionCache.getField(klass));}
        if(hasEvent(klass)) {fields.add(eventCache.getField(klass));}

        return fields;
    }

    @EvoSuiteExclude
    public static List getGeneralFieldsToInject(Class klass) {
        return generalInjection.getFieldsToInject(klass);
    }

    /**
     * Check if all beans that need injection have been properly injected
     *
     * @param instance
     * @param clazz
     * @param 
     * @throws FalsePositiveException
     * @throws IllegalArgumentException
     */
    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public static  void validateBean(@BoundInputVariable(initializer = true, atMostOnceWithSameParameters = true) T instance, Class clazz)
            throws FalsePositiveException, IllegalArgumentException{

        Inputs.checkNull(instance, clazz);


        for(Field f : getAllFieldsToInject(clazz)) {
            f.setAccessible(true);
            try {
                Object obj = f.get(instance);
                if(obj == null){
                    throw new FalsePositiveException("Missing dependency injection for field "+f.getName()+" in class "+clazz.getName());
                }

                //it might be a bean with its own dependency injections.
                //but those should be handled in its instantiation

            } catch (IllegalAccessException e) {
                logger.warn(e.toString());//shouldn't really happen
            }
        }

        Class parent = clazz.getSuperclass();
        if(parent != null && !parent.equals(Object.class)) {
            validateBean(instance, parent);
        }
    }


    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public static  void injectEntityManager(@BoundInputVariable(initializer = true, atMostOnceWithSameParameters = true) T instance, Class clazz)
            throws IllegalArgumentException{

        Inputs.checkNull(instance,clazz);

        String field = entityManagerCache.getFieldName(clazz);
        assert field != null;

        inject(instance, clazz, field, DBManager.getInstance().getCurrentEntityManager());
    }


    @EvoSuiteExclude
    public static boolean hasEntityManager( Class klass) throws IllegalArgumentException{
        Inputs.checkNull(klass);
        return entityManagerCache.hasField(klass);
    }


    /*
        TODO: due to limitations in EvoSuite, in the injectors as input we have to use Class
        instead of Class
     */

    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public static  void injectEntityManagerFactory(@BoundInputVariable(initializer = true, atMostOnceWithSameParameters = true) T instance, Class clazz)
            throws IllegalArgumentException {

        Inputs.checkNull(instance,clazz);

        String field = entityManagerFactoryCache.getFieldName(clazz);
        assert field != null;

        inject(instance, clazz, field, DBManager.getInstance().getDefaultFactory());
    }


    @EvoSuiteExclude
    public static boolean hasEntityManagerFactory( Class klass) throws IllegalArgumentException{
        Inputs.checkNull(klass);
        return entityManagerFactoryCache.hasField(klass);
    }


    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public static  void injectUserTransaction(@BoundInputVariable(initializer = true, atMostOnceWithSameParameters = true) T instance, Class clazz)
        throws IllegalArgumentException {

        Inputs.checkNull(instance,clazz);

        String field = userTransactionCache.getFieldName(clazz);
        assert field != null;

        inject(instance, clazz, field, new EvoUserTransaction()); //TODO this will likely need to change in the future
    }


    @EvoSuiteExclude
    public static boolean hasUserTransaction( Class klass) throws IllegalArgumentException {
        Inputs.checkNull(klass);
        return userTransactionCache.hasField(klass);
    }

    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public static  void injectEvent(@BoundInputVariable(initializer = true, atMostOnceWithSameParameters = true) T instance, Class clazz)
            throws IllegalArgumentException{

        Inputs.checkNull(instance, clazz);

        String field = eventCache.getFieldName(clazz);
        assert field != null;

        inject(instance, clazz, field, new EvoEvent()); //TODO this will likely need to change in the future
    }


    @EvoSuiteExclude
    public static boolean hasEvent( Class klass) throws IllegalArgumentException {
        Inputs.checkNull(klass);
        return eventCache.hasField(klass);
    }

    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public static void executePostConstruct(
            @BoundInputVariable(initializer = true, atMostOnce = true) Object instance) throws IllegalArgumentException {

        Inputs.checkNull(instance);
        executePostConstruct(instance, instance.getClass());
    }

    /**
     * Executed the method annotated with @PostConstruct
     *
     * @param instance
     */
    @Constraints(noNullInputs = true, notMutable = true, noDirectInsertion = true)
    public static void executePostConstruct(
            @BoundInputVariable(initializer = true, atMostOnceWithSameParameters = true) Object instance, Class clazz) throws IllegalArgumentException{

        Inputs.checkNull(instance, clazz);
        if(!clazz.isAssignableFrom(instance.getClass())) {
            throw new IllegalArgumentException("Class "+clazz+" is not assignable from "+instance.getClass());
        }
        if(!hasPostConstruct(clazz)) {
            throw new IllegalArgumentException("The class "+clazz.getName()+" does not have a @PostConstruct");
        }

        Method m = postConstructCache.get(clazz);
        assert m != null;

        try {
            m.invoke(instance);
        } catch (IllegalAccessException e) {
            //should never happen
            logger.error(e.toString());
        } catch (InvocationTargetException e) {
            throw new RuntimeException("Failed to execute @PostConstructor in "+clazz.getName(),e);
        }
    }

    @EvoSuiteExclude
    public static boolean hasPostConstruct(Class clazz) {

        if(!postConstructCache.containsKey(clazz)) {
            Method pc = null;
            outer : for(Method m : ReflectionUtils.getDeclaredMethods(clazz)) {
                for(Annotation annotation : ReflectionUtils.getDeclaredAnnotations(m)) {
                    if(annotation instanceof PostConstruct){
                        pc = m;
                        pc.setAccessible(true);
                        break outer;
                    }
                }
            }
            postConstructCache.put(clazz,pc); //note: it can be null
        }

        Method m = postConstructCache.get(clazz);
        return m != null;
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy