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