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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
<?php
namespace MongoDB\Tests\Model;
use MongoDB\Model\CachingIterator;
use MongoDB\Tests\TestCase;
use Exception;
class CachingIteratorTest extends TestCase
{
/**
* Sanity check for all following tests.
*
* @expectedException \Exception
* @expectedExceptionMessage Cannot traverse an already closed generator
*/
public function testTraversingGeneratorConsumesIt()
{
$iterator = $this->getTraversable([1, 2, 3]);
$this->assertSame([1, 2, 3], iterator_to_array($iterator));
$this->assertSame([1, 2, 3], iterator_to_array($iterator));
}
public function testConstructorRewinds()
{
$iterator = new CachingIterator($this->getTraversable([1, 2, 3]));
$this->assertTrue($iterator->valid());
$this->assertSame(0, $iterator->key());
$this->assertSame(1, $iterator->current());
}
public function testIteration()
{
$iterator = new CachingIterator($this->getTraversable([1, 2, 3]));
$expectedKey = 0;
$expectedItem = 1;
foreach ($iterator as $key => $item) {
$this->assertSame($expectedKey++, $key);
$this->assertSame($expectedItem++, $item);
}
$this->assertFalse($iterator->valid());
}
public function testIterationWithEmptySet()
{
$iterator = new CachingIterator($this->getTraversable([]));
$iterator->rewind();
$this->assertFalse($iterator->valid());
}
public function testPartialIterationDoesNotExhaust()
{
$traversable = $this->getTraversableThatThrows([1, 2, new Exception]);
$iterator = new CachingIterator($traversable);
$expectedKey = 0;
$expectedItem = 1;
foreach ($iterator as $key => $item) {
$this->assertSame($expectedKey++, $key);
$this->assertSame($expectedItem++, $item);
if ($key === 1) {
break;
}
}
$this->assertTrue($iterator->valid());
}
public function testRewindAfterPartialIteration()
{
$iterator = new CachingIterator($this->getTraversable([1, 2, 3]));
$iterator->rewind();
$this->assertTrue($iterator->valid());
$this->assertSame(0, $iterator->key());
$this->assertSame(1, $iterator->current());
$iterator->next();
$this->assertSame([1, 2, 3], iterator_to_array($iterator));
}
public function testCount()
{
$iterator = new CachingIterator($this->getTraversable([1, 2, 3]));
$this->assertCount(3, $iterator);
}
public function testCountAfterPartialIteration()
{
$iterator = new CachingIterator($this->getTraversable([1, 2, 3]));
$iterator->rewind();
$this->assertTrue($iterator->valid());
$this->assertSame(0, $iterator->key());
$this->assertSame(1, $iterator->current());
$iterator->next();
$this->assertCount(3, $iterator);
}
public function testCountWithEmptySet()
{
$iterator = new CachingIterator($this->getTraversable([]));
$this->assertCount(0, $iterator);
}
private function getTraversable($items)
{
foreach ($items as $item) {
yield $item;
}
}
private function getTraversableThatThrows($items)
{
foreach ($items as $item) {
if ($item instanceof Exception) {
throw $item;
} else {
yield $item;
}
}
}
}