redis-8.x-1.x-dev/tests/src/Kernel/RedisFloodTest.php

<?php

namespace Drupal\Tests\redis\Kernel;

use Drupal\KernelTests\KernelTestBase;
use Drupal\redis\Flood\FloodFactory;
use Drupal\Tests\redis\Traits\RedisTestInterfaceTrait;

/**
 * Tests Redis flood backend.
 *
 * @group redis
 */
class RedisFloodTest extends KernelTestBase {

  use RedisTestInterfaceTrait;

  /**
   * Modules to enable.
   *
   * @var array
   */
  protected static $modules = ['redis'];

  /**
   * Test flood control.
   */
  public function testFlood() {
    self::setUpSettings();
    $threshold = 2;
    $window = 1;
    $name = 'flood_test_cleanup';

    $client_factory = \Drupal::service('redis.factory');
    $request_stack = \Drupal::service('request_stack');
    $flood_factory = new FloodFactory($client_factory, $request_stack);
    $flood = $flood_factory->get();

    // By default the event is allowed.
    $this->assertTrue($flood->isAllowed($name, $threshold));

    // Register event.
    $flood->register($name, $window);

    // The event is still allowed.
    $this->assertTrue($flood->isAllowed($name, $threshold));

    $flood->register($name, $window);

    // Verify event is not allowed.
    $this->assertFalse($flood->isAllowed($name, $threshold));

    // "Sleep" two seconds, then the event is allowed again.
    $_SERVER['REQUEST_TIME'] += 2;
    $this->assertTrue($flood->isAllowed($name, $threshold));

  }

}