Club Drupal

  /**

   * Confirm that an extended query has a "tag" added to it.

   */

  function testExtenderHasTag() {

    $query = db_select('test')

      ->extend('SelectQueryExtender');

    $query->addField('test', 'name');

    $query->addField('test', 'age', 'age');

    $query->addTag('test');

    $this->assertTrue($query->hasTag('test'), 'hasTag() returned true.');

    $this->assertFalse($query->hasTag('other'), 'hasTag() returned false.');

  }

  /**

   * Test extended query tagging "has all of these tags" functionality.

   */

  function testExtenderHasAllTags() {

    $query = db_select('test')

      ->extend('SelectQueryExtender');

    $query->addField('test', 'name');

    $query->addField('test', 'age', 'age');

    $query->addTag('test');

    $query->addTag('other');

    $this->assertTrue($query->hasAllTags('test', 'other'), 'hasAllTags() returned true.');

    $this->assertFalse($query->hasAllTags('test', 'stuff'), 'hasAllTags() returned false.');

  }

  /**

   * Test extended query tagging "has at least one of these tags" functionality.

   */

  function testExtenderHasAnyTag() {

    $query = db_select('test')

      ->extend('SelectQueryExtender');

    $query->addField('test', 'name');

    $query->addField('test', 'age', 'age');

    $query->addTag('test');

    $this->assertTrue($query->hasAnyTag('test', 'other'), 'hasAnyTag() returned true.');

    $this->assertFalse($query->hasAnyTag('other', 'stuff'), 'hasAnyTag() returned false.');

  }

    // Check that leading whitespace and comments do not cause problems

    // in the modified query.

    $sql = "

      -- Let's select some rows into a temporary table

      SELECT name FROM {test}

    ";

    $table_name_test = db_query_temporary($sql, array());

    $this->assertEqual($this->countTableRows($table_name_test), $this->countTableRows('test'), 'Leading white space and comments do not interfere with temporary table creation.');