如何模拟自定义JdbcCursorItemReader在Spring Batch上的单个步骤单元测试？

Question

我正在尝试创建一个Junit测试用例，用于测试批处理作业的单个步骤，而批处理作业又有一个自定义的JdbcCursorItemReader，并将数据写入平面文件。我无法模拟可以注入处理器以写入平面文件的数据。请帮忙。

我按照以下链接中的步骤创建了一个测试用例来启动一个单独的步骤：

https://docs.spring.io/spring-batch/trunk/reference/html/testing.html

由于我的reader组件是stepscope，所以我使用上面链接中第10.4节的上下文将所需的域对象放入ExecutionContext中，但在启动该步骤时，它不会识别相同的内容，而是执行我通过配置xml注入的实际SQL查询。

我的配置xml的步骤如下：

<step id="sample-step" next="next-step">
 <tasklet>
    <chunk reader="customJDBCReader" processor="customProcessor" 
     writer="customFlatFileItemWriter" commit-interval="XXX" />
 </tasklet>
</step>

My customreader实现了JdbcCursorItemReader，它定义了一个用于从数据库读取信息的SQL。

在我的单元测试中，我试图模拟数据，而不是依赖于数据库中的实际数据

private StepExecution execution;

@BeforeStep
public StepExecution getStepExection() {
    execution = MetaDataInstanceFactory.createStepExecution();
    return execution;
}

@Test
public void testSampleStep() throws UnexpectedInputException, ParseException, Exception {
    MyDomainObject domainObject= new MyDomainObject ();
    domainObject.setID("sampleID");
    domainObject.setName("sampleName");
    domainObject.setImage("sampleImage");
    execution.getExecutionContext().put("someKey", domainObject);
    execution.setReadCount(1);

    JobExecution jobExecution = jobLauncherTestUtils.launchStep("sample-step", jobParameters, execution.getExecutionContext());
    AssertFile.assertFileEquals(new FileSystemResource(EXPECTED_FILE), new FileSystemResource(OUTPUT_FILE));
}

预期结果应该是，launchstep需要从executionContext获取数据，而不是从数据库将数据写入faltfile。

Answer 1

你的期望是错误的。如果该步骤使用了JdbcBatchItemWriter，那么无论您是否在执行上下文中模拟数据，该读取器都会在测试期间发出SQL查询。步骤的输入数据与存储在执行上下文中的数据不同。

在这种情况下，我建议在测试中使用嵌入式数据库，并用一些虚拟数据填充它。

希望这能有所帮助。