mockit.integration.springframework.BeanLookup Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmockit Show documentation

JMockit is a Java toolkit for automated developer testing. It contains APIs for the creation of the objects to be tested, for mocking dependencies, and for faking external APIs; JUnit (4 & 5) and TestNG test runners are supported. It also contains an advanced code coverage tool.

The newest version!

/*
 * Copyright (c) 2006 JMockit developers
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.integration.springframework;

import edu.umd.cs.findbugs.annotations.NonNull;
import edu.umd.cs.findbugs.annotations.Nullable;

import mockit.internal.injection.BeanExporter;

import org.springframework.beans.factory.BeanNotOfRequiredTypeException;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;

final class BeanLookup {
    private BeanLookup() {
    }

    @NonNull
    static Object getBean(@NonNull BeanExporter beanExporter, @NonNull String name) {
        Object bean = beanExporter.getBean(name);

        if (bean == null) {
            throw new NoSuchBeanDefinitionException(name);
        }

        return bean;
    }

    @NonNull
    @SuppressWarnings("unchecked")
    static  T getBean(@NonNull BeanExporter beanExporter, @NonNull String name, @Nullable Class requiredType) {
        if (requiredType == null) {
            return (T) getBean(beanExporter, name);
        }

        T bean = (T) beanExporter.getBean(name);

        if (bean != null) {
            Class actualType = bean.getClass();

            if (!requiredType.isAssignableFrom(actualType)) {
                throw new BeanNotOfRequiredTypeException(name, requiredType, actualType);
            }
        } else {
            bean = beanExporter.getBean(requiredType);

            if (bean == null) {
                throw new NoSuchBeanDefinitionException(requiredType, "with bean name \"" + name + '"');
            }
        }

        return bean;
    }

    @NonNull
    static  T getBean(@NonNull BeanExporter beanExporter, @NonNull Class requiredType) {
        T bean = beanExporter.getBean(requiredType);

        if (bean == null) {
            throw new NoSuchBeanDefinitionException(requiredType);
        }

        return bean;
    }
}