在SpringBoot中用@SpyBean来表示一个“间谍对象”,允许它的某些方法被模拟,而剩下的方法仍然是真实的方法。
import com.**.hydra.account.HydraAccountApplication;
import com.**.hydra.account.config.DataSourceConfiguration;
import com.**.hydra.client.model.SettleVO;
import com.**.hydra.client.request.SettleRequest;
import com.**.hydra.common.enums.ChannelEnum;
import com.**.infra.rpc.Result;
import org.junit.Test;
import org.junit.runner.RunWith;
import static org.mockito.BDDMockito.*;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.boot.test.mock.mockito.SpyBean;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.jdbc.Sql;
import org.springframework.test.context.junit4.SpringRunner;
import java.util.Date;
import java.util.List;
/**
* @description: 单元测试类,会有一些依赖的对象被mock对象替代
* @author: pilaf
* @create: 2019-04-26 17:22
*/
@ActiveProfiles("ut")
@RunWith(SpringRunner.class)
@SpringBootTest(classes = { HydraAccountApplication.class,SettleServiceImplUnitTest.class,DataSourceConfiguration.class})
@DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_EACH_TEST_METHOD)
public class SettleServiceImplUnitTest {
@SpyBean
private SettleServiceImpl settleService;
@Test
@Sql("classpath:/db/case/settle/test_settle_with_duplication_exception.sql")
public void testSettleWithDuplicateRecordException(){
given(settleService.isAlreadySettled(any(SettleRequest.class))).willReturn(false);
...
}
}
我的本意是只将settleService的isAlreadySettled方法给mock(模拟)掉,不进入它,直接返回false。可是运行的时候却进入了这个方法,由于我mock时指定的参数是any(SettleRequest.class),所以进入方法的时候原来的代码里报了空指针的异常。
上网查了一下才发现自己用错了。正确的写法是
//得先指定返回值,否则就会进入
willReturn(false).given(settleService).isAlreadySettled(any(SettleRequest.class));
之前的错误写法是
//这种写法会进入真实方法,是错误的用法!!!
given(settleService.isAlreadySettled(any(SettleRequest.class))).willReturn(false);
参考文章:
1.https://github.com/spring-projects/spring-boot/issues/8600
2.https://static.javadoc.io/org.mockito/mockito-core/2.7.17/org/mockito/Mockito.html#spy