Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\system\Tests\System;
Chris@0: 
Chris@0: use Drupal\Core\Flood\DatabaseBackend;
Chris@0: use Drupal\Core\Flood\MemoryBackend;
Chris@0: use Drupal\simpletest\WebTestBase;
Chris@0: use Symfony\Component\HttpFoundation\Request;
Chris@0: 
Chris@0: /**
Chris@0:  * Functional tests for the flood control mechanism.
Chris@0:  *
Chris@0:  * @group system
Chris@0:  */
Chris@0: class FloodTest extends WebTestBase {
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function setUp() {
Chris@0:     parent::setUp();
Chris@0: 
Chris@0:     // Flood backends need a request object. Create a dummy one and insert it
Chris@0:     // to the container.
Chris@0:     $request = Request::createFromGlobals();
Chris@0:     $this->container->get('request_stack')->push($request);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Test flood control mechanism clean-up.
Chris@0:    */
Chris@0:   public function testCleanUp() {
Chris@0:     $threshold = 1;
Chris@0:     $window_expired = -1;
Chris@0:     $name = 'flood_test_cleanup';
Chris@0: 
Chris@0:     $flood = \Drupal::flood();
Chris@0:     $this->assertTrue($flood->isAllowed($name, $threshold));
Chris@0:     // Register expired event.
Chris@0:     $flood->register($name, $window_expired);
Chris@0:     // Verify event is not allowed.
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:     // Run cron and verify event is now allowed.
Chris@0:     $this->cronRun();
Chris@0:     $this->assertTrue($flood->isAllowed($name, $threshold));
Chris@0: 
Chris@0:     // Register unexpired event.
Chris@0:     $flood->register($name);
Chris@0:     // Verify event is not allowed.
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:     // Run cron and verify event is still not allowed.
Chris@0:     $this->cronRun();
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Test flood control memory backend.
Chris@0:    */
Chris@0:   public function testMemoryBackend() {
Chris@0:     $threshold = 1;
Chris@0:     $window_expired = -1;
Chris@0:     $name = 'flood_test_cleanup';
Chris@0: 
Chris@0:     $request_stack = \Drupal::service('request_stack');
Chris@0:     $flood = new MemoryBackend($request_stack);
Chris@0:     $this->assertTrue($flood->isAllowed($name, $threshold));
Chris@0:     // Register expired event.
Chris@0:     $flood->register($name, $window_expired);
Chris@0:     // Verify event is not allowed.
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:     // Run cron and verify event is now allowed.
Chris@0:     $flood->garbageCollection();
Chris@0:     $this->assertTrue($flood->isAllowed($name, $threshold));
Chris@0: 
Chris@0:     // Register unexpired event.
Chris@0:     $flood->register($name);
Chris@0:     // Verify event is not allowed.
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:     // Run cron and verify event is still not allowed.
Chris@0:     $flood->garbageCollection();
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Test flood control database backend.
Chris@0:    */
Chris@0:   public function testDatabaseBackend() {
Chris@0:     $threshold = 1;
Chris@0:     $window_expired = -1;
Chris@0:     $name = 'flood_test_cleanup';
Chris@0: 
Chris@0:     $connection = \Drupal::service('database');
Chris@0:     $request_stack = \Drupal::service('request_stack');
Chris@0:     $flood = new DatabaseBackend($connection, $request_stack);
Chris@0:     $this->assertTrue($flood->isAllowed($name, $threshold));
Chris@0:     // Register expired event.
Chris@0:     $flood->register($name, $window_expired);
Chris@0:     // Verify event is not allowed.
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:     // Run cron and verify event is now allowed.
Chris@0:     $flood->garbageCollection();
Chris@0:     $this->assertTrue($flood->isAllowed($name, $threshold));
Chris@0: 
Chris@0:     // Register unexpired event.
Chris@0:     $flood->register($name);
Chris@0:     // Verify event is not allowed.
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:     // Run cron and verify event is still not allowed.
Chris@0:     $flood->garbageCollection();
Chris@0:     $this->assertFalse($flood->isAllowed($name, $threshold));
Chris@0:   }
Chris@0: 
Chris@0: }