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:
<?php
namespace Guzzle\Tests\Batch;
use Guzzle\Batch\FlushingBatch;
use Guzzle\Batch\Batch;
class FlushingBatchTest extends \Guzzle\Tests\GuzzleTestCase
{
public function testFlushesWhenSizeMeetsThreshold()
{
$t = $this->getMock('Guzzle\Batch\BatchTransferInterface', array('transfer'));
$d = $this->getMock('Guzzle\Batch\BatchDivisorInterface', array('createBatches'));
$batch = new Batch($t, $d);
$queue = $this->readAttribute($batch, 'queue');
$d->expects($this->exactly(2))
->method('createBatches')
->will($this->returnCallback(function () use ($queue) {
$items = array();
foreach ($queue as $item) {
$items[] = $item;
}
return array($items);
}));
$t->expects($this->exactly(2))
->method('transfer');
$flush = new FlushingBatch($batch, 3);
$this->assertEquals(3, $flush->getThreshold());
$flush->setThreshold(2);
$flush->add('foo')->add('baz')->add('bar')->add('bee')->add('boo');
$this->assertEquals(1, count($flush));
}
}