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

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

import mockit.*;
import mockit.internal.injection.*;
import mockit.internal.state.*;

import org.springframework.beans.factory.*;
import org.springframework.beans.factory.support.*;

/**
 * If applied, this fake will take over calls to {@link AbstractBeanFactory#getBean(String)} in any implementation
 * class, returning instead a {@link Tested @Tested} or {@link Injectable @Injectable} object with the given field name,
 * or a dependency object injected at any level into a {@code @Tested} object.
 * 
 * In case said calls come (indirectly) from a test class having no {@code @Tested} fields, bean lookup will proceed
 * into the actual {@code getBean(String)} implementation method.
 * 
 * Note this fake is only useful if the code under test makes direct calls to Spring's {@code getBean(name)} method.
 */
public final class FakeBeanFactory extends MockUp
{
   @Mock
   public static Object getBean(Invocation invocation, String name)
   {
      TestedClassInstantiations testedClasses = TestRun.getTestedClassInstantiations();

      if (testedClasses == null) {
         return invocation.proceed();
      }

      Object bean = testedClasses.getBeanExporter().getBean(name);

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

      return bean;
   }
}