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
|
<?php
namespace Illuminate\Tests\Cache;
use Illuminate\Cache\RateLimiter;
use Illuminate\Contracts\Cache\Repository as Cache;
use Mockery as m;
use PHPUnit\Framework\TestCase;
class CacheRateLimiterTest extends TestCase
{
protected function tearDown(): void
{
m::close();
}
public function testTooManyAttemptsReturnTrueIfAlreadyLockedOut()
{
$cache = m::mock(Cache::class);
$cache->shouldReceive('get')->once()->with('key', 0)->andReturn(1);
$cache->shouldReceive('has')->once()->with('key:timer')->andReturn(true);
$cache->shouldReceive('add')->never();
$rateLimiter = new RateLimiter($cache);
$this->assertTrue($rateLimiter->tooManyAttempts('key', 1));
}
public function testHitProperlyIncrementsAttemptCount()
{
$cache = m::mock(Cache::class);
$cache->shouldReceive('add')->once()->with('key:timer', m::type('int'), 1)->andReturn(true);
$cache->shouldReceive('add')->once()->with('key', 0, 1)->andReturn(true);
$cache->shouldReceive('increment')->once()->with('key')->andReturn(1);
$rateLimiter = new RateLimiter($cache);
$rateLimiter->hit('key', 1);
}
public function testHitHasNoMemoryLeak()
{
$cache = m::mock(Cache::class);
$cache->shouldReceive('add')->once()->with('key:timer', m::type('int'), 1)->andReturn(true);
$cache->shouldReceive('add')->once()->with('key', 0, 1)->andReturn(false);
$cache->shouldReceive('increment')->once()->with('key')->andReturn(1);
$cache->shouldReceive('put')->once()->with('key', 1, 1);
$rateLimiter = new RateLimiter($cache);
$rateLimiter->hit('key', 1);
}
public function testRetriesLeftReturnsCorrectCount()
{
$cache = m::mock(Cache::class);
$cache->shouldReceive('get')->once()->with('key', 0)->andReturn(3);
$rateLimiter = new RateLimiter($cache);
$this->assertEquals(2, $rateLimiter->retriesLeft('key', 5));
}
public function testClearClearsTheCacheKeys()
{
$cache = m::mock(Cache::class);
$cache->shouldReceive('forget')->once()->with('key');
$cache->shouldReceive('forget')->once()->with('key:timer');
$rateLimiter = new RateLimiter($cache);
$rateLimiter->clear('key');
}
}
|
