101 lines
3.0 KiB
PHP
101 lines
3.0 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2025 Nextcloud GmbH
|
|
* SPDX-FileContributor: Carl Schwan
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace Test\Preview;
|
|
|
|
use OC\Preview\Db\Preview;
|
|
use OC\Preview\Db\PreviewMapper;
|
|
use OCP\IDBConnection;
|
|
use OCP\Server;
|
|
use OCP\Snowflake\IGenerator;
|
|
use Test\TestCase;
|
|
|
|
#[\PHPUnit\Framework\Attributes\Group('DB')]
|
|
class PreviewMapperTest extends TestCase {
|
|
private PreviewMapper $previewMapper;
|
|
private IDBConnection $connection;
|
|
private IGenerator $snowflake;
|
|
|
|
public function setUp(): void {
|
|
parent::setUp();
|
|
$this->previewMapper = Server::get(PreviewMapper::class);
|
|
$this->connection = Server::get(IDBConnection::class);
|
|
$this->snowflake = Server::get(IGenerator::class);
|
|
|
|
$qb = $this->connection->getQueryBuilder();
|
|
$qb->delete('preview_locations')->executeStatement();
|
|
|
|
$qb = $this->connection->getQueryBuilder();
|
|
$qb->delete('preview_versions')->executeStatement();
|
|
|
|
$qb = $this->connection->getQueryBuilder();
|
|
$qb->delete('previews')->executeStatement();
|
|
}
|
|
|
|
public function tearDown(): void {
|
|
$this->previewMapper->deleteAll();
|
|
parent::tearDown();
|
|
}
|
|
|
|
public function testGetAvailablePreviews(): void {
|
|
// Empty
|
|
$this->assertEquals([], $this->previewMapper->getAvailablePreviews([]));
|
|
|
|
// No preview available
|
|
$this->assertEquals([42 => []], $this->previewMapper->getAvailablePreviews([42]));
|
|
|
|
$this->createPreviewForFileId(42);
|
|
$previews = $this->previewMapper->getAvailablePreviews([42]);
|
|
$this->assertNotEmpty($previews[42]);
|
|
$this->assertNull($previews[42][0]->getLocationId());
|
|
$this->assertNull($previews[42][0]->getBucketName());
|
|
$this->assertNull($previews[42][0]->getObjectStoreName());
|
|
|
|
$this->createPreviewForFileId(43, 2);
|
|
$previews = $this->previewMapper->getAvailablePreviews([43]);
|
|
$this->assertNotEmpty($previews[43]);
|
|
$this->assertEquals('preview-2', $previews[43][0]->getBucketName());
|
|
$this->assertEquals('default', $previews[43][0]->getObjectStoreName());
|
|
}
|
|
|
|
private function createPreviewForFileId(int $fileId, ?int $bucket = null): void {
|
|
$locationId = null;
|
|
if ($bucket) {
|
|
$qb = $this->connection->getQueryBuilder();
|
|
$locationId = $this->snowflake->nextId();
|
|
$qb->insert('preview_locations')
|
|
->values([
|
|
'id' => $locationId,
|
|
'bucket_name' => $qb->createNamedParameter('preview-' . $bucket),
|
|
'object_store_name' => $qb->createNamedParameter('default'),
|
|
]);
|
|
$qb->executeStatement();
|
|
}
|
|
$preview = new Preview();
|
|
$preview->setId($this->snowflake->nextId());
|
|
$preview->setFileId($fileId);
|
|
$preview->setStorageId(1);
|
|
$preview->setCropped(true);
|
|
$preview->setMax(true);
|
|
$preview->setWidth(100);
|
|
$preview->setSourceMimeType('image/jpeg');
|
|
$preview->setHeight(100);
|
|
$preview->setSize(100);
|
|
$preview->setMtime(time());
|
|
$preview->setMimetype('image/jpeg');
|
|
$preview->setEtag('abcdefg');
|
|
|
|
if ($locationId !== null) {
|
|
$preview->setLocationId($locationId);
|
|
}
|
|
$this->previewMapper->insert($preview);
|
|
}
|
|
}
|