BanIpTest.php

<?php

declare (strict_types=1);
namespace Drupal\Tests\ban\Kernel;

use Drupal\ban\BanIpManagerInterface;
use Drupal\Core\Database\Connection;
use Drupal\KernelTests\KernelTestBase;

/**
 * @group ban
 */
class BanIpTest extends KernelTestBase {
  
  /**
   * {@inheritdoc}
   */
  protected static $modules = [
    'ban',
  ];
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->installSchema('ban', [
      'ban_ip',
    ]);
  }
  
  /**
   * Test banning IPs.
   */
  public function testBanIp() : void {
    $banIp = $this->container
      ->get(BanIpManagerInterface::class);
    // Test valid IP addresses.
    $ip = '1.2.3.3';
    $this->assertCount(0, $this->getIpBans($ip));
    $banIp->banIp($ip);
    $this->assertCount(1, $this->getIpBans($ip));
    // Test duplicate ip address are not present in the 'blocked_ips' table
    // when they are entered programmatically.
    $ip = '1.0.0.0';
    $banIp->banIp($ip);
    $banIp->banIp($ip);
    $banIp->banIp($ip);
    $this->assertCount(1, $this->getIpBans($ip));
    $ip = '';
    $banIp->banIp($ip);
    $banIp->banIp($ip);
    $this->assertCount(1, $this->getIpBans($ip));
  }
  
  /**
   * Gets the IP bans.
   */
  protected function getIpBans(string $ip) : array {
    $connection = $this->container
      ->get(Connection::class);
    $query = $connection->select('ban_ip', 'bip');
    $query->fields('bip', [
      'iid',
    ]);
    $query->condition('bip.ip', $ip);
    return $query->execute()
      ->fetchAll();
  }

}