1 files changed, 91 insertions, 0 deletions
diff --git a/vendor/guzzle/guzzle/tests/Guzzle/Tests/Batch/BatchTest.php b/vendor/guzzle/guzzle/tests/Guzzle/Tests/Batch/BatchTest.php
new file mode 100644
index 0000000..296f57a
--- /dev/null
+++ b/vendor/guzzle/guzzle/tests/Guzzle/Tests/Batch/BatchTest.php
@@ -0,0 +1,91 @@
+<?php
+
+namespace Guzzle\Tests\Batch;
+
+use Guzzle\Batch\Batch;
+use Guzzle\Batch\Exception\BatchTransferException;
+
+/**
+ * @covers Guzzle\Batch\Batch
+ */
+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);
+
+        // PHPunit clones objects before passing them to a callback.
+        // Horrible hack to get around this!
+        $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));
+        }
+    }
+}