orocrm / crm

     * @expectedException \InvalidArgumentException

     * @expectedExceptionMessage Context key is missing

     */

    public function testReadFailed()

    {

        $reader = $this->getReader(null);

        /** @var \PHPUnit_Framework_MockObject_MockObject|StepExecution $stepExecution */

        $stepExecution = $this->getMockBuilder('Akeneo\Bundle\BatchBundle\Entity\StepExecution')

            ->disableOriginalConstructor()

            ->getMock();

        $context = $this->getMock('Oro\Bundle\ImportExportBundle\Context\ContextInterface');

        $this->contextRegistry->expects($this->once())

            ->method('getByStepExecution')

            ->with($stepExecution)

            ->will($this->returnValue($context));

        $reader->setStepExecution($stepExecution);

    }

}

        ];

    }

    public function testSetStepExecutionSetToWrappedWriter()

    {

        $wrapped       = $this->getMock('OroCRM\Bundle\MagentoBundle\Tests\Unit\Stub\StepExecutionAwareWriter');

        $writer        = new ProxyEntityWriter($wrapped, $this->databaseHelper);

        $stepExecution = $this->getMockBuilder('Akeneo\Bundle\BatchBundle\Entity\StepExecution')

            ->disableOriginalConstructor()->getMock();

        $wrapped->expects($this->once())->method('setStepExecution')

            ->with($this->equalTo($stepExecution));

        $writer->setStepExecution($stepExecution);

    }

    public function testSetStepExecutionDoesNotProvokeErrorWithRegularWriter()

    {