TemporaryQueryTest.php

<?php

/**
 * @file
 * Definition of Drupal\system\Tests\Database\TemporaryQueryTest.
 */
namespace Drupal\system\Tests\Database;


/**
 * Tests temporary queries.
 */
class TemporaryQueryTest extends DatabaseTestBase {

  /**
   * Modules to enable.
   *
   * @var array
   */
  public static $modules = array(
    'database_test',
  );
  public static function getInfo() {
    return array(
      'name' => 'Temporary query test',
      'description' => 'Test the temporary query functionality.',
      'group' => 'Database',
    );
  }

  /**
   * Returns the number of rows of a table.
   */
  function countTableRows($table_name) {
    return db_select($table_name)
      ->countQuery()
      ->execute()
      ->fetchField();
  }

  /**
   * Confirms that temporary tables work and are limited to one request.
   */
  function testTemporaryQuery() {
    $this
      ->drupalGet('database_test/db_query_temporary');
    $data = json_decode($this
      ->drupalGetContent());
    if ($data) {
      $this
        ->assertEqual($this
        ->countTableRows('test'), $data->row_count, 'The temporary table contains the correct amount of rows.');
      $this
        ->assertFalse(db_table_exists($data->table_name), 'The temporary table is, indeed, temporary.');
    }
    else {
      $this
        ->fail('The creation of the temporary table failed.');
    }

    // Now try to run two db_query_temporary() in the same request.
    $table_name_test = db_query_temporary('SELECT name FROM {test}', array());
    $table_name_task = db_query_temporary('SELECT pid FROM {test_task}', array());
    $this
      ->assertEqual($this
      ->countTableRows($table_name_test), $this
      ->countTableRows('test'), 'A temporary table was created successfully in this request.');
    $this
      ->assertEqual($this
      ->countTableRows($table_name_task), $this
      ->countTableRows('test_task'), 'A second temporary table was created successfully in this request.');
  }

}