使用Mockito时，Spring AOP Aspect无法工作

Question

我有一个@Aspect，它编织了我所有控制器动作方法的执行。当我运行系统时，它工作得很好，但在单元测试中不是这样。我以以下方式使用Mockito和junit:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration("file:**/spring-context.xml")
@WebAppConfiguration
public class UserControllerTest {        
    private MockMvc mockMvc;
    @Mock
    private RoleService roleService;
    @InjectMocks
    private UserController userController;
    @Before
    public void setUp() {
       MockitoAnnotations.initMocks(this);                    
       ...    
       mockMvc = MockMvcBuilders.standaloneSetup(userController).build();
    }    
    ...
}

和一些@Test用mockMvc.perform()。

和我的Aspect是:

@Pointcut("within(@org.springframework.stereotype.Controller *)")
public void controller() { }
@Pointcut("execution(* mypackage.controller.*Controller.*(..))")
public void methodPointcut() { }
@Around("controller() && methodPointcut()")
...

Answer 1

首先，有必要使用webAppContextSetup如Jason建议的:

@Autowired
private WebApplicationContext webApplicationContext;
@Before
public void setUp() throws Exception {
   ...
   mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build();
}

此时应该触发方面，但Mockito不会注入mock。这是因为Spring AOP使用代理对象，并且模拟被注入到这个代理对象而不是被代理的对象中。要解决这个问题，有必要打开对象并使用ReflectionUtils而不是@InjectMocks注释:

private MockMvc mockMvc;
@Mock
private RoleService roleService;
private UserController userController;
@Autowired
private WebApplicationContext webApplicationContext;
@Before
public void setUp() {
   MockitoAnnotations.initMocks(this);                    
   UserController unwrappedController = (UserController) unwrapProxy(userController);
   ReflectionTestUtils.setField(unwrappedController, "roleService", roleService);   
   mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build();
}
...
public static final Object unwrapProxy(Object bean) throws Exception {
/*
 * If the given object is a proxy, set the return value as the object
 * being proxied, otherwise return the given object.
 */
    if (AopUtils.isAopProxy(bean) && bean instanceof Advised) {
        Advised advised = (Advised) bean;
        bean = advised.getTargetSource().getTarget();
    }
    return bean;
}

此时，任何对when(…). thenreturn(…)的调用都应该能正常工作。

说明如下:http://kim.saabye-pedersen.org/2012/12/mockito-and-spring-proxies.html

Answer 2

您可能正在使用Spring AOP，在这种情况下，bean必须是Spring bean才能使AOP工作，通过不在控制器中自动装配，它完全绕过了Spring AOP机制。

我认为修复应该是简单地注入控制器

 @Autowired
 @InjectMocks
 private UserController userController;