drupal/modules/statistics/statistics.test

<?php
// $Id$

class StatisticsBlockVisitorsTestCase extends DrupalWebTestCase {
  /**
   * Implementation of getInfo().
   */
  function getInfo() {
    return array(
      'name' => t('Top visitor blocking'),
      'description' => t('Tests blocking of IP addresses via the top visitors report.'),
      'group' => t('Statistics')
    );
  }

  /**
   * Implementation of setUp().
   */
  function setUp() {
    parent::setUp('statistics');

    // Create user.
    $this->blocking_user = $this->drupalCreateUser(array('block IP addresses', 'access statistics'));

    // Insert dummy access by anonymous user into access log.
    db_query("INSERT INTO {accesslog} (title, path, url, hostname, uid, sid, timer, timestamp) values('%s', '%s', '%s', '%s', %d, '%s', %d, %d)", 'test', 'node/1', 'http://example.com', '192.168.1.1', '0', '10', '10', REQUEST_TIME);
  }

  /**
   * Blocks an IP address via the top visitors report then uses the same page to unblock it.
   */
  function testIPAddressBlocking() {
    // IP address for testing.
    $test_ip_address = '192.168.1.1';

    // Enable access logging (redundant since we insert the data manually).
    variable_set('statistics_enable_access_log', 1);

    // Verify the IP address from accesslog appears on the top visitors page
    // and that a 'block IP adddress' link is displayed.
    $this->drupalLogin($this->blocking_user);
    $this->drupalGet('admin/reports/visitors');
    $this->assertText($test_ip_address, t('IP address found.'));
    $this->assertText(t('block IP address'), t('Block IP link displayed'));

    // Block the IP address.
    $this->clickLink('block IP address');
    $this->assertText(t('IP address blocking'), t('IP blocking page displayed.'));
    $edit = array();
    $edit['ip'] = $test_ip_address;
    $this->drupalPost('admin/settings/ip-blocking', $edit, t('Save'));
    $ip = db_result(db_query("SELECT iid from {blocked_ips} WHERE ip = '%s'", $edit['ip']));
    $this->assertNotNull($ip, t('IP address found in database'));
    $this->assertRaw(t('The IP address %ip has been blocked.', array('%ip' => $edit['ip'])), t('IP address was blocked.'));

    // Verify that the block/unblock link on the top visitors page has been altered.
    $this->drupalGet('admin/reports/visitors');
    $this->assertText(t('unblock IP address'), t('Unblock IP address link displayed'));

    // Unblock the IP address.
    $this->clickLink('unblock IP address');
    $this->assertRaw(t('Are you sure you want to delete %ip?', array('%ip' => $test_ip_address)), t('IP address deletion confirmation found.'));
    $edit = array();
    $this->drupalPost('admin/settings/ip-blocking/delete/1', NULL, t('Delete'));
    $this->assertRaw(t('The IP address %ip was deleted.', array('%ip' => $test_ip_address)), t('IP address deleted.'));
  }
}