Spring AOP 方面无法使用 Mockito

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()。

我的观点是：

@Pointcut("within(@org.springframework.stereotype.Controller *)")
public void controller() { }

@Pointcut("execution(* mypackage.controller.*Controller.*(..))")
public void methodPointcut() { }

@Around("controller() && methodPointcut()")
...

Answer 1

首先需要按照 Jason 的建议使用 webAppContextSetup：

@Autowired
private WebApplicationContext webApplicationContext;

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

此时应该触发方面，但 Mockito 不会注入模拟。这是因为 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;