1: 2: 3: 4: 5: 6: 7: 8: 9: 10: 11: 12: 13: 14: 15: 16: 17: 18: 19: 20: 21: 22: 23: 24: 25: 26: 27: 28: 29: 30: 31: 32: 33: 34: 35: 36: 37: 38: 39: 40: 41: 42: 43: 44: 45: 46: 47: 48: 49: 50: 51: 52: 53: 54: 55: 56: 57: 58: 59: 60: 61: 62: 63: 64: 65: 66: 67: 68: 69: 70: 71: 72: 73: 74: 75: 76: 77: 78: 79: 80: 81: 82: 83: 84: 85: 86: 87: 88: 89: 90: 91: 92:
<?php
namespace Guzzle\Tests\Batch;
use Guzzle\Batch\Batch;
use Guzzle\Batch\Exception\BatchTransferException;
class BatchTest extends \Guzzle\Tests\GuzzleTestCase
{
private function getMockTransfer()
{
return $this->getMock('Guzzle\Batch\BatchTransferInterface');
}
private function getMockDivisor()
{
return $this->getMock('Guzzle\Batch\BatchDivisorInterface');
}
public function testAddsItemsToQueue()
{
$batch = new Batch($this->getMockTransfer(), $this->getMockDivisor());
$this->assertSame($batch, $batch->add('foo'));
$this->assertEquals(1, count($batch));
}
public function testFlushReturnsItems()
{
$transfer = $this->getMockTransfer();
$transfer->expects($this->exactly(2))
->method('transfer');
$divisor = $this->getMockDivisor();
$divisor->expects($this->once())
->method('createBatches')
->will($this->returnValue(array(array('foo', 'baz'), array('bar'))));
$batch = new Batch($transfer, $divisor);
$batch->add('foo')->add('baz')->add('bar');
$items = $batch->flush();
$this->assertEquals(array('foo', 'baz', 'bar'), $items);
}
public function testThrowsExceptionContainingTheFailedBatch()
{
$called = 0;
$originalException = new \Exception('Foo!');
$transfer = $this->getMockTransfer();
$transfer->expects($this->exactly(2))
->method('transfer')
->will($this->returnCallback(function () use (&$called, $originalException) {
if (++$called == 2) {
throw $originalException;
}
}));
$divisor = $this->getMockDivisor();
$batch = new Batch($transfer, $divisor);
$queue = $this->readAttribute($batch, 'queue');
$divisor->expects($this->once())
->method('createBatches')
->will($this->returnCallback(function ($batch) use ($queue) {
foreach ($queue as $item) {
$items[] = $item;
}
return array_chunk($items, 2);
}));
$batch->add('foo')->add('baz')->add('bar')->add('bee')->add('boo');
$this->assertFalse($batch->isEmpty());
try {
$items = $batch->flush();
$this->fail('Expected exception');
} catch (BatchTransferException $e) {
$this->assertEquals($originalException, $e->getPrevious());
$this->assertEquals(array('bar', 'bee'), array_values($e->getBatch()));
$this->assertEquals(1, count($batch));
}
}
}