使用 mockito 来模拟我正在测试的课程

Question

假设我有一个名为 UserController 的控制器类，它有 2 个方法：getUserCount() 和 getLatestUser()， getUserCount 调用 getLatestUser。

@Controller
class UserController{

public long getUserCount(){
#code
getLatestUser();
#code
}

public User getLatestUser(){}
}

我应该使用 Junit 和 Mockito 测试这些方法中的每一个，因此我有这样的东西：

class UserControllerTest{
@Autowired
UserController userController;

@Test
public void testing_get_user_count(){
User user = new User();
when(userController.getLastestUser()).thenReturn(user);
}
}

我的问题是我无法模拟 UserController，因为我已经自动连接了它，所以我不能在 getLatestUser 上使用 when().thenReturn()。

有没有办法让我模拟它？

Answer 1

您可以使用@SpyBean 代替@Autowired。它在 bean 上应用 Mockito spy。

class UserControllerTest {
  @SpyBean
  UserController userController;

  @Test
  public void testing_get_user_count(){
    User user = new User();
    when(userController.getLastestUser()).thenReturn(user);
  }
}