Spring Boot+Mockito+JUnit中的@Mock注入@InjectMocks失效

问题描述

测试代码如下：

@RunWith(SpringRunner.class)
@SpringBootTest
public class CheckConfirmPayServiceTest {

    private static final Logger logger = LoggerFactory.getLogger(CheckConfirmPayServiceTest.class);

    @Mock
    private CashierService cashierService;

    @Autowired
    @InjectMocks // 被注入mock对象的类一般是被测试类
    private CheckConfirmPayService checkConfirmPayService;

    @Autowired
    private TScanStatementJobMapper tScanStatementJobMapper;


    private String jobFlowNo = FlowNoUtil.getTxNo();

    private String sysOrderNo = "S" + FlowNoUtil.getTxNo();

    @Before
    public void setUp() throws Exception {
        // Mock
        MockitoAnnotations.initMocks(this);
        Mockito.doAnswer(new Answer<Object>() {
            @Override
            public Object answer(InvocationOnMock invocation) {
                Object[] args = invocation.getArguments();
                CashierCheckConfirmPayRequest request = (CashierCheckConfirmPayRequest) args[0];
                logger.info("=====Mock 接口测试=====");
                CashierCheckConfirmPayResponse response = new CashierCheckConfirmPayResponse();
                if (request == null) {
                    throw new BizException(ErrorCodeEnum.NULL_POINTER_EXCEPTION.code,
                            ErrorCodeEnum.NULL_POINTER_EXCEPTION.msg);
                }
                response.setReqsysNo(request.getReqsysNo());
                response.setVersion(request.getVersion());
                return response;
            }
        }).when(cashierService).checkConfirmPay(Mockito.any(CashierCheckConfirmPayRequest.class));
        insertTScanStatementJob();
    }

    /**
     * 插入对账状态数据
     *
     * @author shenhaiwen
     * @time 2018年8月27日 下午2:12:31
     */
    private void insertTScanStatementJob() {
        String checkDate = TimeUtil.addDays(-1); // 当前日期前一天
        TScanStatementJob job = new TScanStatementJob();
        job.setJobFlowNo(jobFlowNo);
        job.setCheckDate(checkDate);
        job.setStatementFlag(StatementFlagEnum.FINISH.getCode());
        job.setCreateTime(TimeUtil.getTimeStamp());
        job.setUpdateTime(TimeUtil.getTimeStamp());
        job.setRemark("单元测试数据");
        tScanStatementJobMapper.insert(job);
    }

    @After
    public void delData() {
        tScanStatementJobMapper.deleteByPrimaryKey(jobFlowNo);
    }

    /**
     * 批量处理确认支付场景
     *
     * @author shenhaiwen
     * @time 2018年8月4日 上午11:26:16
     */
    @Test
    public void testRemoteCheckConfirmPay() {
        CallBackStatusEnum callBackStatusEnum = null;
        try {
            callBackStatusEnum = checkConfirmPayService.remoteCheckConfirmPay();
        } catch (Exception e) {
            TestCase.fail();
        }
        TestCase.assertEquals(CallBackStatusEnum.SUCCESS.getStatus(), callBackStatusEnum.getStatus());

    }

测试类里将cashierService对象mock后注入到checkConfirmPayService中,然后在测试方法中执行cashierService的方法。

但是现在问题是checkConfirmPayService的cashierService属性没有被@Mock标签注入,而是调用了@Autowired标签,用的是spring生成的bean 而不是mock的cashierService.

运行结果如下：

问题原因

当前版本只支持setter 的方式进行注入，Mockito 首先尝试类型注入，如果有多个类型相同的mock 对象，那么它会根据名称进行注入。当注入失败的时候Mockito不会抛出任何异常，所以你可能需要手动去验证它的安全性。

因此需要在checkConfirmPayService中对cashierService采用setter 的方式进行注入，示例代码代码如下：

@Before
    public void setUp() throws Exception {
        // Mock
        MockitoAnnotations.initMocks(this);
        checkConfirmPayService.setCashierService(cashierService);
        ............

测试如预期进入cashierService的mock中，如下所示：