Merge pull request #22135 from nextcloud/enh/noid/occ-preview-repair

Add occ preview:migrate to migrate previews from the old flat structure to a subfolder structure

1 files changed, 141 insertions, 0 deletions

diff --git a/tests/Core/Command/Preview/RepairTest.php b/tests/Core/Command/Preview/RepairTest.php
new file mode 100644
index 00000000000..68f32c1f498
--- /dev/null
+++ b/tests/Core/Command/Preview/RepairTest.php
@@ -0,0 +1,141 @@
+<?php
+
+namespace Tests\Core\Command\Preview;
+
+use bantu\IniGetWrapper\IniGetWrapper;
+use OC\Core\Command\Preview\Repair;
+use OCP\Files\Folder;
+use OCP\Files\IRootFolder;
+use OCP\Files\Node;
+use OCP\IConfig;
+use OCP\ILogger;
+use PHPUnit\Framework\MockObject\MockObject;
+use Symfony\Component\Console\Formatter\OutputFormatterInterface;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+use Test\TestCase;
+
+class RepairTest extends TestCase {
+	/** @var IConfig|MockObject */
+	private $config;
+	/** @var IRootFolder|MockObject */
+	private $rootFolder;
+	/** @var ILogger|MockObject */
+	private $logger;
+	/** @var IniGetWrapper|MockObject */
+	private $iniGetWrapper;
+	/** @var InputInterface|MockObject */
+	private $input;
+	/** @var OutputInterface|MockObject */
+	private $output;
+	/** @var string */
+	private $outputLines = '';
+	/** @var Repair */
+	private $repair;
+
+	protected function setUp(): void {
+		parent::setUp();
+		$this->config = $this->getMockBuilder(IConfig::class)
+			->getMock();
+		$this->rootFolder = $this->getMockBuilder(IRootFolder::class)
+			->getMock();
+		$this->logger = $this->getMockBuilder(ILogger::class)
+			->getMock();
+		$this->iniGetWrapper = $this->getMockBuilder(IniGetWrapper::class)
+			->getMock();
+		$this->repair = new Repair($this->config, $this->rootFolder, $this->logger, $this->iniGetWrapper);
+		$this->input = $this->getMockBuilder(InputInterface::class)
+			->getMock();
+		$this->input->expects($this->any())
+			->method('getOption')
+			->willReturnCallback(function ($parameter) {
+				if ($parameter === 'batch') {
+					return true;
+				}
+				return null;
+			});
+		$this->output = $this->getMockBuilder(OutputInterface::class)
+			->setMethods(['section', 'writeln', 'write', 'setVerbosity', 'getVerbosity', 'isQuiet', 'isVerbose', 'isVeryVerbose', 'isDebug', 'setDecorated', 'isDecorated', 'setFormatter', 'getFormatter'])
+			->getMock();
+		$self = $this;
+		$this->output->expects($this->any())
+			->method('section')
+			->willReturn($this->output);
+		$this->output->expects($this->any())
+			->method('getFormatter')
+			->willReturn($this->getMockBuilder(OutputFormatterInterface::class)->getMock());
+		$this->output->expects($this->any())
+			->method('writeln')
+			->willReturnCallback(function ($line) use ($self) {
+				$self->outputLines .= $line . "\n";
+			});
+	}
+
+	public function emptyTestDataProvider() {
+		/** directoryNames, expectedOutput */
+		return [
+			[
+				[],
+				'All previews are already migrated.'
+			],
+			[
+				[['name' => 'a'], ['name' => 'b'], ['name' => 'c']],
+				'All previews are already migrated.'
+			],
+			[
+				[['name' => '0', 'content' => ['folder', 'folder']], ['name' => 'b'], ['name' => 'c']],
+				'All previews are already migrated.'
+			],
+			[
+				[['name' => '0', 'content' => ['file', 'folder', 'folder']], ['name' => 'b'], ['name' => 'c']],
+				'A total of 1 preview files need to be migrated.'
+			],
+			[
+				[['name' => '23'], ['name' => 'b'], ['name' => 'c']],
+				'A total of 1 preview files need to be migrated.'
+			],
+		];
+	}
+
+	/**
+	 * @dataProvider emptyTestDataProvider
+	 */
+	public function testEmptyExecute($directoryNames, $expectedOutput) {
+		$previewFolder = $this->getMockBuilder(Folder::class)
+			->getMock();
+		$directories = array_map(function ($element) {
+			$dir = $this->getMockBuilder(Folder::class)
+				->getMock();
+			$dir->expects($this->any())
+				->method('getName')
+				->willReturn($element['name']);
+			if (isset($element['content'])) {
+				$list = [];
+				foreach ($element['content'] as $item) {
+					if ($item === 'file') {
+						$list[] = $this->getMockBuilder(Node::class)
+							->getMock();
+					} elseif ($item === 'folder') {
+						$list[] = $this->getMockBuilder(Folder::class)
+							->getMock();
+					}
+				}
+				$dir->expects($this->once())
+					->method('getDirectoryListing')
+					->willReturn($list);
+			}
+			return $dir;
+		}, $directoryNames);
+		$previewFolder->expects($this->once())
+			->method('getDirectoryListing')
+			->willReturn($directories);
+		$this->rootFolder->expects($this->at(0))
+			->method('get')
+			->with("appdata_/preview")
+			->willReturn($previewFolder);
+
+		$this->repair->run($this->input, $this->output);
+
+		$this->assertStringContainsString($expectedOutput, $this->outputLines);
+	}
+}