我有一个针对tasklet的Junit测试,类似于:
@ContextConfiguration(locations = {"/context/job-runner-context.xml"})
@TestExecutionListeners({ DependencyInjectionTestExecutionListener.class, StepScopeTestExecutionListener.class })
@RunWith(SpringJUnit4ClassRunner.class)
public class InicializaTaskletTest extends BaseTeste {
@Resource
private ChunkContext chunkContext;
@Before
public void setUp() throws Exception {
}
@Test
public void testExecutaTaskletInicializacao() throws Exception {
AtividadesContext atividadesContext = create();
ExecutionContext ctx = chunkContext.getStepContext().getStepExecution().getJobExecution().getExecutionContext();
ctx.put(ATIVIDADES_FOLHA, atividadesContext);
when(service.criarArquivo(diretorio, atividadesContext)).thenReturn(new Emissao();
RepeatStatus retorno = tasklet.execute(mock(StepContribution.class), chunkContext);
}
}
在tasklet中,当我用" diretorio"来调用service.criarArquivo时,我想返回一个emissao对象。和" atividadesContext"参数。
tasklet类似于:
@Value("#{jobExecutionContext['atividadesFolha']}")
private AtividadesContext atividadesContext;
@Override
public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) throws IOException, AtividadeJaIniciadaException {
EmissaoArquivoSpai emissao = arquivoSpaiService.criarArquivo(diretorioBaseArquivo, atividadesContext);
return RepeatStatus.FINISHED;
}
}
问题是当我执行junit测试时,注入atividadesContext不起作用或者它无法在上下文中正确放置。而test的when子句返回一个null emissao对象,因为它不是方法所期望的参数。我可以使用Mockito.any()来做到这一点,但首先我想知道为什么我不能在junit测试中注入上下文对象。
当我以正常方式运行作业时,我可以正常注入atividadesContext。
答案 0 :(得分:1)
在Spring Batch单元测试中,我们使用一个名为JobSynchronizationManager
的组件(步骤也相同)。在调用作业范围的bean之前,请使用JobExecution
注册JobSynchronizationManager
。这使ApplicationContext
可用于接线目的。
所以看看你的测试,你的结构如下:
@Test
public void testExecutaTaskletInicializacao() throws Exception {
AtividadesContext atividadesContext = create();
JobExecution jobExecution = new JobExecution(5l);
ExecutionContext ctx = new ExecutionContext();
ctx.put(ATIVIDADES_FOLHA, atividadesContext);
jobExecution.setExecutionContext(ctx);
JobSynchronizationManager.register(jobExecution);
when(service.criarArquivo(diretorio, atividadesContext)).thenReturn(new Emissao());
RepeatStatus retorno = tasklet.execute(mock(StepContribution.class), chunkContext);
JobSynchronizationManager.release();
}
您可以在此处的文档中详细了解JobSynchronizationManager
:http://docs.spring.io/spring-batch/trunk/apidocs/org/springframework/batch/core/scope/context/JobSynchronizationManager.html