|
| 1 | +<?php |
| 2 | +namespace Elastica\Test\Aggregation; |
| 3 | + |
| 4 | +use Elastica\Aggregation\BucketSelector; |
| 5 | +use Elastica\Aggregation\DateHistogram; |
| 6 | +use Elastica\Aggregation\Max; |
| 7 | +use Elastica\Document; |
| 8 | +use Elastica\Query; |
| 9 | + |
| 10 | +class BucketSelectorTest extends BaseAggregationTest |
| 11 | +{ |
| 12 | + protected function _getIndexForTest() |
| 13 | + { |
| 14 | + $index = $this->_createIndex(); |
| 15 | + |
| 16 | + $index->getType('test')->addDocuments([ |
| 17 | + new Document(1, ['date' => '2018-12-01', 'value' => 1]), |
| 18 | + new Document(2, ['date' => '2018-12-02', 'value' => 2]), |
| 19 | + new Document(3, ['date' => '2018-12-03', 'value' => 5]), |
| 20 | + new Document(4, ['date' => '2018-12-04', 'value' => 4]), |
| 21 | + new Document(5, ['date' => '2018-12-05', 'value' => 6]), |
| 22 | + new Document(6, ['date' => '2018-12-06', 'value' => 9]), |
| 23 | + new Document(7, ['date' => '2018-12-07', 'value' => 11]), |
| 24 | + new Document(8, ['date' => '2018-12-08', 'value' => 4]), |
| 25 | + new Document(9, ['date' => '2018-12-09', 'value' => 7]), |
| 26 | + new Document(10, ['date' => '2018-12-10', 'value' => 4]), |
| 27 | + ]); |
| 28 | + |
| 29 | + $index->refresh(); |
| 30 | + |
| 31 | + return $index; |
| 32 | + } |
| 33 | + |
| 34 | + /** |
| 35 | + * @group unit |
| 36 | + */ |
| 37 | + public function testToArray() |
| 38 | + { |
| 39 | + $expected = [ |
| 40 | + 'max' => [ |
| 41 | + 'field' => 'value', |
| 42 | + ], |
| 43 | + 'aggs' => [ |
| 44 | + 'selector_agg' => [ |
| 45 | + 'bucket_selector' => [ |
| 46 | + 'buckets_path' => ['max' => 'max_agg'], |
| 47 | + 'script' => 'params.max > 5', |
| 48 | + ], |
| 49 | + ], |
| 50 | + ], |
| 51 | + ]; |
| 52 | + |
| 53 | + $maxAgg = new Max('max_agg'); |
| 54 | + $maxAgg->setField('value'); |
| 55 | + |
| 56 | + $selectorAgg = new BucketSelector('selector_agg', ['max' => 'max_agg'], 'params.max > 5'); |
| 57 | + $maxAgg->addAggregation($selectorAgg); |
| 58 | + |
| 59 | + $this->assertEquals($expected, $maxAgg->toArray()); |
| 60 | + } |
| 61 | + |
| 62 | + /** |
| 63 | + * @group functional |
| 64 | + */ |
| 65 | + public function testMaxAggregation() |
| 66 | + { |
| 67 | + $index = $this->_getIndexForTest(); |
| 68 | + |
| 69 | + $dateHistogramAgg = new DateHistogram('histogram_agg', 'date', 'day'); |
| 70 | + $dateHistogramAgg->setFormat('yyyy-MM-dd'); |
| 71 | + |
| 72 | + $maxAgg = new Max('max_agg'); |
| 73 | + $maxAgg->setField('value'); |
| 74 | + $dateHistogramAgg->addAggregation($maxAgg); |
| 75 | + |
| 76 | + $bucketAgg = new BucketSelector('selector_agg', ['max' => 'max_agg'], 'params.max > 5'); |
| 77 | + $dateHistogramAgg->addAggregation($bucketAgg); |
| 78 | + |
| 79 | + $query = new Query(); |
| 80 | + $query->addAggregation($dateHistogramAgg); |
| 81 | + |
| 82 | + $dateHistogramAggResult = $index->search($query)->getAggregation('histogram_agg')['buckets']; |
| 83 | + |
| 84 | + $this->assertEquals(4, count($dateHistogramAggResult)); |
| 85 | + $this->assertEquals(6, $dateHistogramAggResult[0]['max_agg']['value']); |
| 86 | + $this->assertEquals(9, $dateHistogramAggResult[1]['max_agg']['value']); |
| 87 | + $this->assertEquals(11, $dateHistogramAggResult[2]['max_agg']['value']); |
| 88 | + $this->assertEquals(7, $dateHistogramAggResult[3]['max_agg']['value']); |
| 89 | + } |
| 90 | +} |
0 commit comments