implement tests for unlimited quota restrictions

Signed-off-by: Julien Veyssier <eneiluj@posteo.net>
nextcloud · Jul 19, 2021 · d4e6a65 · d4e6a65
1 parent d2cf158
commit d4e6a65
Showing 1 changed file with 105 additions and 0 deletions.
diff --git a/tests/lib/User/UserTest.php b/tests/lib/User/UserTest.php
@@ -724,6 +724,111 @@ public function testSetQuota() {
 		$user->setQuota('23 TB');
 	}
 
+	public function testGetDefaultUnlimitedQuota() {
+		/**
+		 * @var UserInterface | \PHPUnit\Framework\MockObject\MockObject $backend
+		 */
+		$backend = $this->createMock(\Test\Util\User\Dummy::class);
+
+		/** @var PublicEmitter|\PHPUnit\Framework\MockObject\MockObject $emitter */
+		$emitter = $this->createMock(PublicEmitter::class);
+		$emitter->expects($this->never())
+			->method('emit');
+
+		$config = $this->createMock(IConfig::class);
+		$user = new User('foo', $backend, $this->dispatcher, $emitter, $config);
+		$config->expects($this->at(0))
+			->method('getUserValue')
+			->with(
+				'foo',
+				'files',
+				'quota',
+				'default'
+			)
+			->willReturn('default');
+		$config->expects($this->at(1))
+			->method('getAppValue')
+			->with(
+				'files',
+				'default_quota',
+				'none'
+			)
+			->willReturn('none');
+		// allow unlimited quota
+		$config->expects($this->at(2))
+			->method('getAppValue')
+			->with(
+				'files',
+				'allow_unlimited_quota',
+				'1'
+			)
+			->willReturn('1');
+
+		$quota = $user->getQuota();
+		$this->assertEquals('none', $quota);
+	}
+
+	public function testGetDefaultUnlimitedQuotaForbidden() {
+		/**
+		 * @var UserInterface | \PHPUnit\Framework\MockObject\MockObject $backend
+		 */
+		$backend = $this->createMock(\Test\Util\User\Dummy::class);
+
+		/** @var PublicEmitter|\PHPUnit\Framework\MockObject\MockObject $emitter */
+		$emitter = $this->createMock(PublicEmitter::class);
+		$emitter->expects($this->never())
+			->method('emit');
+
+		$config = $this->createMock(IConfig::class);
+		$user = new User('foo', $backend, $this->dispatcher, $emitter, $config);
+		$config->expects($this->at(0))
+			->method('getUserValue')
+			->with(
+				'foo',
+				'files',
+				'quota',
+				'default'
+			)
+			->willReturn('default');
+		$config->expects($this->at(1))
+			->method('getAppValue')
+			->with(
+				'files',
+				'default_quota',
+				'none'
+			)
+			->willReturn('none');
+		// do not allow unlimited quota
+		$config->expects($this->at(2))
+			->method('getAppValue')
+			->with(
+				'files',
+				'allow_unlimited_quota',
+				'1'
+			)
+			->willReturn('0');
+		$config->expects($this->at(3))
+			->method('getAppValue')
+			->with(
+				'files',
+				'quota_preset',
+				'1 GB, 5 GB, 10 GB'
+			)
+			->willReturn('1 GB, 5 GB, 10 GB');
+		// expect seeing 1 GB used as fallback value
+		$config->expects($this->at(4))
+			->method('getAppValue')
+			->with(
+				'files',
+				'default_quota',
+				'1 GB'
+			)
+			->willReturn('1 GB');
+
+		$quota = $user->getQuota();
+		$this->assertEquals('1 GB', $quota);
+	}
+
 	public function testSetQuotaAddressNoChange() {
 		/**
 		 * @var UserInterface | \PHPUnit\Framework\MockObject\MockObject $backend