在Junit测试中检索上下文值。

时间:2015-06-23 16:12:55

标签: spring spring-batch

我有一个针对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。

1 个答案:

答案 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();
}

您可以在此处的文档中详细了解JobSynchronizationManagerhttp://docs.spring.io/spring-batch/trunk/apidocs/org/springframework/batch/core/scope/context/JobSynchronizationManager.html