Proyectos de Subversion Moodle

<?php

// This file is part of Moodle - http://moodle.org/

//

// Moodle is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

//

// Moodle is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

// GNU General Public License for more details.

//

// You should have received a copy of the GNU General Public License

// along with Moodle.  If not, see <http://www.gnu.org/licenses/>.

namespace core_ai;

defined('MOODLE_INTERNAL') || die();

global $CFG;

require_once($CFG->libdir . '/testing/classes/incrementing_clock.php');

/**

 * Test ai subsystem rate limiter methods.

 *

 * @package    core_ai

 * @copyright  2024 Matt Porritt

 * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later

 * @covers     \core_ai\rate_limiter

 */

final class rate_limiter_test extends \advanced_testcase {

    /**

     * Set up before tests.

     */

    protected function setUp(): void {

        parent::setUp();

        $this->resetAfterTest(true);

    }

    /**

     * Test global rate limit is enforced correctly.

     */

    public function test_global_rate_limit(): void {

        $ratelimiter = \core\di::get(rate_limiter::class);

        $component = 'testcomponent';

        $ratelimit = 5;

        // Make 5 requests, all should be allowed.

        for ($i = 0; $i < 5; $i++) {

            $this->assertTrue($ratelimiter->check_global_rate_limit($component, $ratelimit));

        }

        // The 6th request should be denied.

        $this->assertFalse($ratelimiter->check_global_rate_limit($component, $ratelimit));

    }

    /**

     * Test user rate limit is enforced correctly.

     */

    public function test_user_rate_limit(): void {

        $ratelimiter = \core\di::get(rate_limiter::class);

        $component = 'testcomponent';

        $ratelimit = 3;

        $userid = 12345;

        // Make 3 requests for the user, all should be allowed.

        for ($i = 0; $i < 3; $i++) {

            $this->assertTrue($ratelimiter->check_user_rate_limit($component, $ratelimit, $userid));

        }

        // The 4th request should be denied.

        $this->assertFalse($ratelimiter->check_user_rate_limit($component, $ratelimit, $userid));

    }

    /**

     * Test global rate limit resets after time window.

     */

    public function test_global_rate_limit_reset(): void {

        $clock = $this->mock_clock_with_incrementing(0);

        $ratelimiter = \core\di::get(rate_limiter::class);

        $component = 'testcomponent';

        $ratelimit = 3;

        // Make 3 requests, all should be allowed.

        for ($i = 0; $i < 3; $i++) {

            $this->assertTrue($ratelimiter->check_global_rate_limit($component, $ratelimit));

        }

        // Simulate moving time forward by TIME_WINDOW to reset the rate limit.

        $clock->set_to(rate_limiter::TIME_WINDOW + 1);

        // The next request should be allowed again.

        $this->assertTrue($ratelimiter->check_global_rate_limit($component, $ratelimit));

    }

    /**

     * Test user rate limit resets after time window.

     */

    public function test_user_rate_limit_reset(): void {

        $clock = $this->mock_clock_with_incrementing(0);

        $ratelimiter = \core\di::get(rate_limiter::class);

        $component = 'testcomponent';

        $ratelimit = 3;

        $userid = 12345;

        // Make 3 requests for the user, all should be allowed.

        for ($i = 0; $i < 3; $i++) {

            $this->assertTrue($ratelimiter->check_user_rate_limit($component, $ratelimit, $userid));

        }

        // Simulate moving time forward by TIME_WINDOW to reset the rate limit.

        $clock->set_to(rate_limiter::TIME_WINDOW + 1);

        // The next user request should be allowed again.

        $this->assertTrue($ratelimiter->check_user_rate_limit($component, $ratelimit, $userid));

    }

}