View source
<?php
namespace Drupal\system\Tests\Entity;
class EntityQueryAggregateTest extends EntityUnitTestBase {
public static $modules = array();
protected $entityStorage;
protected $queryResult;
public $factory;
protected function setUp() {
parent::setUp();
$this->entityStorage = $this->entityManager
->getStorage('entity_test');
$this->factory = $this->container
->get('entity.query');
for ($i = 1; $i <= 2; $i++) {
$field_name = 'field_test_' . $i;
entity_create('field_storage_config', array(
'field_name' => $field_name,
'entity_type' => 'entity_test',
'type' => 'integer',
'cardinality' => 2,
))
->save();
entity_create('field_config', array(
'field_name' => $field_name,
'entity_type' => 'entity_test',
'bundle' => 'entity_test',
))
->save();
}
$entity = $this->entityStorage
->create(array(
'id' => 1,
'user_id' => 1,
'field_test_1' => 1,
'field_test_2' => 2,
));
$entity
->enforceIsNew();
$entity
->save();
$entity = $this->entityStorage
->create(array(
'id' => 2,
'user_id' => 2,
'field_test_1' => 1,
'field_test_2' => 7,
));
$entity
->enforceIsNew();
$entity
->save();
$entity = $this->entityStorage
->create(array(
'id' => 3,
'user_id' => 2,
'field_test_1' => 2,
'field_test_2' => 1,
));
$entity
->enforceIsNew();
$entity
->save();
$entity = $this->entityStorage
->create(array(
'id' => 4,
'user_id' => 2,
'field_test_1' => 2,
'field_test_2' => 8,
));
$entity
->enforceIsNew();
$entity
->save();
$entity = $this->entityStorage
->create(array(
'id' => 5,
'user_id' => 3,
'field_test_1' => 2,
'field_test_2' => 2,
));
$entity
->enforceIsNew();
$entity
->save();
$entity = $this->entityStorage
->create(array(
'id' => 6,
'user_id' => 3,
'field_test_1' => 3,
'field_test_2' => 8,
));
$entity
->enforceIsNew();
$entity
->save();
}
public function testAggregation() {
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 1,
),
array(
'user_id' => 2,
),
array(
'user_id' => 3,
),
));
$function_expected = array();
$function_expected['count'] = array(
array(
'id_count' => 6,
),
);
$function_expected['min'] = array(
array(
'id_min' => 1,
),
);
$function_expected['max'] = array(
array(
'id_max' => 6,
),
);
$function_expected['sum'] = array(
array(
'id_sum' => 21,
),
);
$function_expected['avg'] = array(
array(
'id_avg' => 21.0 / 6.0,
),
);
foreach ($function_expected as $aggregation_function => $expected) {
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', $aggregation_function)
->execute();
$this
->assertEqual($this->queryResult, $expected);
}
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 1,
'id_count' => 1,
),
array(
'user_id' => 2,
'id_count' => 3,
),
array(
'user_id' => 3,
'id_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->groupBy('id')
->conditionAggregate('id', 'COUNT', 8)
->execute();
$this
->assertResults(array());
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('id')
->conditionAggregate('id', 'COUNT', 8)
->execute();
$this
->assertResults(array());
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'count')
->groupBy('id')
->conditionAggregate('id', 'COUNT', 6)
->execute();
$this
->assertResults(array(
array(
'id_count' => 6,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'count')
->conditionAggregate('id', 'count', 2)
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'id_count' => 2,
'user_id' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'count')
->conditionAggregate('id', 'COUNT', 1, '>')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'id_count' => 2,
'user_id' => 3,
),
array(
'id_count' => 3,
'user_id' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->sortAggregate('id', 'COUNT')
->execute();
$this
->assertSortedResults(array(
array(
'id_count' => 6,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->sortAggregate('id', 'COUNT')
->execute();
$this
->assertSortedResults(array(
array(
'id_count' => 6,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->groupBy('user_id')
->sortAggregate('id', 'COUNT', 'DESC')
->execute();
$this
->assertSortedResults(array(
array(
'user_id' => 2,
'id_count' => 3,
),
array(
'user_id' => 3,
'id_count' => 2,
),
array(
'user_id' => 1,
'id_count' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->groupBy('user_id')
->sortAggregate('id', 'COUNT', 'ASC')
->execute();
$this
->assertSortedResults(array(
array(
'user_id' => 1,
'id_count' => 1,
),
array(
'user_id' => 3,
'id_count' => 2,
),
array(
'user_id' => 2,
'id_count' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->groupBy('user_id')
->sortAggregate('id', 'COUNT')
->conditionAggregate('id', 'COUNT', 2)
->execute();
$this
->assertSortedResults(array(
array(
'id_count' => 2,
'user_id' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->groupBy('user_id')
->sortAggregate('id', 'COUNT', 'ASC')
->conditionAggregate('id', 'COUNT', 3, '<')
->execute();
$this
->assertSortedResults(array(
array(
'id_count' => 1,
'user_id' => 1,
),
array(
'id_count' => 2,
'user_id' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('id', 'COUNT')
->groupBy('user_id')
->sortAggregate('id', 'COUNT', 'DESC')
->conditionAggregate('id', 'COUNT', 3, '<')
->execute();
$this
->assertSortedResults(array(
array(
'id_count' => 2,
'user_id' => 3,
),
array(
'id_count' => 1,
'user_id' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
),
array(
'field_test_1' => 2,
),
array(
'field_test_1' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('user_id', 'COUNT')
->groupBy('field_test_1')
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'user_id_count' => 2,
),
array(
'field_test_1' => 2,
'user_id_count' => 3,
),
array(
'field_test_1' => 3,
'user_id_count' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('field_test_1', 'COUNT')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 1,
'field_test_1_count' => 1,
),
array(
'user_id' => 2,
'field_test_1_count' => 3,
),
array(
'user_id' => 3,
'field_test_1_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('field_test_1', 'SUM')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 1,
'field_test_1_sum' => 1,
),
array(
'user_id' => 2,
'field_test_1_sum' => 5,
),
array(
'user_id' => 3,
'field_test_1_sum' => 5,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('field_test_1', 'SUM')
->aggregate('field_test_2', 'SUM')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 1,
'field_test_1_sum' => 1,
'field_test_2_sum' => 2,
),
array(
'user_id' => 2,
'field_test_1_sum' => 5,
'field_test_2_sum' => 16,
),
array(
'user_id' => 3,
'field_test_1_sum' => 5,
'field_test_2_sum' => 10,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('field_test_1', 'SUM')
->aggregate('field_test_1', 'COUNT')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 1,
'field_test_1_sum' => 1,
'field_test_1_count' => 1,
),
array(
'user_id' => 2,
'field_test_1_sum' => 5,
'field_test_1_count' => 3,
),
array(
'user_id' => 3,
'field_test_1_sum' => 5,
'field_test_1_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->aggregate('field_test_2', 'COUNT')
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
),
array(
'field_test_1' => 2,
'field_test_2_count' => 3,
),
array(
'field_test_1' => 3,
'field_test_2_count' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->aggregate('field_test_2', 'COUNT')
->aggregate('field_test_2', 'SUM')
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
'field_test_2_sum' => 9,
),
array(
'field_test_1' => 2,
'field_test_2_count' => 3,
'field_test_2_sum' => 11,
),
array(
'field_test_1' => 3,
'field_test_2_count' => 1,
'field_test_2_sum' => 8,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->conditionAggregate('field_test_1', 'COUNT', 3)
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 2,
'field_test_1_count' => 3,
),
array(
'user_id' => 3,
'field_test_1_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('field_test_1', 'SUM')
->conditionAggregate('field_test_1', 'COUNT', 2, '>')
->groupBy('user_id')
->execute();
$this
->assertResults(array(
array(
'user_id' => 2,
'field_test_1_sum' => 5,
'field_test_1_count' => 3,
),
array(
'user_id' => 3,
'field_test_1_sum' => 5,
'field_test_1_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->conditionAggregate('user_id', 'COUNT', 2)
->groupBy('field_test_1')
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'user_id_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->conditionAggregate('user_id', 'COUNT', 2, '>')
->groupBy('field_test_1')
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'user_id_count' => 2,
),
array(
'field_test_1' => 2,
'user_id_count' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->conditionAggregate('field_test_2', 'COUNT', 2)
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->conditionAggregate('field_test_2', 'COUNT', 2, '>')
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
),
array(
'field_test_1' => 2,
'field_test_2_count' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->conditionAggregate('field_test_2', 'COUNT', 2)
->conditionAggregate('field_test_2', 'SUM', 8)
->execute();
$this
->assertResults(array());
$this->queryResult = $this->factory
->getAggregate('entity_test', 'OR')
->groupBy('field_test_1')
->conditionAggregate('field_test_2', 'COUNT', 2)
->conditionAggregate('field_test_2', 'SUM', 8)
->execute();
$this
->assertResults(array(
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
'field_test_2_sum' => 9,
),
array(
'field_test_1' => 3,
'field_test_2_count' => 1,
'field_test_2_sum' => 8,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('field_test_1', 'COUNT')
->groupBy('user_id')
->sort('user_id', 'DESC')
->execute();
$this
->assertSortedResults(array(
array(
'user_id' => 3,
'field_test_1_count' => 2,
),
array(
'user_id' => 2,
'field_test_1_count' => 3,
),
array(
'user_id' => 1,
'field_test_1_count' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->aggregate('field_test_1', 'COUNT')
->groupBy('user_id')
->sort('user_id', 'ASC')
->execute();
$this
->assertSortedResults(array(
array(
'user_id' => 1,
'field_test_1_count' => 1,
),
array(
'user_id' => 2,
'field_test_1_count' => 3,
),
array(
'user_id' => 3,
'field_test_1_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->conditionAggregate('field_test_1', 'COUNT', 2, '>')
->groupBy('user_id')
->sort('user_id', 'ASC')
->execute();
$this
->assertSortedResults(array(
array(
'user_id' => 2,
'field_test_1_count' => 3,
),
array(
'user_id' => 3,
'field_test_1_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->sortAggregate('field_test_1', 'COUNT', 'DESC')
->groupBy('user_id')
->execute();
$this
->assertSortedResults(array(
array(
'user_id' => 2,
'field_test_1_count' => 3,
),
array(
'user_id' => 3,
'field_test_1_count' => 2,
),
array(
'user_id' => 1,
'field_test_1_count' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->sortAggregate('field_test_1', 'COUNT', 'ASC')
->groupBy('user_id')
->execute();
$this
->assertSortedResults(array(
array(
'user_id' => 1,
'field_test_1_count' => 1,
),
array(
'user_id' => 3,
'field_test_1_count' => 2,
),
array(
'user_id' => 2,
'field_test_1_count' => 3,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->aggregate('field_test_2', 'COUNT')
->sort('field_test_1', 'ASC')
->execute();
$this
->assertSortedResults(array(
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
),
array(
'field_test_1' => 2,
'field_test_2_count' => 3,
),
array(
'field_test_1' => 3,
'field_test_2_count' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->aggregate('field_test_2', 'COUNT')
->sort('field_test_1', 'DESC')
->execute();
$this
->assertSortedResults(array(
array(
'field_test_1' => 3,
'field_test_2_count' => 1,
),
array(
'field_test_1' => 2,
'field_test_2_count' => 3,
),
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->sortAggregate('field_test_2', 'COUNT', 'DESC')
->execute();
$this
->assertSortedResults(array(
array(
'field_test_1' => 2,
'field_test_2_count' => 3,
),
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
),
array(
'field_test_1' => 3,
'field_test_2_count' => 1,
),
));
$this->queryResult = $this->factory
->getAggregate('entity_test')
->groupBy('field_test_1')
->sortAggregate('field_test_2', 'COUNT', 'ASC')
->execute();
$this
->assertSortedResults(array(
array(
'field_test_1' => 3,
'field_test_2_count' => 1,
),
array(
'field_test_1' => 1,
'field_test_2_count' => 2,
),
array(
'field_test_1' => 2,
'field_test_2_count' => 3,
),
));
}
protected function assertResults($expected, $sorted = FALSE) {
$found = TRUE;
$expected_keys = array_keys($expected);
foreach ($this->queryResult as $key => $row) {
$keys = $sorted ? array(
$key,
) : $expected_keys;
foreach ($keys as $key) {
$expected_row = $expected[$key];
if (!array_diff_assoc($row, $expected_row) && !array_diff_assoc($expected_row, $row)) {
continue 2;
}
}
$found = FALSE;
break;
}
return $this
->assertTrue($found, strtr('!expected expected, !found found', array(
'!expected' => print_r($expected, TRUE),
'!found' => print_r($this->queryResult, TRUE),
)));
}
protected function assertSortedResults($expected) {
return $this
->assertResults($expected, TRUE);
}
}