database_test.module

<?php

/**
 * Implements hook_query_alter().
 */
function database_test_query_alter(QueryAlterableInterface $query) {
  if ($query
    ->hasTag('database_test_alter_add_range')) {
    $query
      ->range(0, 2);
  }
  if ($query
    ->hasTag('database_test_alter_add_join')) {
    $people_alias = $query
      ->join('test', 'people', "test_task.pid = %alias.id");
    $name_field = $query
      ->addField($people_alias, 'name', 'name');
    $query
      ->condition($people_alias . '.id', 2);
  }
  if ($query
    ->hasTag('database_test_alter_change_conditional')) {
    $conditions =& $query
      ->conditions();
    $conditions[0]['value'] = 2;
  }
  if ($query
    ->hasTag('database_test_alter_change_fields')) {
    $fields =& $query
      ->getFields();
    unset($fields['age']);
  }
  if ($query
    ->hasTag('database_test_alter_change_expressions')) {
    $expressions =& $query
      ->getExpressions();
    $expressions['double_age']['expression'] = 'age*3';
  }
}

/**
 * Implements hook_query_TAG_alter().
 *
 * Called by DatabaseTestCase::testAlterRemoveRange.
 */
function database_test_query_database_test_alter_remove_range_alter(QueryAlterableInterface $query) {
  $query
    ->range();
}

/**
 * Implements hook_menu().
 */
function database_test_menu() {
  $items['database_test/db_query_temporary'] = array(
    'access callback' => TRUE,
    'page callback' => 'database_test_db_query_temporary',
  );
  $items['database_test/pager_query_even'] = array(
    'access callback' => TRUE,
    'page callback' => 'database_test_even_pager_query',
  );
  $items['database_test/pager_query_odd'] = array(
    'access callback' => TRUE,
    'page callback' => 'database_test_odd_pager_query',
  );
  $items['database_test/tablesort'] = array(
    'access callback' => TRUE,
    'page callback' => 'database_test_tablesort',
  );
  $items['database_test/tablesort_first'] = array(
    'access callback' => TRUE,
    'page callback' => 'database_test_tablesort_first',
  );
  $items['database_test/tablesort_default_sort'] = array(
    'access callback' => TRUE,
    'page callback' => 'drupal_get_form',
    'page arguments' => array(
      'database_test_theme_tablesort',
    ),
  );
  return $items;
}

/**
 * Run a db_query_temporary and output the table name and its number of rows.
 *
 * We need to test that the table created is temporary, so we run it here, in a
 * separate menu callback request; After this request is done, the temporary
 * table should automatically dropped.
 */
function database_test_db_query_temporary() {
  $table_name = db_query_temporary('SELECT status FROM {system}', array());
  drupal_json_output(array(
    'table_name' => $table_name,
    'row_count' => db_select($table_name)
      ->countQuery()
      ->execute()
      ->fetchField(),
  ));
  exit;
}

/**
 * Run a pager query and return the results.
 *
 * This function does care about the page GET parameter, as set by the
 * simpletest HTTP call.
 */
function database_test_even_pager_query($limit) {
  $query = db_select('test', 't');
  $query
    ->fields('t', array(
    'name',
  ))
    ->orderBy('age');

  // This should result in 2 pages of results.
  $query = $query
    ->extend('PagerDefault')
    ->limit($limit);
  $names = $query
    ->execute()
    ->fetchCol();
  drupal_json_output(array(
    'names' => $names,
  ));
  exit;
}

/**
 * Run a pager query and return the results.
 *
 * This function does care about the page GET parameter, as set by the
 * simpletest HTTP call.
 */
function database_test_odd_pager_query($limit) {
  $query = db_select('test_task', 't');
  $query
    ->fields('t', array(
    'task',
  ))
    ->orderBy('pid');

  // This should result in 4 pages of results.
  $query = $query
    ->extend('PagerDefault')
    ->limit($limit);
  $names = $query
    ->execute()
    ->fetchCol();
  drupal_json_output(array(
    'names' => $names,
  ));
  exit;
}

/**
 * Run a tablesort query and return the results.
 *
 * This function does care about the page GET parameter, as set by the
 * simpletest HTTP call.
 */
function database_test_tablesort() {
  $header = array(
    'tid' => array(
      'data' => t('Task ID'),
      'field' => 'tid',
      'sort' => 'desc',
    ),
    'pid' => array(
      'data' => t('Person ID'),
      'field' => 'pid',
    ),
    'task' => array(
      'data' => t('Task'),
      'field' => 'task',
    ),
    'priority' => array(
      'data' => t('Priority'),
      'field' => 'priority',
    ),
  );
  $query = db_select('test_task', 't');
  $query
    ->fields('t', array(
    'tid',
    'pid',
    'task',
    'priority',
  ));
  $query = $query
    ->extend('TableSort')
    ->orderByHeader($header);

  // We need all the results at once to check the sort.
  $tasks = $query
    ->execute()
    ->fetchAll();
  drupal_json_output(array(
    'tasks' => $tasks,
  ));
  exit;
}

/**
 * Run a tablesort query with a second order_by after and return the results.
 *
 * This function does care about the page GET parameter, as set by the
 * simpletest HTTP call.
 */
function database_test_tablesort_first() {
  $header = array(
    'tid' => array(
      'data' => t('Task ID'),
      'field' => 'tid',
      'sort' => 'desc',
    ),
    'pid' => array(
      'data' => t('Person ID'),
      'field' => 'pid',
    ),
    'task' => array(
      'data' => t('Task'),
      'field' => 'task',
    ),
    'priority' => array(
      'data' => t('Priority'),
      'field' => 'priority',
    ),
  );
  $query = db_select('test_task', 't');
  $query
    ->fields('t', array(
    'tid',
    'pid',
    'task',
    'priority',
  ));
  $query = $query
    ->extend('TableSort')
    ->orderByHeader($header)
    ->orderBy('priority');

  // We need all the results at once to check the sort.
  $tasks = $query
    ->execute()
    ->fetchAll();
  drupal_json_output(array(
    'tasks' => $tasks,
  ));
  exit;
}

/**
 * Output a form without setting a header sort.
 */
function database_test_theme_tablesort($form, &$form_state) {
  $header = array(
    'username' => array(
      'data' => t('Username'),
      'field' => 'u.name',
    ),
    'status' => array(
      'data' => t('Status'),
      'field' => 'u.status',
    ),
  );
  $query = db_select('users', 'u');
  $query
    ->condition('u.uid', 0, '<>');
  user_build_filter_query($query);
  $count_query = clone $query;
  $count_query
    ->addExpression('COUNT(u.uid)');
  $query = $query
    ->extend('PagerDefault')
    ->extend('TableSort');
  $query
    ->fields('u', array(
    'uid',
    'name',
    'status',
    'created',
    'access',
  ))
    ->limit(50)
    ->orderByHeader($header)
    ->setCountQuery($count_query);
  $result = $query
    ->execute();
  $options = array();
  $status = array(
    t('blocked'),
    t('active'),
  );
  $accounts = array();
  foreach ($result as $account) {
    $options[$account->uid] = array(
      'username' => check_plain($account->name),
      'status' => $status[$account->status],
    );
  }
  $form['accounts'] = array(
    '#type' => 'tableselect',
    '#header' => $header,
    '#options' => $options,
    '#empty' => t('No people available.'),
  );
  return $form;
}