View source
<?php
namespace Drupal\KernelTests\Core\Database;
use Drupal\Component\Render\FormattableMarkup;
use Drupal\Core\Database\Database;
use Drupal\Core\Database\Query\Condition;
use Drupal\Core\Database\RowCountException;
use Drupal\user\Entity\User;
class SelectComplexTest extends DatabaseTestBase {
public static $modules = [
'system',
'user',
'node_access_test',
'field',
];
public function testDefaultJoin() {
$query = $this->connection
->select('test_task', 't');
$people_alias = $query
->join('test', 'p', 't.pid = p.id');
$name_field = $query
->addField($people_alias, 'name', 'name');
$query
->addField('t', 'task', 'task');
$priority_field = $query
->addField('t', 'priority', 'priority');
$query
->orderBy($priority_field);
$result = $query
->execute();
$num_records = 0;
$last_priority = 0;
foreach ($result as $record) {
$num_records++;
$this
->assertTrue($record->{$priority_field} >= $last_priority, 'Results returned in correct order.');
$this
->assertNotEqual($record->{$name_field}, 'Ringo', 'Taskless person not selected.');
$last_priority = $record->{$priority_field};
}
$this
->assertEqual($num_records, 7, 'Returned the correct number of rows.');
}
public function testLeftOuterJoin() {
$query = $this->connection
->select('test', 'p');
$people_alias = $query
->leftJoin('test_task', 't', 't.pid = p.id');
$name_field = $query
->addField('p', 'name', 'name');
$query
->addField($people_alias, 'task', 'task');
$query
->addField($people_alias, 'priority', 'priority');
$query
->orderBy($name_field);
$result = $query
->execute();
$num_records = 0;
$last_name = 0;
foreach ($result as $record) {
$num_records++;
$this
->assertTrue(strcmp($record->{$name_field}, $last_name) >= 0, 'Results returned in correct order.');
}
$this
->assertEqual($num_records, 8, 'Returned the correct number of rows.');
}
public function testGroupBy() {
$query = $this->connection
->select('test_task', 't');
$count_field = $query
->addExpression('COUNT(task)', 'num');
$task_field = $query
->addField('t', 'task');
$query
->orderBy($count_field);
$query
->groupBy($task_field);
$result = $query
->execute();
$num_records = 0;
$last_count = 0;
$records = [];
foreach ($result as $record) {
$num_records++;
$this
->assertTrue($record->{$count_field} >= $last_count, 'Results returned in correct order.');
$last_count = $record->{$count_field};
$records[$record->{$task_field}] = $record->{$count_field};
}
$correct_results = [
'eat' => 1,
'sleep' => 2,
'code' => 1,
'found new band' => 1,
'perform at superbowl' => 1,
];
foreach ($correct_results as $task => $count) {
$this
->assertEqual($records[$task], $count, new FormattableMarkup("Correct number of '@task' records found.", [
'@task' => $task,
]));
}
$this
->assertEqual($num_records, 6, 'Returned the correct number of total rows.');
}
public function testGroupByAndHaving() {
$query = $this->connection
->select('test_task', 't');
$count_field = $query
->addExpression('COUNT(task)', 'num');
$task_field = $query
->addField('t', 'task');
$query
->orderBy($count_field);
$query
->groupBy($task_field);
$query
->having('COUNT(task) >= 2');
$result = $query
->execute();
$num_records = 0;
$last_count = 0;
$records = [];
foreach ($result as $record) {
$num_records++;
$this
->assertTrue($record->{$count_field} >= 2, 'Record has the minimum count.');
$this
->assertTrue($record->{$count_field} >= $last_count, 'Results returned in correct order.');
$last_count = $record->{$count_field};
$records[$record->{$task_field}] = $record->{$count_field};
}
$correct_results = [
'sleep' => 2,
];
foreach ($correct_results as $task => $count) {
$this
->assertEqual($records[$task], $count, new FormattableMarkup("Correct number of '@task' records found.", [
'@task' => $task,
]));
}
$this
->assertEqual($num_records, 1, 'Returned the correct number of total rows.');
}
public function testRange() {
$query = $this->connection
->select('test');
$query
->addField('test', 'name');
$query
->addField('test', 'age', 'age');
$query
->range(0, 2);
$query_result = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($query_result, 2, 'Returned the correct number of rows.');
}
public function testRangeUndo() {
$query = $this->connection
->select('test');
$query
->addField('test', 'name');
$query
->addField('test', 'age', 'age');
$query
->range(0, 2);
$query
->range(NULL, NULL);
$query_result = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($query_result, 4, 'Returned the correct number of rows.');
}
public function testDistinct() {
$query = $this->connection
->select('test_task');
$query
->addField('test_task', 'task');
$query
->distinct();
$query_result = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($query_result, 6, 'Returned the correct number of rows.');
}
public function testCountQuery() {
$query = $this->connection
->select('test');
$name_field = $query
->addField('test', 'name');
$age_field = $query
->addField('test', 'age', 'age');
$query
->orderBy('name');
$count = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($count, 4, 'Counted the correct number of records.');
$record = $query
->execute()
->fetch();
$this
->assertEqual($record->{$name_field}, 'George', 'Correct data retrieved.');
$this
->assertEqual($record->{$age_field}, 27, 'Correct data retrieved.');
}
public function testHavingCountQuery() {
$query = $this->connection
->select('test')
->extend('Drupal\\Core\\Database\\Query\\PagerSelectExtender')
->groupBy('age')
->having('age + 1 > 0');
$query
->addField('test', 'age');
$query
->addExpression('age + 1');
$count = count($query
->execute()
->fetchCol());
$this
->assertEqual($count, 4, 'Counted the correct number of records.');
}
public function testCountQueryRemovals() {
$query = $this->connection
->select('test');
$query
->fields('test');
$query
->orderBy('name');
$count = $query
->countQuery();
$tables = $query
->getTables();
$this
->assertEqual($tables['test']['all_fields'], 1, 'Query correctly sets \'all_fields\' statement.');
$tables = $count
->getTables();
$this
->assertFalse(isset($tables['test']['all_fields']), 'Count query correctly unsets \'all_fields\' statement.');
$orderby = $query
->getOrderBy();
$db_type = Database::getConnection()
->databaseType();
$this
->assertEqual($orderby['name'], $db_type == 'pgsql' ? 'ASC NULLS FIRST' : 'ASC', 'Query correctly sets ordering clause.');
$orderby = $count
->getOrderBy();
$this
->assertFalse(isset($orderby['name']), 'Count query correctly unsets ordering clause.');
$count = $count
->execute()
->fetchField();
$this
->assertEqual($count, 4, 'Counted the correct number of records.');
}
public function testCountQueryFieldRemovals() {
$query = $this->connection
->select('test');
$query
->fields('test', [
'fail',
]);
$this
->assertEqual(4, $query
->countQuery()
->execute()
->fetchField(), 'Count Query removed fields');
$query = $this->connection
->select('test');
$query
->addExpression('fail');
$this
->assertEqual(4, $query
->countQuery()
->execute()
->fetchField(), 'Count Query removed expressions');
}
public function testCountQueryDistinct() {
$query = $this->connection
->select('test_task');
$query
->addField('test_task', 'task');
$query
->distinct();
$count = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($count, 6, 'Counted the correct number of records.');
}
public function testCountQueryGroupBy() {
$query = $this->connection
->select('test_task');
$query
->addField('test_task', 'pid');
$query
->groupBy('pid');
$count = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($count, 3, 'Counted the correct number of records.');
$query = $this->connection
->select('test_task');
$query
->addField('test_task', 'pid', 'pid_alias');
$query
->addExpression('COUNT(test_task.task)', 'count');
$query
->groupBy('pid_alias');
$query
->orderBy('pid_alias', 'asc');
$count = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($count, 3, 'Counted the correct number of records.');
}
public function testNestedConditions() {
$query = $this->connection
->select('test');
$query
->addField('test', 'job');
$query
->condition('name', 'Paul');
$query
->condition((new Condition('OR'))
->condition('age', 26)
->condition('age', 27));
$job = $query
->execute()
->fetchField();
$this
->assertEqual($job, 'Songwriter', 'Correct data retrieved.');
}
public function testJoinTwice() {
$query = $this->connection
->select('test')
->fields('test');
$alias = $query
->join('test', 'test', 'test.job = %alias.job');
$query
->addField($alias, 'name', 'othername');
$query
->addField($alias, 'job', 'otherjob');
$query
->where("{$alias}.name <> test.name");
$crowded_job = $query
->execute()
->fetch();
$this
->assertEqual($crowded_job->job, $crowded_job->otherjob, 'Correctly joined same table twice.');
$this
->assertNotEqual($crowded_job->name, $crowded_job->othername, 'Correctly joined same table twice.');
}
public function testJoinSubquery() {
$this
->installSchema('system', 'sequences');
$account = User::create([
'name' => $this
->randomMachineName(),
'mail' => $this
->randomMachineName() . '@example.com',
]);
$query = Database::getConnection('replica')
->select('test_task', 'tt');
$query
->addExpression('tt.pid + 1', 'abc');
$query
->condition('priority', 1, '>');
$query
->condition('priority', 100, '<');
$subquery = $this->connection
->select('test', 'tp');
$subquery
->join('test_one_blob', 'tpb', 'tp.id = tpb.id');
$subquery
->join('node', 'n', 'tp.id = n.nid');
$subquery
->addTag('node_access');
$subquery
->addMetaData('account', $account);
$subquery
->addField('tp', 'id');
$subquery
->condition('age', 5, '>');
$subquery
->condition('age', 500, '<');
$query
->leftJoin($subquery, 'sq', 'tt.pid = sq.id');
$query
->join('test_one_blob', 'tb3', 'tt.pid = tb3.id');
$query
->preExecute();
$query
->getArguments();
$str = (string) $query;
$pos = strpos($str, 'db_condition_placeholder_0', 0);
$pos2 = strpos($str, 'db_condition_placeholder_0', $pos + 1);
$this
->assertFalse($pos2, 'Condition placeholder is not repeated.');
}
public function testSelectWithRowCount() {
$query = $this->connection
->select('test');
$query
->addField('test', 'name');
$result = $query
->execute();
try {
$result
->rowCount();
$exception = FALSE;
} catch (RowCountException $e) {
$exception = TRUE;
}
$this
->assertTrue($exception, 'Exception was thrown');
}
public function testJoinConditionObject() {
$query = $this->connection
->select('test_task', 't');
$join_cond = (new Condition('AND'))
->where('t.pid = p.id');
$people_alias = $query
->join('test', 'p', $join_cond);
$name_field = $query
->addField($people_alias, 'name', 'name');
$query
->addField('t', 'task', 'task');
$priority_field = $query
->addField('t', 'priority', 'priority');
$query
->orderBy($priority_field);
$result = $query
->execute();
$num_records = 0;
$last_priority = 0;
foreach ($result as $record) {
$num_records++;
$this
->assertTrue($record->{$priority_field} >= $last_priority, 'Results returned in correct order.');
$this
->assertNotEqual($record->{$name_field}, 'Ringo', 'Taskless person not selected.');
$last_priority = $record->{$priority_field};
}
$this
->assertEqual($num_records, 7, 'Returned the correct number of rows.');
$t1_name = 'John';
$t2_name = 'George';
$join_cond = (new Condition('AND'))
->condition('t1.name', $t1_name)
->condition('t2.name', $t2_name);
$query = $this->connection
->select('test', 't1');
$query
->innerJoin('test', 't2', $join_cond);
$query
->addField('t1', 'name', 't1_name');
$query
->addField('t2', 'name', 't2_name');
$num_records = $query
->countQuery()
->execute()
->fetchField();
$this
->assertEqual($num_records, 1, 'Query expected to return 1 row. Actual: ' . $num_records);
if ($num_records == 1) {
$record = $query
->execute()
->fetchObject();
$this
->assertEqual($record->t1_name, $t1_name, 'Query expected to retrieve name ' . $t1_name . ' from table t1. Actual: ' . $record->t1_name);
$this
->assertEqual($record->t2_name, $t2_name, 'Query expected to retrieve name ' . $t2_name . ' from table t2. Actual: ' . $record->t2_name);
}
}
}