123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171 |
- <?php
- /**
- * SPDX-FileCopyrightText: 2016-2024 Nextcloud GmbH and Nextcloud contributors
- * SPDX-FileCopyrightText: 2013-2016 ownCloud, Inc.
- * SPDX-License-Identifier: AGPL-3.0-only
- */
- namespace OCA\DAV\Tests\unit\Connector\Sabre;
- use OC\Files\View;
- use OCA\DAV\Connector\Sabre\QuotaPlugin;
- use OCP\Files\FileInfo;
- use Test\TestCase;
- class QuotaPluginTest extends TestCase {
- /** @var \Sabre\DAV\Server | \PHPUnit\Framework\MockObject\MockObject */
- private $server;
- /** @var QuotaPlugin|\PHPUnit\Framework\MockObject\MockObject */
- private $plugin;
- private function init($quota, $checkedPath = ''): void {
- $view = $this->buildFileViewMock($quota, $checkedPath);
- $this->server = new \Sabre\DAV\Server();
- $this->plugin = $this->getMockBuilder(QuotaPlugin::class)
- ->setConstructorArgs([$view])
- ->setMethods(['getFileChunking'])
- ->getMock();
- $this->plugin->initialize($this->server);
- }
- /**
- * @dataProvider lengthProvider
- */
- public function testLength($expected, $headers): void {
- $this->init(0);
- $this->plugin->expects($this->never())
- ->method('getFileChunking');
- $this->server->httpRequest = new \Sabre\HTTP\Request('POST', 'dummy.file', $headers);
- $length = $this->plugin->getLength();
- $this->assertEquals($expected, $length);
- }
- /**
- * @dataProvider quotaOkayProvider
- */
- public function testCheckQuota($quota, $headers): void {
- $this->init($quota);
- $this->plugin->expects($this->never())
- ->method('getFileChunking');
- $this->server->httpRequest = new \Sabre\HTTP\Request('POST', 'dummy.file', $headers);
- $result = $this->plugin->checkQuota('');
- $this->assertTrue($result);
- }
- /**
- * @dataProvider quotaExceededProvider
- */
- public function testCheckExceededQuota($quota, $headers): void {
- $this->expectException(\Sabre\DAV\Exception\InsufficientStorage::class);
- $this->init($quota);
- $this->plugin->expects($this->never())
- ->method('getFileChunking');
- $this->server->httpRequest = new \Sabre\HTTP\Request('POST', 'dummy.file', $headers);
- $this->plugin->checkQuota('');
- }
- /**
- * @dataProvider quotaOkayProvider
- */
- public function testCheckQuotaOnPath($quota, $headers): void {
- $this->init($quota, 'sub/test.txt');
- $this->plugin->expects($this->never())
- ->method('getFileChunking');
- $this->server->httpRequest = new \Sabre\HTTP\Request('POST', 'dummy.file', $headers);
- $result = $this->plugin->checkQuota('/sub/test.txt');
- $this->assertTrue($result);
- }
- public function quotaOkayProvider() {
- return [
- [1024, []],
- [1024, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [1024, ['CONTENT-LENGTH' => '512']],
- [1024, ['OC-TOTAL-LENGTH' => '1024', 'CONTENT-LENGTH' => '512']],
- [FileInfo::SPACE_UNKNOWN, []],
- [FileInfo::SPACE_UNKNOWN, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [FileInfo::SPACE_UNKNOWN, ['CONTENT-LENGTH' => '512']],
- [FileInfo::SPACE_UNKNOWN, ['OC-TOTAL-LENGTH' => '1024', 'CONTENT-LENGTH' => '512']],
- [FileInfo::SPACE_UNLIMITED, []],
- [FileInfo::SPACE_UNLIMITED, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [FileInfo::SPACE_UNLIMITED, ['CONTENT-LENGTH' => '512']],
- [FileInfo::SPACE_UNLIMITED, ['OC-TOTAL-LENGTH' => '1024', 'CONTENT-LENGTH' => '512']],
- ];
- }
- public function quotaExceededProvider() {
- return [
- [1023, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [511, ['CONTENT-LENGTH' => '512']],
- [2047, ['OC-TOTAL-LENGTH' => '2048', 'CONTENT-LENGTH' => '1024']],
- ];
- }
- public function lengthProvider() {
- return [
- [null, []],
- [1024, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [512, ['CONTENT-LENGTH' => '512']],
- [2048, ['OC-TOTAL-LENGTH' => '2048', 'CONTENT-LENGTH' => '1024']],
- [4096, ['OC-TOTAL-LENGTH' => '2048', 'X-EXPECTED-ENTITY-LENGTH' => '4096']],
- [null, ['X-EXPECTED-ENTITY-LENGTH' => 'A']],
- [null, ['CONTENT-LENGTH' => 'A']],
- [1024, ['OC-TOTAL-LENGTH' => 'A', 'CONTENT-LENGTH' => '1024']],
- [1024, ['OC-TOTAL-LENGTH' => 'A', 'X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [2048, ['OC-TOTAL-LENGTH' => '2048', 'X-EXPECTED-ENTITY-LENGTH' => 'A']],
- [2048, ['OC-TOTAL-LENGTH' => '2048', 'CONTENT-LENGTH' => 'A']],
- ];
- }
- public function quotaChunkedOkProvider() {
- return [
- [1024, 0, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [1024, 0, ['CONTENT-LENGTH' => '512']],
- [1024, 0, ['OC-TOTAL-LENGTH' => '1024', 'CONTENT-LENGTH' => '512']],
- // with existing chunks (allowed size = total length - chunk total size)
- [400, 128, ['X-EXPECTED-ENTITY-LENGTH' => '512']],
- [400, 128, ['CONTENT-LENGTH' => '512']],
- [400, 128, ['OC-TOTAL-LENGTH' => '512', 'CONTENT-LENGTH' => '500']],
- // \OCP\Files\FileInfo::SPACE-UNKNOWN = -2
- [-2, 0, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [-2, 0, ['CONTENT-LENGTH' => '512']],
- [-2, 0, ['OC-TOTAL-LENGTH' => '1024', 'CONTENT-LENGTH' => '512']],
- [-2, 128, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [-2, 128, ['CONTENT-LENGTH' => '512']],
- [-2, 128, ['OC-TOTAL-LENGTH' => '1024', 'CONTENT-LENGTH' => '512']],
- ];
- }
- public function quotaChunkedFailProvider() {
- return [
- [400, 0, ['X-EXPECTED-ENTITY-LENGTH' => '1024']],
- [400, 0, ['CONTENT-LENGTH' => '512']],
- [400, 0, ['OC-TOTAL-LENGTH' => '1024', 'CONTENT-LENGTH' => '512']],
- // with existing chunks (allowed size = total length - chunk total size)
- [380, 128, ['X-EXPECTED-ENTITY-LENGTH' => '512']],
- [380, 128, ['CONTENT-LENGTH' => '512']],
- [380, 128, ['OC-TOTAL-LENGTH' => '512', 'CONTENT-LENGTH' => '500']],
- ];
- }
- private function buildFileViewMock($quota, $checkedPath) {
- // mock filesysten
- $view = $this->getMockBuilder(View::class)
- ->setMethods(['free_space'])
- ->disableOriginalConstructor()
- ->getMock();
- $view->expects($this->any())
- ->method('free_space')
- ->with($this->identicalTo($checkedPath))
- ->willReturn($quota);
- return $view;
- }
- }
|