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
71
72
73
74
75
76
77
78
79
80
|
<?php
namespace Illuminate\Tests\Cache;
use Illuminate\Cache\ArrayStore;
use Illuminate\Cache\RateLimiter;
use Illuminate\Cache\RateLimiting\Limit;
use Illuminate\Cache\Repository;
use Illuminate\Contracts\Cache\Repository as Cache;
use PHPUnit\Framework\Attributes\DataProvider;
use PHPUnit\Framework\TestCase;
use ReflectionProperty;
class RateLimiterTest extends TestCase
{
public static function registerNamedRateLimiterDataProvider(): array
{
return [
'uses BackedEnum' => [BackedEnumNamedRateLimiter::API, 'api'],
'uses UnitEnum' => [UnitEnumNamedRateLimiter::THIRD_PARTY, 'THIRD_PARTY'],
'uses normal string' => ['yolo', 'yolo'],
'uses int' => [100, '100'],
];
}
#[DataProvider('registerNamedRateLimiterDataProvider')]
public function testRegisterNamedRateLimiter(mixed $name, string $expected): void
{
$reflectedLimitersProperty = new ReflectionProperty(RateLimiter::class, 'limiters');
$reflectedLimitersProperty->setAccessible(true);
$rateLimiter = new RateLimiter($this->createMock(Cache::class));
$rateLimiter->for($name, fn () => Limit::perMinute(100));
$limiters = $reflectedLimitersProperty->getValue($rateLimiter);
$this->assertArrayHasKey($expected, $limiters);
$limiterClosure = $rateLimiter->limiter($name);
$this->assertNotNull($limiterClosure);
}
public function testShouldUseOriginKeyAsPrefixWhenMultipleLimiterWithSameKey()
{
$rateLimiter = new RateLimiter(new Repository(new ArrayStore));
$rateLimiter->for('user_limiter', fn (string $userId) => [
Limit::perSecond(3)->by($userId),
Limit::perMinute(5)->by($userId),
]);
$userId1 = '123';
$userId2 = '456';
$limiterForUser1 = $rateLimiter->limiter('user_limiter')($userId1);
$limiterForUser2 = $rateLimiter->limiter('user_limiter')($userId2);
for ($i = 0; $i < 3; $i++) {
$this->assertFalse($rateLimiter->tooManyAttempts($limiterForUser1[0]->key, $limiterForUser1[0]->maxAttempts));
$this->assertFalse($rateLimiter->tooManyAttempts($limiterForUser2[0]->key, $limiterForUser2[0]->maxAttempts));
$rateLimiter->hit($limiterForUser1[0]->key, $limiterForUser1[0]->decaySeconds);
$rateLimiter->hit($limiterForUser2[0]->key, $limiterForUser2[0]->decaySeconds);
}
$this->assertNotSame($limiterForUser1[0]->key, $limiterForUser2[0]->key);
$this->assertNotSame($limiterForUser1[1]->key, $limiterForUser2[1]->key);
}
}
enum BackedEnumNamedRateLimiter: string
{
case API = 'api';
}
enum UnitEnumNamedRateLimiter
{
case THIRD_PARTY;
}
|
