org.kubek2k.springockito.annotations.SpringockitoContextLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of springockito-annotations Show documentation
There is a newer version: 1.0.9
Show newest version
package org.kubek2k.springockito.annotations;

import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.support.GenericApplicationContext;
import org.springframework.test.context.support.GenericXmlContextLoader;

import java.util.*;

public class SpringockitoContextLoader extends GenericXmlContextLoader {

    /**
     * Have to be sorted since we want a nice caching of customized application contexts
     */
    private Map> mockedBeans
            = new TreeMap>();
    private Set spiedBeans = new TreeSet();

    private DesiredMockitoBeansFinder mockedBeansFinder = new DesiredMockitoBeansFinder();

    private MockitoBeansDefiner mockitoBeansDefiner = new MockitoBeansDefiner();
    private MockitoSpiesDefiner mockitoSpiesDefiner = new MockitoSpiesDefiner();

    @Override
    protected void customizeContext(GenericApplicationContext context) {
        super.customizeContext(context);
        registerMocks(context, mockedBeans);
        registerSpies(context, spiedBeans);
    }

    private void registerMocks(GenericApplicationContext context,
                               Map> mockedBeans) {
        for (Map.Entry> beanEntry : this.mockedBeans.entrySet()) {
            DesiredMockitoBeansFinder.MockProperties mockProperties = beanEntry.getValue();
            ReplaceWithMock replaceWithMockAnnotation = mockProperties.getAnnotationInstance();
            context.registerBeanDefinition(beanEntry.getKey(),
                    mockitoBeansDefiner.createMockFactoryBeanDefinition(mockProperties.getMockClass(),
                            replaceWithMockAnnotation.extraInterfaces(),
                            replaceWithMockAnnotation.name(),
                            replaceWithMockAnnotation.defaultAnswer()
                    ));
        }
    }

    private void registerSpies(GenericApplicationContext context, Set spiedBeanNames) {
        for (String beanName : spiedBeanNames) {
            BeanDefinition beanDefinition = context.getBeanDefinition(beanName);
            String wrappedBeanName = beanName + "$$WRAPPED_WITH_SPY";
            context.registerBeanDefinition(wrappedBeanName, beanDefinition);
            context.registerBeanDefinition(beanName, mockitoSpiesDefiner.createSpyDefinition(wrappedBeanName));
        }
    }

    private void defineMocksAndSpies(Class clazz) {
        this.mockedBeans.putAll(mockedBeansFinder.findMockedBeans(clazz));
        this.spiedBeans.addAll(mockedBeansFinder.findSpiedBeans(clazz));
    }

    private List generateLocationForMocksAndSpies() {
        List result = new ArrayList(mockedBeans.size() + spiedBeans.size());

        for (Map.Entry> mockDefinitionEntry : mockedBeans.entrySet()) {
            result.add("classpath*:/mock-" + mockDefinitionEntry.getKey() + "=" + mockDefinitionEntry.getValue().getClasspathRepresentation());
        }

        for (String spiedBean : spiedBeans) {
            result.add("classpath*:/spy-" + spiedBean);
        }

        return result;
    }

    // I know I could use commons or sth but I don't want to introduce more deps than are actually really really needed
    private  List merge(List list1, List list2) {
        List result = new ArrayList(list1.size() + list2.size());

        result.addAll(list1);
        result.addAll(list2);

        return result;
    }

    private String[] addFakeLocationsOfBeansAndSpies(String[] locations) {
        List locationOfMocksAndSpies = generateLocationForMocksAndSpies();
        return merge(Arrays.asList(locations), locationOfMocksAndSpies)
                .toArray(new String[locations.length + locationOfMocksAndSpies.size()]);
    }

    @Override
    protected String[] generateDefaultLocations(Class clazz) {
        String[] resultingLocations = super.generateDefaultLocations(clazz);

        defineMocksAndSpies(clazz);

        return addFakeLocationsOfBeansAndSpies(resultingLocations);

    }

    @Override
    protected String[] modifyLocations(Class clazz, String... passedLocations) {
        String[] resultingLocations = super.modifyLocations(clazz, passedLocations);

        defineMocksAndSpies(clazz);

        return addFakeLocationsOfBeansAndSpies(resultingLocations);
    }
}