summaryrefslogtreecommitdiff
path: root/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/Elastic/QueryEngine/AggregationsTest.php
blob: cdcc45da83563282cf9ffe3f0aa85886a8f7f091 (plain)
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
<?php

namespace SMW\Tests\Elastic\QueryEngine;

use SMW\Elastic\QueryEngine\Aggregations;
use SMW\Elastic\QueryEngine\FieldMapper;

/**
 * @covers \SMW\Elastic\QueryEngine\Aggregations
 * @group semantic-mediawiki
 *
 * @license GNU GPL v2+
 * @since 3.0
 *
 * @author mwjames
 */
class AggregationsTest extends \PHPUnit_Framework_TestCase {

	public function testCanConstruct() {

		$this->assertInstanceOf(
			Aggregations::class,
			new Aggregations()
		);
	}

	/**
	 * @dataProvider parametersProvider
	 */
	public function testResolve( $parameters, $expected ) {

		$instance = new Aggregations( $parameters );

		$this->assertEquals(
			$expected,
			(string)$instance
		);
	}

	public function parametersProvider() {

		$fieldMapper = new FieldMapper();

		yield [
			[],
			'[]'
		];

		yield [
			$fieldMapper->aggs_terms( 'test_1', 'category' ),
			'{"aggregations":{"test_1":{"terms":{"field":"category"}}}}'
		];

		yield [
			$fieldMapper->aggs_terms( 'test_2', 'category', [ 'size' => 5 ] ),
			'{"aggregations":{"test_2":{"terms":{"field":"category","size":5}}}}'
		];

		yield [
			[
				$fieldMapper->aggs_terms( 'test_1', 'foo' ),
				$fieldMapper->aggs_terms( 'test_2', 'bar' )
			],
			'{"aggregations":{"test_1":{"terms":{"field":"foo"}},"test_2":{"terms":{"field":"bar"}}}}'
		];

		yield [
			[
				new Aggregations( $fieldMapper->aggs_terms( 'test_1', 'foo' ) ),
				new Aggregations( $fieldMapper->aggs_terms( 'test_2', 'bar' ) )
			],
			'{"aggregations":{"test_1":{"terms":{"field":"foo"}},"test_2":{"terms":{"field":"bar"}}}}'
		];

		// https://www.elastic.co/blog/intro-to-Aggregations-pt-2-sub-Aggregations
		$aggs = new Aggregations(
			$fieldMapper->aggs_terms( 'all_boroughs', 'borough' )
		);

		$aggs->addSubAggregations(
			new Aggregations(
				$fieldMapper->aggs_terms( 'cause', 'contributing_factor_vehicle', [ 'size' => 3 ] )
			)
		);

		yield [
			$aggs,
			'{"aggregations":{"all_boroughs":{"terms":{"field":"borough"},"aggregations":{"cause":{"terms":{"field":"contributing_factor_vehicle","size":3}}}}}}'
		];

		$cause = new Aggregations(
			$fieldMapper->aggs_terms( 'cause', 'contributing_factor_vehicle', [ 'size' => 3 ] )
		);

		$cause->addSubAggregations(
			new Aggregations(
				$fieldMapper->aggs_date_histogram( 'incidents_per_month', '@timestamp', 'month' )
			)
		);

		$aggs = new Aggregations(
			$fieldMapper->aggs_terms( 'all_boroughs', 'borough' )
		);

		$aggs->addSubAggregations(
			$cause
		);

		yield [
			$aggs,
			'{"aggregations":{"all_boroughs":{"terms":{"field":"borough"},"aggregations":{"cause":{"terms":{"field":"contributing_factor_vehicle","size":3},"aggregations":{"incidents_per_month":{"date_histogram":{"field":"@timestamp","interval":"month"}}}}}}}}'
		];
	}

}