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
|
<?php
namespace Elastica\Test;
use Elastica\Document;
use Elastica\Query;
use Elastica\ResultSet;
use Elastica\Scroll;
use Elastica\Search;
class ScrollTest extends Base
{
/**
* Full foreach test.
*
* @group functional
*/
public function testForeach()
{
$scroll = new Scroll($this->_prepareSearch());
$count = 1;
/** @var ResultSet $resultSet */
foreach ($scroll as $scrollId => $resultSet) {
$this->assertNotEmpty($scrollId);
$results = $resultSet->getResults();
switch (true) {
case $count === 1:
// hits: 1 - 5
$this->assertEquals(5, $resultSet->count());
$this->assertEquals('1', $results[0]->getId());
$this->assertEquals('5', $results[4]->getId());
break;
case $count === 2:
// hits: 6 - 10
$this->assertEquals(5, $resultSet->count());
$this->assertEquals('6', $results[0]->getId());
$this->assertEquals('10', $results[4]->getId());
break;
case $count === 3:
// hit: 11
$this->assertEquals(1, $resultSet->count());
$this->assertEquals('11', $results[0]->getId());
break;
case $count === 4:
$this->assertEquals(0, $resultSet->count());
break;
default:
$this->fail('too many iterations');
}
$count++;
}
}
/**
* Scroll must not overwrite options.
*
* @group functional
*/
public function testSearchRevert()
{
$search = $this->_prepareSearch();
$search->setOption(Search::OPTION_SCROLL, 'must');
$search->setOption(Search::OPTION_SCROLL_ID, 'not');
$search->setOption(Search::OPTION_SEARCH_TYPE, 'change');
$old = $search->getOptions();
$scroll = new Scroll($search);
$scroll->rewind();
$this->assertEquals($old, $search->getOptions());
$scroll->next();
$this->assertEquals($old, $search->getOptions());
}
/**
* index: 11 docs
* query size: 5.
*
* @return Search
*/
private function _prepareSearch()
{
$index = $this->_createIndex();
$index->refresh();
$docs = array();
for ($x = 1; $x <= 11; $x++) {
$docs[] = new Document($x, array('id' => $x, 'key' => 'value'));
}
$type = $index->getType('scrollTest');
$type->addDocuments($docs);
$index->refresh();
$search = new Search($this->_getClient());
$search->addIndex($index)->addType($type);
$search->getQuery()->setSize(5);
return $search;
}
}
|
