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
|
<?php
namespace Elastica\Test\Aggregation;
use Elastica\Aggregation\Avg;
use Elastica\Aggregation\Filter;
use Elastica\Aggregation\Filters;
use Elastica\Document;
use Elastica\Filter\Term;
use Elastica\Query;
class FiltersTest extends BaseAggregationTest
{
protected function _getIndexForTest()
{
$index = $this->_createIndex('filter');
$index->getType('test')->addDocuments(array(
new Document(1, array('price' => 5, 'color' => 'blue')),
new Document(2, array('price' => 8, 'color' => 'blue')),
new Document(3, array('price' => 1, 'color' => 'red')),
new Document(4, array('price' => 3, 'color' => 'green')),
));
$index->refresh();
return $index;
}
/**
* @group unit
*/
public function testToArrayUsingNamedFilters()
{
$expected = array(
'filters' => array(
'filters' => array(
'blue' => array(
'term' => array('color' => 'blue'),
),
'red' => array(
'term' => array('color' => 'red'),
),
),
),
'aggs' => array(
'avg_price' => array('avg' => array('field' => 'price')),
),
);
$agg = new Filters('by_color');
$agg->addFilter(new Term(array('color' => 'blue')), 'blue');
$agg->addFilter(new Term(array('color' => 'red')), 'red');
$avg = new Avg('avg_price');
$avg->setField('price');
$agg->addAggregation($avg);
$this->assertEquals($expected, $agg->toArray());
}
/**
* @group unit
*/
public function testToArrayUsingAnonymousFilters()
{
$expected = array(
'filters' => array(
'filters' => array(
array(
'term' => array('color' => 'blue'),
),
array(
'term' => array('color' => 'red'),
),
),
),
'aggs' => array(
'avg_price' => array('avg' => array('field' => 'price')),
),
);
$agg = new Filters('by_color');
$agg->addFilter(new Term(array('color' => 'blue')));
$agg->addFilter(new Term(array('color' => 'red')));
$avg = new Avg('avg_price');
$avg->setField('price');
$agg->addAggregation($avg);
$this->assertEquals($expected, $agg->toArray());
}
/**
* @group functional
*/
public function testFilterAggregation()
{
$agg = new Filters('by_color');
$agg->addFilter(new Term(array('color' => 'blue')), 'blue');
$agg->addFilter(new Term(array('color' => 'red')), 'red');
$avg = new Avg('avg_price');
$avg->setField('price');
$agg->addAggregation($avg);
$query = new Query();
$query->addAggregation($agg);
$results = $this->_getIndexForTest()->search($query)->getAggregation('by_color');
$resultsForBlue = $results['buckets']['blue'];
$resultsForRed = $results['buckets']['red'];
$this->assertEquals(2, $resultsForBlue['doc_count']);
$this->assertEquals(1, $resultsForRed['doc_count']);
$this->assertEquals((5 + 8) / 2, $resultsForBlue['avg_price']['value']);
$this->assertEquals(1, $resultsForRed['avg_price']['value']);
}
}
|