function SelectTest::testUnionCount

Tests that we can get a count query for a UNION Select query.

File

core/tests/Drupal/KernelTests/Core/Database/SelectTest.php, line 353

Class

SelectTest

Tests the Select query builder.

Namespace

Code

public function testUnionCount() {
    $query_1 = $this->connection
        ->select('test', 't')
        ->fields('t', [
        'name',
        'age',
    ])
        ->condition('age', [
        27,
        28,
    ], 'IN');
    $query_2 = $this->connection
        ->select('test', 't')
        ->fields('t', [
        'name',
        'age',
    ])
        ->condition('age', 28);
    $query_1->union($query_2, 'ALL');
    $names = $query_1->execute()
        ->fetchCol();
    $query_3 = $query_1->countQuery();
    $count = $query_3->execute()
        ->fetchField();
    // Ensure the counts match.
    $this->assertEqual(count($names), $count, "The count query's result matched the number of rows in the UNION query.");
}