IpAddressBlockingTest.php in Drupal 8

Namespace

Drupal\Tests\ban\Functional

File

core/modules/ban/tests/src/Functional/IpAddressBlockingTest.php

<?php

namespace Drupal\Tests\ban\Functional;

use Drupal\Tests\BrowserTestBase;
use Drupal\Core\Database\Database;
use Drupal\ban\BanIpManager;

/**
 * Tests IP address banning.
 *
 * @group ban
 */
class IpAddressBlockingTest extends BrowserTestBase {

  /**
   * Modules to install.
   *
   * @var array
   */
  public static $modules = [
    'ban',
  ];

  /**
   * {@inheritdoc}
   */
  protected $defaultTheme = 'stark';

  /**
   * Tests various user input to confirm correct validation and saving of data.
   */
  public function testIPAddressValidation() {

    // Create user.
    $admin_user = $this
      ->drupalCreateUser([
      'ban IP addresses',
    ]);
    $this
      ->drupalLogin($admin_user);
    $this
      ->drupalGet('admin/config/people/ban');
    $connection = Database::getConnection();

    // Ban a valid IP address.
    $edit = [];
    $edit['ip'] = '1.2.3.3';
    $this
      ->drupalPostForm('admin/config/people/ban', $edit, t('Add'));
    $ip = $connection
      ->select('ban_ip', 'bi')
      ->fields('bi', [
      'iid',
    ])
      ->condition('ip', $edit['ip'])
      ->execute()
      ->fetchField();
    $this
      ->assertNotEmpty($ip, 'IP address found in database.');
    $this
      ->assertRaw(t('The IP address %ip has been banned.', [
      '%ip' => $edit['ip'],
    ]), 'IP address was banned.');

    // Try to block an IP address that's already blocked.
    $edit = [];
    $edit['ip'] = '1.2.3.3';
    $this
      ->drupalPostForm('admin/config/people/ban', $edit, t('Add'));
    $this
      ->assertText(t('This IP address is already banned.'));

    // Try to block a reserved IP address.
    $edit = [];
    $edit['ip'] = '255.255.255.255';
    $this
      ->drupalPostForm('admin/config/people/ban', $edit, t('Add'));
    $this
      ->assertText(t('Enter a valid IP address.'));

    // Try to block a reserved IP address.
    $edit = [];
    $edit['ip'] = 'test.example.com';
    $this
      ->drupalPostForm('admin/config/people/ban', $edit, t('Add'));
    $this
      ->assertText(t('Enter a valid IP address.'));

    // Submit an empty form.
    $edit = [];
    $edit['ip'] = '';
    $this
      ->drupalPostForm('admin/config/people/ban', $edit, t('Add'));
    $this
      ->assertText(t('Enter a valid IP address.'));

    // Pass an IP address as a URL parameter and submit it.
    $submit_ip = '1.2.3.4';
    $this
      ->drupalPostForm('admin/config/people/ban/' . $submit_ip, [], t('Add'));
    $ip = $connection
      ->select('ban_ip', 'bi')
      ->fields('bi', [
      'iid',
    ])
      ->condition('ip', $submit_ip)
      ->execute()
      ->fetchField();
    $this
      ->assertNotEmpty($ip, 'IP address found in database');
    $this
      ->assertRaw(t('The IP address %ip has been banned.', [
      '%ip' => $submit_ip,
    ]), 'IP address was banned.');

    // Submit your own IP address. This fails, although it works when testing
    // manually.
    // TODO: On some systems this test fails due to a bug/inconsistency in cURL.
    // $edit = array();
    // $edit['ip'] = \Drupal::request()->getClientIP();
    // $this->drupalPostForm('admin/config/people/ban', $edit, t('Save'));
    // $this->assertText(t('You may not ban your own IP address.'));
    // Test duplicate ip address are not present in the 'blocked_ips' table.
    // when they are entered programmatically.
    $banIp = new BanIpManager($connection);
    $ip = '1.0.0.0';
    $banIp
      ->banIp($ip);
    $banIp
      ->banIp($ip);
    $banIp
      ->banIp($ip);
    $query = $connection
      ->select('ban_ip', 'bip');
    $query
      ->fields('bip', [
      'iid',
    ]);
    $query
      ->condition('bip.ip', $ip);
    $ip_count = $query
      ->execute()
      ->fetchAll();
    $this
      ->assertCount(1, $ip_count);
    $ip = '';
    $banIp
      ->banIp($ip);
    $banIp
      ->banIp($ip);
    $query = $connection
      ->select('ban_ip', 'bip');
    $query
      ->fields('bip', [
      'iid',
    ]);
    $query
      ->condition('bip.ip', $ip);
    $ip_count = $query
      ->execute()
      ->fetchAll();
    $this
      ->assertCount(1, $ip_count);
  }

}

Classes

Name	Description
IpAddressBlockingTest	Tests IP address banning.