如何在Spring使用Mockito只嘲笑一个bean的方法？

Question

我在Spring测试中使用@主和@Profile来模拟bean：

@Profile("test")
@Configuration
public class TestBeanConf {

@Bean
@Primary
public UserService userService() {
    UserService userService = Mockito.mock(UserService.class);
    TokenValidationUrl validation = new TokenValidationUrl();
    validation.setValid(true);
    validation.setUid("123456789");
    Mockito.when(userService.tokenValidation("23456")).thenReturn(validation);
    return userService;
}

但是UserServicebean的其他方法返回null，我如何监视真正创建的bean，而只监视mocktokenValidation方法？

Answer 1

为了监视UserServicebean的一种方法，在构建该bean的过程中，我们通过将UserService的现有实例作为参数发送，并使用Mockito的监视功能，从Spring上下文自动连接UserService的现有实例：

@Profile("test")
@Configuration
public class TestBeanConf {

@Bean
@Primary
public UserService userServiceTest(UserService userService) {
    UserService userService = Mockito.spy(userService);
    TokenValidationUrl validation = new TokenValidationUrl();
    validation.setValid(true);
    validation.setUid("123456789");
    Mockito.when(userService.tokenValidation("23456")).thenReturn(validation);
    return userService;
}