nextcloud/tests/lib/Preview/BackgroundCleanupJobTest.php

160 lines
3.8 KiB
PHP
Raw Normal View History

<?php
/**
* @copyright Copyright (c) 2018, Roeland Jago Douma <roeland@famdouma.nl>
*
* @author Roeland Jago Douma <roeland@famdouma.nl>
*
* @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/>
*
*/
namespace Test\Preview;
use OC\Files\AppData\Factory;
use OC\Preview\BackgroundCleanupJob;
use OC\PreviewManager;
use OCP\Files\IRootFolder;
use OCP\IDBConnection;
use Test\Traits\MountProviderTrait;
use Test\Traits\UserTrait;
/**
* Class BackgroundCleanupJobTest
*
* @group DB
*
* @package Test\Preview
*/
class BackgroundCleanupJobTest extends \Test\TestCase {
use MountProviderTrait;
use UserTrait;
/** @var string */
private $userId;
/** @var bool */
private $trashEnabled;
/** @var Factory */
private $appDataFactory;
/** @var IDBConnection */
private $connection;
/** @var PreviewManager */
private $previewManager;
/** @var IRootFolder */
private $rootFolder;
public function setUp() {
parent::setUp();
$this->userId = $this->getUniqueID();
$this->createUser($this->userId, $this->userId);
$storage = new \OC\Files\Storage\Temporary([]);
$this->registerMount($this->userId, $storage, '');
$this->loginAsUser($this->userId);
$this->logout();
$this->loginAsUser($this->userId);
$appManager = \OC::$server->getAppManager();
$this->trashEnabled = $appManager->isEnabledForUser('files_trashbin', $this->userId);
$appManager->disableApp('files_trashbin');
$this->appDataFactory = \OC::$server->query(Factory::class);
$this->connection = \OC::$server->getDatabaseConnection();
$this->previewManager = \OC::$server->getPreviewManager();
$this->rootFolder = \OC::$server->getRootFolder();
}
public function tearDown() {
if ($this->trashEnabled) {
$appManager = \OC::$server->getAppManager();
$appManager->enableApp('files_trashbin');
}
$this->logout();
return parent::tearDown();
}
private function setup11Previews(): array {
$userFolder = $this->rootFolder->getUserFolder($this->userId);
$files = [];
for ($i = 0; $i < 11; $i++) {
$file = $userFolder->newFile($i.'.txt');
$file->putContent('hello world!');
$this->previewManager->getPreview($file);
$files[] = $file;
}
return $files;
}
public function testCleanupSystemCron() {
$files = $this->setup11Previews();
$preview = $this->appDataFactory->get('preview');
$previews = $preview->getDirectoryListing();
$this->assertCount(11, $previews);
$job = new BackgroundCleanupJob($this->connection, $this->appDataFactory, true);
$job->run([]);
foreach ($files as $file) {
$file->delete();
}
$this->assertCount(11, $previews);
$job->run([]);
$previews = $preview->getDirectoryListing();
$this->assertCount(0, $previews);
}
public function testCleanupAjax() {
$files = $this->setup11Previews();
$preview = $this->appDataFactory->get('preview');
$previews = $preview->getDirectoryListing();
$this->assertCount(11, $previews);
$job = new BackgroundCleanupJob($this->connection, $this->appDataFactory, false);
$job->run([]);
foreach ($files as $file) {
$file->delete();
}
$this->assertCount(11, $previews);
$job->run([]);
$previews = $preview->getDirectoryListing();
$this->assertCount(1, $previews);
$job->run([]);
$previews = $preview->getDirectoryListing();
$this->assertCount(0, $previews);
}
}