com.oneandone.iocunit.analyzer.SetupCreator Maven / Gradle / Ivy

Go to download
package com.oneandone.iocunit.analyzer;

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;

import javax.annotation.Priority;
import javax.enterprise.inject.spi.Extension;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.oneandone.cdi.weldstarter.WeldSetupClass;
import com.oneandone.cdi.weldstarter.spi.TestExtensionService;

public class SetupCreator {
    static Logger logger = LoggerFactory.getLogger(SetupCreator.class);
    private final Configuration configuration;

    public SetupCreator(Configuration configuration) {
        this.configuration = configuration;
    }

    private Collection findExtensions() {
        List result = new ArrayList<>();
        for (TestExtensionService testExtensionService : configuration.testerExtensionsConfigsFinder.testExtensionServices) {
            result.addAll(testExtensionService.getExtensions());
        }
        return result;
    }

    private void handleWeldExtensions(final Method method, final WeldSetupClass weldSetup) {
        try {
            for (Class extensionClass : configuration.getElseClasses().extensionClasses) {
                if(configuration.excludedExtensions != null && configuration.excludedExtensions.contains(extensionClass)) {
                    continue;
                }
                if(extensionClass.getName().contains(".ProducerConfigExtension")) {
                    Constructor constructor =
                            extensionClass.getConstructor(Method.class);
                    Extension producerConfig = constructor.newInstance(method);
                    weldSetup.addExtensionObject(producerConfig);
                }
                else {
                    weldSetup.addExtensionObject(extensionClass.newInstance());
                }
            }
            for (Extension e : configuration.getElseClasses().extensionObjects) {
                Class extensionClass = e.getClass();
                final Constructor[] declaredConstructors = extensionClass.getDeclaredConstructors();
                if(declaredConstructors.length == 1 && declaredConstructors[0].getParameters().length == 0) {
                    weldSetup.addExtensionObject(extensionClass.newInstance());
                }
                else {
                    weldSetup.addExtensionObject(e);
                }
            }
        } catch (Exception e1) {
            throw new RuntimeException(e1);
        }
    }

    public WeldSetupClass buildWeldSetup(Method method) {
        WeldSetupClass weldSetup = new WeldSetupClass();
        weldSetup.setBeanClasses(configuration.getObligatory());
        weldSetup.setAlternativeClasses(configuration.getEnabledAlternatives());
        weldSetup.setEnabledAlternativeStereotypes(configuration.getElseClasses().foundAlternativeStereotypes);
        if(logger.isTraceEnabled()) {
            for (Class i : configuration.getElseClasses().decorators) {
                logger.trace("buildWeldSetup Decorator:   {}", i);
            }
            for (Class i : configuration.getElseClasses().interceptors) {
                logger.trace("buildWeldSetup Interceptor: {}", i);
            }
        }
        weldSetup.setEnabledDecorators(configuration.getElseClasses().decorators);
        List> interceptorsToEnable = configuration.getElseClasses().interceptors;
        if (!WeldSetupClass.isWeld1()) {
            try {
                interceptorsToEnable = configuration.getElseClasses().interceptors.stream()
                        .filter(c -> c.getAnnotation(Priority.class) == null)
                        .collect(Collectors.toList());
            } catch (NoClassDefFoundError e) {
                ;
            }
        }
        weldSetup.setEnabledInterceptors(interceptorsToEnable);
        handleWeldExtensions(method, weldSetup);
        for (Extension e : findExtensions()) {
            weldSetup.addExtensionObject(e);
        }
        return weldSetup;
    }
}