使用mockito初始化spring批处理作业执行

Question

问题：

我正在为我的一个spring批处理作业方法编写单元测试。我使用mockito来模拟我的批处理作业依赖关系。在jobExecution发挥作用之前，一切都很好。我要测试的方法调用了jobExecution变量，但它给了我NPE（NullPointerException）并且我没有成功地用mockito模拟它。

    @Override
    public List<Date> getValidDates() {
        List<Date> allDates = dateService.findAllFailedDates(getJobName(), false);
        List<Date> datesThatCanBeWorkedOn = getAllDatesThatHaveNecessaryDependencies(allDates);
        Collection<Date> datesThatCannotBeWorkedOn = CollectionUtils.subtract(allDates, datesThatCanBeWorkedOn);
        currentJobExecution.getExecutionContext().put("datesThatCannotBeWorkedOn", datesThatCannotBeWorkedOn);
        return datesThatCanBeWorkedOn;
    }

@RunWith(SpringJUnit4ClassRunner.class)
public class ValidDateReaderTest extends BaseTest {

    @InjectMocks
    private ValidDateReader validDateReader;

    @Mock
    private DateService dateService;

    @Mock
    private JobExecution currentJobExecution;

    @Mock
    private JobInstance jobInstance;

    @Test
    public void getDatesNeededProcessingTest() {

        when(jobInstance.getJobName()).thenReturn(BatchNames.VALID_DATE_READER);
        when(currentJobExecution.getJobInstance()).thenReturn(jobInstance);
        List<Date> allDates = new ArrayList<>();
        allDates.add(DateUtils.addDays(new Date(), -5));
        allDates.add(DateUtils.addDays(new Date(), -4));
        allDates.add(DateUtils.addDays(new Date(), -3));
        allDates.add(DateUtils.addDays(new Date(), -2));
        allDates.add(DateUtils.addDays(new Date(), -1));
        when(dateService.findAllFailedDates(BatchNames.VALID_DATE_READER, false))
                .thenReturn(allDates);
        assertEquals(1, validDateReader.getValidDates().size());
    }

}

删除此currentJobExecution时

currentJobExecution.getExecutionContext().put("datesThatCannotBeWorkedOn", datesThatCannotBeWorkedOn);

从我要测试的方法，然后测试成功完成。但我不能删除它，因为它是业务逻辑的一部分。

Answer 1

CurrentJobExecution.GetExecutionContext()似乎返回NULL。你可能也想嘲弄一下：

when(currentJobExecution.getExecutionContext()).thenReturn(new ExecutionContext());