File size: 1.31Kb
<?php
namespace Guzzle\Tests\Batch;
use Guzzle\Batch\ExceptionBufferingBatch;
use Guzzle\Batch\Batch;
use Guzzle\Batch\BatchSizeDivisor;
/**
* @covers Guzzle\Batch\ExceptionBufferingBatch
*/
class ExceptionBufferingBatchTest extends \Guzzle\Tests\GuzzleTestCase
{
public function testFlushesEntireBatchWhileBufferingErroredBatches()
{
$t = $this->getMockBuilder('Guzzle\Batch\BatchTransferInterface')
->setMethods(array('transfer'))
->getMock();
$d = new BatchSizeDivisor(1);
$batch = new Batch($t, $d);
$called = 0;
$t->expects($this->exactly(3))
->method('transfer')
->will($this->returnCallback(function ($batch) use (&$called) {
if (++$called === 2) {
throw new \Exception('Foo');
}
}));
$decorator = new ExceptionBufferingBatch($batch);
$decorator->add('foo')->add('baz')->add('bar');
$result = $decorator->flush();
$e = $decorator->getExceptions();
$this->assertEquals(1, count($e));
$this->assertEquals(array('baz'), $e[0]->getBatch());
$decorator->clearExceptions();
$this->assertEquals(0, count($decorator->getExceptions()));
$this->assertEquals(array('foo', 'bar'), $result);
}
}