java - Spring 测试和Mockito Controller 测试

@Test
public void testWelcomePage() throws Exception {
    UserDto dto = new UserDto("admin");
    UserEntity user = new UserEntity("admin");
    when(userServiceMock.getUser(dto)).thenReturn(user);

    mockMvc.perform(get("/main/user/welcome?loginId=admin"))
        .andExpect(status().isOk())
        .andExpect(view().name("user/welcome"))
        .andExpect(forwardedUrl("/WEB-INF/pages/user/welcome.jsp"))
        .andExpect(model().attribute("user", hasProperty("loginId", is("admin")))); //-->java.lang.AssertionError: Model attribute 'user' .... but: was null...

    verify(userServiceMock, times(1)).getUser(dto); //-->Argument(s) are different! Wanted:
    verifyNoMoreInteractions(userServiceMock);
}

UserDto是spring mvc表单对象传递的对象。

@RequestMapping(value="/welcome", method = RequestMethod.GET)
public String welcome(UserDto userDto, ModelMap model, Locale locale) {
    UserEntity user = null;
    try {
        user = userService.getUser(userDto);
    } catch (DataNotFoundException e) {
        e.printStackTrace();
        model.addAttribute("message", messageSource.getMessage("msg.data.notfound", null, locale));
    }
    model.addAttribute("user", user);
    return "user/welcome";
}

但是，mockito在传递的参数上引发断言错误（UserDto id不同）。
我该如何解决？

最佳答案

UserDto对象需要重写equals（）方法。

@Override
public boolean equals(Object obj) {
    return EqualsBuilder.reflectionEquals(this, obj);
}

完成了！