aboutsummaryrefslogtreecommitdiffstats
path: root/tests/phpunit/unit/includes/editpage/Constraint/LinkPurgeRateLimitConstraintTest.php
blob: 7363601e0f7218b9ad76627f0cdafc9103df4ff7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
<?php
/**
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 * http://www.gnu.org/copyleft/gpl.html
 *
 * @file
 */

use MediaWiki\EditPage\Constraint\IEditConstraint;
use MediaWiki\EditPage\Constraint\LinkPurgeRateLimitConstraint;
use MediaWiki\Permissions\RateLimiter;
use MediaWiki\Permissions\RateLimitSubject;
use MediaWiki\User\UserIdentityValue;
use PHPUnit\Framework\MockObject\MockObject;

/**
 * Tests the LinkPurgeRateLimitConstraint
 *
 * @author DannyS712
 *
 * @covers \MediaWiki\EditPage\Constraint\LinkPurgeRateLimitConstraint
 */
class LinkPurgeRateLimitConstraintTest extends MediaWikiUnitTestCase {
	use EditConstraintTestTrait;

	/**
	 * @param bool $fail
	 *
	 * @return MockObject&RateLimiter
	 */
	private function getRateLimiter( $fail ) {
		$mock = $this->createNoOpMock( RateLimiter::class, [ 'limit' ] );
		$mock->expects( $this->once() )
			->method( 'limit' )
			->with( self::anything(), 'linkpurge', 0 )
			->willReturn( $fail );
		return $mock;
	}

	public function testPass() {
		$limiter = $this->getRateLimiter( false );

		$subject = new RateLimitSubject( new UserIdentityValue( 1, 'test' ), null, [] );

		$constraint = new LinkPurgeRateLimitConstraint( $limiter, $subject );
		$this->assertConstraintPassed( $constraint );
	}

	public function testFailure() {
		$limiter = $this->getRateLimiter( true );

		$subject = new RateLimitSubject( new UserIdentityValue( 1, 'test' ), null, [] );

		$constraint = new LinkPurgeRateLimitConstraint( $limiter, $subject );
		$this->assertConstraintFailed( $constraint, IEditConstraint::AS_RATE_LIMITED );
	}

}