1 files changed, 158 insertions, 0 deletions
diff --git a/tests/lib/Files/Storage/Wrapper/AvailabilityTest.php b/tests/lib/Files/Storage/Wrapper/AvailabilityTest.php
new file mode 100644
index 00000000000..d890081cbb6
--- /dev/null
+++ b/tests/lib/Files/Storage/Wrapper/AvailabilityTest.php
@@ -0,0 +1,158 @@
+<?php
+
+/**
+ * SPDX-FileCopyrightText: 2019-2024 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-FileCopyrightText: 2016 ownCloud, Inc.
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+namespace Test\Files\Storage\Wrapper;
+
+use OC\Files\Cache\Storage as StorageCache;
+use OC\Files\Storage\Temporary;
+use OC\Files\Storage\Wrapper\Availability;
+use OCP\Files\StorageNotAvailableException;
+
+class AvailabilityTest extends \Test\TestCase {
+	/** @var \PHPUnit\Framework\MockObject\MockObject|StorageCache */
+	protected $storageCache;
+	/** @var \PHPUnit\Framework\MockObject\MockObject|Temporary */
+	protected $storage;
+	/** @var Availability */
+	protected $wrapper;
+
+	protected function setUp(): void {
+		parent::setUp();
+
+		$this->storageCache = $this->createMock(StorageCache::class);
+
+		$this->storage = $this->createMock(Temporary::class);
+		$this->storage->expects($this->any())
+			->method('getStorageCache')
+			->willReturn($this->storageCache);
+
+		$this->wrapper = new Availability(['storage' => $this->storage]);
+	}
+
+	/**
+	 * Storage is available
+	 */
+	public function testAvailable(): void {
+		$this->storage->expects($this->once())
+			->method('getAvailability')
+			->willReturn(['available' => true, 'last_checked' => 0]);
+		$this->storage->expects($this->never())
+			->method('test');
+		$this->storage->expects($this->once())
+			->method('mkdir');
+
+		$this->wrapper->mkdir('foobar');
+	}
+
+	/**
+	 * Storage marked unavailable, TTL not expired
+	 *
+	 */
+	public function testUnavailable(): void {
+		$this->expectException(StorageNotAvailableException::class);
+
+		$this->storage->expects($this->once())
+			->method('getAvailability')
+			->willReturn(['available' => false, 'last_checked' => time()]);
+		$this->storage->expects($this->never())
+			->method('test');
+		$this->storage->expects($this->never())
+			->method('mkdir');
+
+		$this->wrapper->mkdir('foobar');
+	}
+
+	/**
+	 * Storage marked unavailable, TTL expired
+	 */
+	public function testUnavailableRecheck(): void {
+		$this->storage->expects($this->once())
+			->method('getAvailability')
+			->willReturn(['available' => false, 'last_checked' => 0]);
+		$this->storage->expects($this->once())
+			->method('test')
+			->willReturn(true);
+		$calls = [
+			false, // prevents concurrent rechecks
+			true, // sets correct availability
+		];
+		$this->storage->expects($this->exactly(2))
+			->method('setAvailability')
+			->willReturnCallback(function ($value) use (&$calls): void {
+				$expected = array_shift($calls);
+				$this->assertEquals($expected, $value);
+			});
+		$this->storage->expects($this->once())
+			->method('mkdir');
+
+		$this->wrapper->mkdir('foobar');
+	}
+
+	/**
+	 * Storage marked available, but throws StorageNotAvailableException
+	 *
+	 */
+	public function testAvailableThrowStorageNotAvailable(): void {
+		$this->expectException(StorageNotAvailableException::class);
+
+		$this->storage->expects($this->once())
+			->method('getAvailability')
+			->willReturn(['available' => true, 'last_checked' => 0]);
+		$this->storage->expects($this->never())
+			->method('test');
+		$this->storage->expects($this->once())
+			->method('mkdir')
+			->willThrowException(new StorageNotAvailableException());
+		$this->storageCache->expects($this->once())
+			->method('setAvailability')
+			->with($this->equalTo(false));
+
+		$this->wrapper->mkdir('foobar');
+	}
+
+	/**
+	 * Storage available, but call fails
+	 * Method failure does not indicate storage unavailability
+	 */
+	public function testAvailableFailure(): void {
+		$this->storage->expects($this->once())
+			->method('getAvailability')
+			->willReturn(['available' => true, 'last_checked' => 0]);
+		$this->storage->expects($this->never())
+			->method('test');
+		$this->storage->expects($this->once())
+			->method('mkdir')
+			->willReturn(false);
+		$this->storage->expects($this->never())
+			->method('setAvailability');
+
+		$this->wrapper->mkdir('foobar');
+	}
+
+	/**
+	 * Storage available, but throws exception
+	 * Standard exception does not indicate storage unavailability
+	 *
+	 */
+	public function testAvailableThrow(): void {
+		$this->expectException(\Exception::class);
+
+		$this->storage->expects($this->once())
+			->method('getAvailability')
+			->willReturn(['available' => true, 'last_checked' => 0]);
+		$this->storage->expects($this->never())
+			->method('test');
+		$this->storage->expects($this->once())
+			->method('mkdir')
+			->willThrowException(new \Exception());
+		$this->storage->expects($this->never())
+			->method('setAvailability');
+
+		$this->wrapper->mkdir('foobar');
+	}
+}