de.hilling.junit.cdi.ContextControlWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cdi-test-core Show documentation
implementation of cdi-test
The newest version!
package de.hilling.junit.cdi;

import jakarta.enterprise.inject.spi.Bean;

import static java.util.logging.Level.INFO;

import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.List;
import java.util.ServiceLoader;
import java.util.Set;
import java.util.logging.Logger;

import org.jboss.weld.bootstrap.api.Service;
import org.jboss.weld.construction.api.WeldCreationalContext;
import org.jboss.weld.environment.se.Weld;
import org.jboss.weld.environment.se.WeldContainer;
import org.jboss.weld.manager.api.WeldManager;

import de.hilling.junit.cdi.annotations.BypassTestInterceptor;
import de.hilling.junit.cdi.util.UserLogger;

/**
 * Singleton for booting the container and starting and stopping the standard CDI contexts.
 */
@BypassTestInterceptor
public class ContextControlWrapper {
    private static final Logger LOG = UserLogger.getInstance();

    private static final ContextControlWrapper INSTANCE = new ContextControlWrapper();
    private final Weld           weld;
    private final WeldManager    weldManager;
    private final ContextControl contextControl;

    /**
     * Returns the singleton.
     * The first invocation of this method will boot the container, if it's not already running.
     *
     * @return the single instance of this class
     */
    public static ContextControlWrapper getInstance() {
        return INSTANCE;
    }

    private ContextControlWrapper() {
        weld = new Weld();
        LOG.info("booting cdi container");
        long start = System.currentTimeMillis();
        weld.addServices(resolveServices());
        WeldContainer weldContainer = weld.initialize();
        long end = System.currentTimeMillis();
        LOG.log(INFO, "booting cdi container finished in {0} ms", end - start);
        if(!weldContainer.isRunning()) {
            throw new CdiTestException("couldn't start weld");
        }
        weldManager = (WeldManager) weldContainer.getBeanManager();
        contextControl = getContextualReference(ContextControl.class);
    }

    private Service[] resolveServices() {
        ServiceLoader services = ServiceLoader.load(Service.class);
        List serviceList = new ArrayList<>();
        for (Service service : services) {
            serviceList.add(service);
        }
        return serviceList.toArray(new Service[serviceList.size()]);
    }

    @SuppressWarnings({ "unchecked" })
    public  T getContextualReference(Class beanType, Annotation ... qualifiers) {
        Bean bean = resolveBean(beanType, qualifiers);
        WeldCreationalContext creationalContext = weldManager.createCreationalContext(null);
        return (T) weldManager.getReference(bean, beanType, creationalContext);
    }

    @SuppressWarnings({ "unchecked" })
    private  Bean resolveBean(Class beanType, Annotation[] qualifiers) {
        Set> beans = weldManager.getBeans(beanType, qualifiers);
        return (Bean) weldManager.resolve(beans);
    }

    public  boolean hasNormalScope(Class beanType, Annotation ... qualifiers) {
        Class bean = resolveBean(beanType, qualifiers).getScope();
        return weldManager.isNormalScope(bean);
    }

    public void startContexts() {
        contextControl.startContexts();
    }

    public void stopContexts() {
        contextControl.stopContexts();
    }

    public void addServices(Service... services) {
        weld.addServices(services);
    }
}