1 files changed, 40 insertions, 30 deletions
diff --git a/tests/lib/Files/ObjectStore/MapperTest.php b/tests/lib/Files/ObjectStore/MapperTest.php
index 307096d8dbc..6448d5ce1f5 100644
--- a/tests/lib/Files/ObjectStore/MapperTest.php
+++ b/tests/lib/Files/ObjectStore/MapperTest.php
@@ -1,54 +1,64 @@
 <?php
+
 /**
- * @author Roeland Jago Douma <rullzer@owncloud.com>
- *
- * @copyright Copyright (c) 2016, ownCloud, Inc.
- * @license AGPL-3.0
- *
- * This code is free software: you can redistribute it and/or modify
- * it under the terms of the GNU Affero General Public License, version 3,
- * as published by the Free Software Foundation.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU Affero General Public License for more details.
- *
- * You should have received a copy of the GNU Affero General Public License, version 3,
- * along with this program.  If not, see <http://www.gnu.org/licenses/>
- *
+ * SPDX-FileCopyrightText: 2016-2024 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-FileCopyrightText: 2016 ownCloud, Inc.
+ * SPDX-License-Identifier: AGPL-3.0-only
  */
 
 namespace Test\Files\ObjectStore;
 
 use OC\Files\ObjectStore\Mapper;
+use OCP\IConfig;
 use OCP\IUser;
 
 class MapperTest extends \Test\TestCase {
-	public function dataGetBucket() {
+	/** @var IUser|\PHPUnit\Framework\MockObject\MockObject */
+	private $user;
+
+	/** @var IConfig|\PHPUnit\Framework\MockObject\MockObject */
+	private $config;
+
+	/** @var Mapper */
+	private $mapper;
+
+	protected function setUp(): void {
+		parent::setUp();
+
+		$this->user = $this->createMock(IUser::class);
+		$this->config = $this->createMock(IConfig::class);
+		$this->mapper = new Mapper($this->user, $this->config);
+	}
+
+	public static function dataGetBucket(): array {
 		return [
-			['user', 64, '17'],
-			['USER', 64, '0'],
-			['bc0e8b52-a66c-1035-90c6-d9663bda9e3f', 64, '56'],
-			['user', 8, '1'],
-			['user', 2, '1'],
-			['USER', 2, '0'],
+			['user', 64, 0, '17'],
+			['USER', 64, 0, '0'],
+			['bc0e8b52-a66c-1035-90c6-d9663bda9e3f', 64, 0, '56'],
+			['user', 8, 0, '1'],
+			['user', 2, 0, '1'],
+			['USER', 2, 0, '0'],
+			['user', 128, 64, '81'],
 		];
 	}
 
 	/**
-	 * @dataProvider dataGetBucket
 	 * @param string $username
 	 * @param int $numBuckets
 	 * @param string $expectedBucket
 	 */
-	public function testGetBucket($username, $numBuckets, $expectedBucket) {
-		$user = $this->createMock(IUser::class);
-		$user->method('getUID')
+	#[\PHPUnit\Framework\Attributes\DataProvider('dataGetBucket')]
+	public function testGetBucket($username, $numBuckets, $bucketShift, $expectedBucket): void {
+		$this->user->expects($this->once())
+			->method('getUID')
 			->willReturn($username);
 
-		$mapper = new Mapper($user);
+		$this->config->expects($this->once())
+			->method('getSystemValue')
+			->with('objectstore_multibucket')
+			->willReturn(['arguments' => ['min_bucket' => $bucketShift]]);
 
-		$this->assertSame($expectedBucket, $mapper->getBucket($numBuckets));
+		$result = $this->mapper->getBucket($numBuckets);
+		$this->assertEquals($expectedBucket, $result);
 	}
 }