FlushingBatchTest.php

<?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));
    }
}