1 files changed, 35 insertions, 11 deletions
diff --git a/tests/lib/Files/ObjectStore/MapperTest.php b/tests/lib/Files/ObjectStore/MapperTest.php
index 307096d8dbc..d8427f993fe 100644
--- a/tests/lib/Files/ObjectStore/MapperTest.php
+++ b/tests/lib/Files/ObjectStore/MapperTest.php
@@ -22,17 +22,37 @@
 namespace Test\Files\ObjectStore;
 
 use OC\Files\ObjectStore\Mapper;
+use OCP\IConfig;
 use OCP\IUser;
 
 class MapperTest extends \Test\TestCase {
+
+	/** @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 function dataGetBucket() {
 		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'],
 		];
 	}
 
@@ -42,13 +62,17 @@ class MapperTest extends \Test\TestCase {
 	 * @param int $numBuckets
 	 * @param string $expectedBucket
 	 */
-	public function testGetBucket($username, $numBuckets, $expectedBucket) {
-		$user = $this->createMock(IUser::class);
-		$user->method('getUID')
+	public function testGetBucket($username, $numBuckets, $bucketShift, $expectedBucket) {
+		$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);
 	}
 }