getBeansOfType（類<T>型）

Question

我想寫的JUnit我的API之一，該API中我使用如下圖：

Map<String, T> beansMap = ctx.getBeansOfType(clazz); 

凡

ctx = org.springframework.context.ApplicationContext 
clazz = Class<T> 

我需要模擬ctx.getBeansOfType(clazz)並得到這Map<Spring, T>的回報，我無法做到這一點。

Answer 1

通常，從ApplicationContext直接檢索bean被認爲是一種糟糕的做法，因爲它引入了耦合。 看到爲什麼https://stackoverflow.com/a/9663099/6604329。

使用字段，構造函數或查找方法注入將消除對模擬ApplicationContext的需要。

任何方式，這裏是你如何嘲笑ApplicationContext.getBeansOfType(clazz)

import org.junit.Test; 
import org.mockito.Mockito; 
import org.mockito.invocation.InvocationOnMock; 
import org.mockito.stubbing.Answer; 
import org.springframework.context.ApplicationContext; 

import java.util.HashMap; 
import java.util.Map; 

import static org.junit.Assert.assertFalse; 
import static org.mockito.Matchers.any; 
import static org.mockito.Mockito.mock; 
import static org.mockito.Mockito.when; 

/** 
* @author mponomarev 
*/ 
public class ApiTest { 
    @Test 
    public void testSomething() throws Exception { 
     ApplicationContext applicationContext = mock(ApplicationContext.class); 
     final Map beans = new HashMap(); 

     when(applicationContext.getBeansOfType(any(Class.class))) 
      .thenAnswer(new Answer<Map<String,Object>>() { 
       @Override 
       public Map<String,Object> answer(InvocationOnMock invocation) 
       throws Throwable { 
        Class clazz = invocation.getArgumentAt(0, Class.class); 
        beans.put("beanName", mock(clazz)); 
        return beans; 
       } 
      }); 

     Api api = new Api(applicationContext); 
     api.perform(); 

     assertFalse("beans shouldn't be empty", beans.isEmpty()); 
     for(Object o : beans.values()) { 
      Component component = (Component)o; 
      Mockito.verify(component).doSomething(); 
     } 
    } 

    public static class Api { 
     private final Map<String,Component> components; 

     Api(ApplicationContext applicationContext) { 
      this.components = applicationContext.getBeansOfType(Component.class); 
     } 

     void perform() { 
      for(Component component : components.values()) { 
       component.doSomething(); 
      } 
     } 
    } 

    public interface Component { 
     void doSomething(); 
    } 
}