DeleteTruncateTest.php 2.53 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
<?php

/**
 * @file
 * Definition of Drupal\system\Tests\Database\DeleteTruncateTest.
 */

namespace Drupal\system\Tests\Database;

/**
 * Delete/Truncate tests.
 *
 * The DELETE tests are not as extensive, as all of the interesting code for
 * DELETE queries is in the conditional which is identical to the UPDATE and
 * SELECT conditional handling.
 *
 * The TRUNCATE tests are not extensive either, because the behavior of
 * TRUNCATE queries is not consistent across database engines. We only test
 * that a TRUNCATE query actually deletes all rows from the target table.
 */
class DeleteTruncateTest extends DatabaseTestBase {

  public static function getInfo() {
    return array(
      'name' => 'Delete/Truncate tests',
      'description' => 'Test the Delete and Truncate query builders.',
      'group' => 'Database',
    );
  }

  /**
32
   * Confirms that we can use a subselect in a delete successfully.
33 34 35 36 37 38 39 40 41 42 43 44 45
   */
  function testSubselectDelete() {
    $num_records_before = db_query('SELECT COUNT(*) FROM {test_task}')->fetchField();
    $pid_to_delete = db_query("SELECT * FROM {test_task} WHERE task = 'sleep'")->fetchField();

    $subquery = db_select('test', 't')
      ->fields('t', array('id'))
      ->condition('t.id', array($pid_to_delete), 'IN');
    $delete = db_delete('test_task')
      ->condition('task', 'sleep')
      ->condition('pid', $subquery, 'IN');

    $num_deleted = $delete->execute();
46
    $this->assertEqual($num_deleted, 1, 'Deleted 1 record.');
47 48

    $num_records_after = db_query('SELECT COUNT(*) FROM {test_task}')->fetchField();
49
    $this->assertEqual($num_records_before, $num_records_after + $num_deleted, 'Deletion adds up.');
50 51 52
  }

  /**
53
   * Confirms that we can delete a single record successfully.
54 55 56 57 58 59 60
   */
  function testSimpleDelete() {
    $num_records_before = db_query('SELECT COUNT(*) FROM {test}')->fetchField();

    $num_deleted = db_delete('test')
      ->condition('id', 1)
      ->execute();
61
    $this->assertIdentical($num_deleted, 1, 'Deleted 1 record.');
62 63

    $num_records_after = db_query('SELECT COUNT(*) FROM {test}')->fetchField();
64
    $this->assertEqual($num_records_before, $num_records_after + $num_deleted, 'Deletion adds up.');
65 66 67
  }

  /**
68
   * Confirms that we can truncate a whole table successfully.
69 70 71 72 73 74 75
   */
  function testTruncate() {
    $num_records_before = db_query("SELECT COUNT(*) FROM {test}")->fetchField();

    db_truncate('test')->execute();

    $num_records_after = db_query("SELECT COUNT(*) FROM {test}")->fetchField();
76
    $this->assertEqual(0, $num_records_after, 'Truncate really deletes everything.');
77 78
  }
}