/** * Test for stats aggregation toArray() method. */ public function testToArray() { $aggregation = new StatsAggregation('test_agg'); $aggregation->setField('test_field'); $expectedResult = ['agg_test_agg' => ['stats' => ['field' => 'test_field']]]; $this->assertEquals($expectedResult, $aggregation->toArray()); }
/** * Test for stats aggregation when script is set. */ public function testStatsAggregationWithScriptSet() { /** @var Repository $repo */ $repo = $this->getManager()->getRepository('AcmeTestBundle:Product'); $aggregation = new StatsAggregation('test_agg'); $aggregation->setField('price'); $aggregation->setScript('_value * 1.2'); $search = $repo->createSearch()->addAggregation($aggregation); $results = $repo->execute($search, Repository::RESULTS_RAW); $expectedResult = ['agg_test_agg' => ['count' => 3, 'min' => 12.54, 'max' => 38.4, 'sum' => 69.06, 'avg' => 23.02]]; $this->assertArrayHasKey('aggregations', $results); foreach ($expectedResult['agg_test_agg'] as $checkKey => $checkValue) { $this->assertEquals($checkValue, $results['aggregations']['agg_test_agg'][$checkKey], '', 0.01); } }
/** * Data provider for testNestedAggregation. * * @return array */ public function getNestedAggregationData() { $out = []; $mapping = ['product' => ['properties' => ['sub_products' => ['type' => 'nested', 'properties' => ['id' => ['type' => 'string', 'index' => 'not_analyzed'], 'title' => ['type' => 'string'], 'price' => ['type' => 'float']]]]]]; // Case #0 simple terms aggregation. $aggregation = new TermsAggregation('test_terms_agg'); $aggregation->setField('sub_products.title'); $result = ['doc_count' => 4, 'agg_test_terms_agg' => ['buckets' => [['key' => 'foo', 'doc_count' => 2], ['key' => 'bar', 'doc_count' => 1], ['key' => 'baz', 'doc_count' => 1]]]]; $out[] = [$aggregation, $result, $mapping]; // Case #1 simple stats aggregation. $aggregation = new StatsAggregation('test_stats_agg'); $aggregation->setField('sub_products.price'); $result = ['doc_count' => 4, 'agg_test_stats_agg' => ['count' => 4, 'min' => 10, 'max' => 1000, 'sum' => 1135, 'avg' => 283.75]]; $out[] = [$aggregation, $result, $mapping]; return $out; }
/** * Test for histogram aggregation. Order and min doc count set. Elasticsearch version >= 1.5.0. */ public function testHistogramAggregationWithOrderAndMinDocCountSet() { /** @var Repository $repo */ $repo = $this->getManager()->getRepository('AcmeTestBundle:Product'); $aggregation = new HistogramAggregation('test_agg'); $aggregation->setField('price'); $aggregation->setInterval(5); $aggregation->setMinDocCount(2); $statsAggregation = new StatsAggregation('price_stats'); $statsAggregation->setField('price'); $aggregation->setOrder($statsAggregation->getName() . '.min', HistogramAggregation::DIRECTION_ASC); $aggregation->addAggregation($statsAggregation); $expectedResults = ['doc_count' => 2, 'agg_price_stats' => ['count' => 2, 'min' => 2.0, 'max' => 3.0, 'avg' => 2.5, 'sum' => 5.0]]; $search = $repo->createSearch()->addAggregation($aggregation); $results = $repo->execute($search, Repository::RESULTS_RAW)['aggregations']; $resultBucket = $results['agg_test_agg']['buckets'][0]; $this->assertEquals($expectedResults['doc_count'], $resultBucket['doc_count']); foreach ($expectedResults['agg_price_stats'] as $checkKey => $checkValue) { $this->assertEquals($checkValue, $resultBucket['agg_price_stats'][$checkKey], '', 0.01); } }
/** * {@inheritdoc} */ public function preProcessSearch(Search $search, Search $relatedSearch, FilterState $state = null) { $stateAgg = new StatsAggregation('date_range_agg'); $stateAgg->setField($this->getField()); $search->addAggregation($stateAgg); }