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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of evosuite-runtime Show documentation
There is a newer version: 1.0.6
/**
 * 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.Reflection;
import org.evosuite.runtime.util.Inputs;
import org.evosuite.runtime.util.ReflectionUtils;

import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * Class used to determine which fields should be injected
 * in a "general" way: ie, instance object is created in the
 * test as any other input parameter. This is different
 * from "special" fields that need to be adhoc initialized
 * by EvoSuite, like EntityManager that needs to be connected
 * to an actual database
 *
 * Created by Andrea Arcuri on 13/07/15.
 */
public class GeneralInjection {

    private final Map, List> cache;

    /**
     *  Caches of injected fields that are treated specially
     */
    private final Map, InjectionCache> specials;

    public GeneralInjection(InjectionCache... specials) throws IllegalArgumentException{
        this.cache = new LinkedHashMap<>();
        this.specials = new LinkedHashMap<>();
        for(InjectionCache ic : specials){
            Class fieldClass = ic.getFieldClass();
            if(this.specials.containsKey(fieldClass)){
                throw new IllegalArgumentException("Field of type "+
                        fieldClass + "has more than one cache");
            }
            this.specials.put(fieldClass , ic);
        }
    }

    public void reset(){
        cache.clear();
    }

    public List getFieldsToInject(Class klass) throws IllegalArgumentException{
        Inputs.checkNull(klass);

        List fields = cache.get(klass);
        if(fields==null){
            fields = new ArrayList<>();

            for(Field f : Reflection.getDeclaredFields(klass)){
                Class fieldClass = f.getType();
                if(specials.containsKey(fieldClass)){
                    //this is already handled, so skip it
                    continue;
                }
                for(Annotation annotation : Reflection.getDeclaredAnnotations(f)){
                    Class annotationClass = annotation.annotationType();
                    if(InjectionList.isValidForInjection(annotationClass)){
                        fields.add(f);
                        break;
                    }
                }
            }

            fields.sort((a,b) -> a.getName().compareTo(b.getName()));
            cache.put(klass, fields);
        }

        return new ArrayList<>(fields);
    }
}