9

我有一个@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()")
...
4

2 回答 2

10

首先,有必要按照 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

于 2014-04-12T22:07:30.537 回答
0

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

我认为解决方法应该是简单地注入控制器

 @Autowired
 @InjectMocks
 private UserController userController;
于 2013-06-03T18:11:46.430 回答