|
|
|
|
@ -12,21 +12,26 @@ namespace Test\Security\RateLimiting;
|
|
|
|
|
use OC\Security\RateLimiting\Backend\IBackend;
|
|
|
|
|
use OC\Security\RateLimiting\Limiter;
|
|
|
|
|
use OCP\IUser;
|
|
|
|
|
use OCP\Security\RateLimiting\ILimiter;
|
|
|
|
|
use PHPUnit\Framework\MockObject\MockObject;
|
|
|
|
|
use Psr\Log\LoggerInterface;
|
|
|
|
|
use Test\TestCase;
|
|
|
|
|
|
|
|
|
|
class LimiterTest extends TestCase {
|
|
|
|
|
/** @var IBackend|\PHPUnit\Framework\MockObject\MockObject */
|
|
|
|
|
private $backend;
|
|
|
|
|
/** @var Limiter */
|
|
|
|
|
private $limiter;
|
|
|
|
|
|
|
|
|
|
private IBackend&MockObject $backend;
|
|
|
|
|
private ILimiter $limiter;
|
|
|
|
|
private LoggerInterface $logger;
|
|
|
|
|
|
|
|
|
|
protected function setUp(): void {
|
|
|
|
|
parent::setUp();
|
|
|
|
|
|
|
|
|
|
$this->backend = $this->createMock(IBackend::class);
|
|
|
|
|
$this->logger = $this->createMock(LoggerInterface::class);
|
|
|
|
|
|
|
|
|
|
$this->limiter = new Limiter(
|
|
|
|
|
$this->backend
|
|
|
|
|
$this->backend,
|
|
|
|
|
$this->logger,
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@ -43,6 +48,8 @@ class LimiterTest extends TestCase {
|
|
|
|
|
'4664f0d9c88dcb7552be47b37bb52ce35977b2e60e1ac13757cf625f31f87050a41f3da064887fa87d49fd042e4c8eb20de8f10464877d3959677ab011b73a47'
|
|
|
|
|
)
|
|
|
|
|
->willReturn(101);
|
|
|
|
|
$this->logger->expects($this->once())
|
|
|
|
|
->method('info');
|
|
|
|
|
|
|
|
|
|
$this->limiter->registerAnonRequest('MyIdentifier', 100, 100, '127.0.0.1');
|
|
|
|
|
}
|
|
|
|
|
@ -64,6 +71,8 @@ class LimiterTest extends TestCase {
|
|
|
|
|
'4664f0d9c88dcb7552be47b37bb52ce35977b2e60e1ac13757cf625f31f87050a41f3da064887fa87d49fd042e4c8eb20de8f10464877d3959677ab011b73a47',
|
|
|
|
|
100
|
|
|
|
|
);
|
|
|
|
|
$this->logger->expects($this->never())
|
|
|
|
|
->method('info');
|
|
|
|
|
|
|
|
|
|
$this->limiter->registerAnonRequest('MyIdentifier', 100, 100, '127.0.0.1');
|
|
|
|
|
}
|
|
|
|
|
@ -87,6 +96,8 @@ class LimiterTest extends TestCase {
|
|
|
|
|
'ddb2ec50fa973fd49ecf3d816f677c8095143e944ad10485f30fb3dac85c13a346dace4dae2d0a15af91867320957bfd38a43d9eefbb74fe6919e15119b6d805'
|
|
|
|
|
)
|
|
|
|
|
->willReturn(101);
|
|
|
|
|
$this->logger->expects($this->once())
|
|
|
|
|
->method('info');
|
|
|
|
|
|
|
|
|
|
$this->limiter->registerUserRequest('MyIdentifier', 100, 100, $user);
|
|
|
|
|
}
|
|
|
|
|
@ -115,6 +126,8 @@ class LimiterTest extends TestCase {
|
|
|
|
|
'ddb2ec50fa973fd49ecf3d816f677c8095143e944ad10485f30fb3dac85c13a346dace4dae2d0a15af91867320957bfd38a43d9eefbb74fe6919e15119b6d805',
|
|
|
|
|
100
|
|
|
|
|
);
|
|
|
|
|
$this->logger->expects($this->never())
|
|
|
|
|
->method('info');
|
|
|
|
|
|
|
|
|
|
$this->limiter->registerUserRequest('MyIdentifier', 100, 100, $user);
|
|
|
|
|
}
|
|
|
|
|
|