如何将数据模型模拟注入弹簧控制器？

Question

在我的单元测试中，我需要在 spring @Controller 中模拟一个数据模型实例，否则 @RequestMapping 方法的返回值会出错。

为此，我尝试了以下方法：

1. 创建一个User mock，调用user.login()，需要返回“true”
2. 在 LoginController 中注入模拟对象
3. 存根登录方法以返回 true
4. 使用 Spring 测试中的 MockMVC 执行 POST /Login
5. 验证是否调用了 mockUser.login

这里是控制器方法：

@RequestMapping(value = "/Login", method = RequestMethod.POST)
  public String updateUI(Locale locale, Model model, @RequestParam("username") String username,
      @RequestParam("hashedPW") String hashedPW, HttpServletRequest request) {
    model.addAttribute("username", username);
    user = new User(username, username, hashedPW.getBytes(), LoginHandler.getInstance());

    boolean loginResult = user.login();
    if(loginResult == true) {
      return "profile";
    }
    String output = "Failed login (" + username + ") requested, locale = " + locale;
    log(output);
    return "home";
  }

以及我通过注入初始化模拟对象：

@Mock
  private User mockUser;

  @InjectMocks
  private LoginController injectedLoginController;


  @Before
  public void setup() throws ServletException {
    MockitoAnnotations.initMocks(this);
    mvc = MockMvcBuilders.standaloneSetup(injectedLoginController).build();
    LOGFILE = new File("logs/general.log");
  }

最后是单元测试：

@Test
  public void testLoginSuccess() throws Exception {
    String username = "Stefan";
    byte[] hashedPW = "".getBytes();
    when(mockUser.login()).thenReturn(true);
    ResultActions ra = mvc
        .perform(post("/Login").param("username", username).param("hashedPW", hashedPW.toString()))
        .andExpect(status().isOk());

    verify(mockUser).login();
  }

总的来说，我希望控制器处理的 User 对象是 mockUser 类型而不是 User，并且 login() 方法会被调用一次（相应地）并返回“true”。

但我得到的只是 “需要但未调用：mockUser.login() 实际上，与此模拟的交互为零。”

感谢任何解决我的问题的建议，因为我已经为此工作了很长一段时间，但我似乎不明白这一点。

Answer 1

1) 这似乎是一个 IT 测试.. 你不应该强制任何注入.. spring 会这样做：

@InjectMocks
private LoginController injectedLoginController;

2) 你需要监视你的控制器：

@SpyBean
private LoginController injectedLoginController;

3)您需要将User的创建移动到控制器内部的包级方法中：

user = createUser(username, username, hashedPW.getBytes(), LoginHandler.getInstance()); 


...

User createUser(...){
   return new User(username, username, hashedPW.getBytes(), LoginHandler.getInstance());
}

4) 使该方法返回一个模拟用户：

  doReturn(mockUser).when(injectedLoginController).createUser(...);
  when(mockUser.login()).thenReturn(true);

Answer 2

如果您不想更改代码，可以从 PowerMockito 获得帮助，它可以帮助模拟新对象的构造。

@RunWith(PowerMockRunner.class)
@PrepareForTest({User.class})
public class ControllerUnderTest {
  // here User is class for which we want to mock object creation
}

现在，让我们模拟对象

@Test
public void testLoginSuccess() throws Exception {
 ....
 User userMock = PowerMockito.createMock(User.class);

 PowerMockito.whenNew(User.class).withArguments(username, username, hashedPW.getBytes(), LoginHandler.getInstance()).thenReturn(userMock);

 expect(userMock.login()).andReturn(true);

 verify(mockUser).login();
 ...
}

有关powermock的更多详细信息，请查看PowerMockito