/**
* @param array $contextSharedKeys
* @param array $context
* @param array $job
* @param bool $isJobSuccess
* @param int $jobExecutions
* @param array $expectedContext
*
* @dataProvider executeDataProvider
*/
public function testExecute(array $contextSharedKeys, array $context, array $job, $isJobSuccess = true, $jobExecutions = 0, $expectedContext = [])
{
if ($job) {
list($jobType, $jobName) = $job;
$this->executor->addPostProcessingJob($jobType, $jobName);
}
$this->executor->setContextSharedKeys($contextSharedKeys);
/** @var \PHPUnit_Framework_MockObject_MockObject|StepExecution $stepExecution */
$stepExecution = $this->getMockBuilder('Akeneo\\Bundle\\BatchBundle\\Entity\\StepExecution')->disableOriginalConstructor()->getMock();
$executionContext = new ExecutionContext();
foreach ($context as $key => $value) {
$executionContext->put($key, $value);
}
$jobExecution = $this->getMock('Akeneo\\Bundle\\BatchBundle\\Entity\\JobExecution');
$jobInstance = $this->getMock('Akeneo\\Bundle\\BatchBundle\\Entity\\JobInstance');
$jobExecution->expects($this->any())->method('getJobInstance')->will($this->returnValue($jobInstance));
$jobExecution->expects($this->any())->method('getExecutionContext')->will($this->returnValue($executionContext));
$stepExecution->expects($this->any())->method('getJobExecution')->will($this->returnValue($jobExecution));
$this->executor->setStepExecution($stepExecution);
$jobResult = new JobResult();
$jobResult->setSuccessful($isJobSuccess);
if (!$isJobSuccess) {
$this->setExpectedException('Oro\\Bundle\\ImportExportBundle\\Exception\\RuntimeException');
}
$this->jobExecutor->expects($this->exactly($jobExecutions))->method('executeJob')->willReturn($jobResult);
$this->processor->expects($this->any())->method('process')->willReturnArgument(0);
$this->reader->expects($this->atLeastOnce())->method('read')->willReturnOnConsecutiveCalls(new \stdClass(), new \stdClass(), new \stdClass(), null);
$this->executor->execute();
$this->assertEquals($expectedContext, $executionContext->getKeys());
}
/**
* {@inheritdoc}
*/
public function doExecute(StepExecution $stepExecution)
{
$this->initializeStepElements($stepExecution);
$stepExecutor = new PostProcessStepExecutor();
$stepExecutor->setStepExecution($stepExecution)->setJobExecutor($this->jobExecutor)->setReader($this->reader)->setProcessor($this->processor)->setWriter($this->writer);
if (null !== $this->batchSize) {
$stepExecutor->setBatchSize($this->batchSize);
}
if ($this->contextSharedKeys) {
$stepExecutor->setContextSharedKeys($this->contextSharedKeys);
}
if ($this->postProcessingJobs) {
$jobType = $stepExecution->getJobExecution()->getJobInstance()->getType();
foreach ($this->postProcessingJobs as $jobName) {
$stepExecutor->addPostProcessingJob($jobType, $jobName);
}
}
$stepExecutor->execute($this);
$this->flushStepElements();
}
