1 files changed, 40 insertions, 0 deletions
diff --git a/vendor/guzzle/guzzle/tests/Guzzle/Tests/Batch/FlushingBatchTest.php b/vendor/guzzle/guzzle/tests/Guzzle/Tests/Batch/FlushingBatchTest.php
new file mode 100644
index 0000000..9b37a48
--- /dev/null
+++ b/vendor/guzzle/guzzle/tests/Guzzle/Tests/Batch/FlushingBatchTest.php
@@ -0,0 +1,40 @@
+<?php
+
+namespace Guzzle\Tests\Batch;
+
+use Guzzle\Batch\FlushingBatch;
+use Guzzle\Batch\Batch;
+
+/**
+ * @covers Guzzle\Batch\FlushingBatch
+ */
+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));
+    }
+}