add some tests for mimetype check
Signed-off-by: Robin Appelman <robin@icewind.nl>
This commit is contained in:
Robin Appelman
parent
dcbf88868f
commit
de77189186
|
|
@ -0,0 +1,192 @@
|
||||||
|
<?php
|
||||||
|
|
||||||
|
declare(strict_types=1);
|
||||||
|
/**
|
||||||
|
* @copyright Copyright (c) 2021 Robin Appelman <robin@icewind.nl>
|
||||||
|
*
|
||||||
|
* @license GNU AGPL version 3 or any later version
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU Affero General Public License as
|
||||||
|
* published by the Free Software Foundation, either version 3 of the
|
||||||
|
* License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* 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
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
namespace OCA\WorkflowEngine\Tests\Check;
|
||||||
|
|
||||||
|
use OC\Files\Storage\Temporary;
|
||||||
|
use OCA\WorkflowEngine\Check\FileMimeType;
|
||||||
|
use OCP\Files\IMimeTypeDetector;
|
||||||
|
use OCP\IL10N;
|
||||||
|
use OCP\IRequest;
|
||||||
|
use Test\TestCase;
|
||||||
|
|
||||||
|
class TemporaryNoLocal extends Temporary {
|
||||||
|
public function instanceOfStorage($className) {
|
||||||
|
if ($className === '\OC\Files\Storage\Local') {
|
||||||
|
return false;
|
||||||
|
} else {
|
||||||
|
return parent::instanceOfStorage($className);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @group DB
|
||||||
|
*/
|
||||||
|
class FileMimeTypeTest extends TestCase {
|
||||||
|
/** @var IL10N */
|
||||||
|
private $l10n;
|
||||||
|
/** @var IRequest */
|
||||||
|
private $request;
|
||||||
|
/** @var IMimeTypeDetector */
|
||||||
|
private $mimeDetector;
|
||||||
|
|
||||||
|
private $extensions = [
|
||||||
|
'.txt' => 'text/plain-path-detected',
|
||||||
|
];
|
||||||
|
|
||||||
|
private $content = [
|
||||||
|
'text-content' => 'text/plain-content-detected',
|
||||||
|
];
|
||||||
|
|
||||||
|
protected function setUp(): void {
|
||||||
|
parent::setUp();
|
||||||
|
|
||||||
|
$this->l10n = $this->createMock(IL10N::class);
|
||||||
|
$this->request = $this->createMock(IRequest::class);
|
||||||
|
$this->mimeDetector = $this->createMock(IMimeTypeDetector::class);
|
||||||
|
$this->mimeDetector->method('detectPath')
|
||||||
|
->willReturnCallback(function ($path) {
|
||||||
|
foreach ($this->extensions as $extension => $mime) {
|
||||||
|
if (strpos($path, $extension) !== false) {
|
||||||
|
return $mime;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 'application/octet-stream';
|
||||||
|
});
|
||||||
|
$this->mimeDetector->method('detectContent')
|
||||||
|
->willReturnCallback(function ($path) {
|
||||||
|
$body = file_get_contents($path);
|
||||||
|
foreach ($this->content as $match => $mime) {
|
||||||
|
if (strpos($body, $match) !== false) {
|
||||||
|
return $mime;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 'application/octet-stream';
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testUseCachedMimetype() {
|
||||||
|
$storage = new Temporary([]);
|
||||||
|
$storage->mkdir('foo');
|
||||||
|
$storage->file_put_contents('foo/bar.txt', 'asd');
|
||||||
|
$storage->getScanner()->scan('');
|
||||||
|
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain'));
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testNonCachedNotExists() {
|
||||||
|
$storage = new Temporary([]);
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain-path-detected'));
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testNonCachedLocal() {
|
||||||
|
$storage = new Temporary([]);
|
||||||
|
$storage->mkdir('foo');
|
||||||
|
$storage->file_put_contents('foo/bar.txt', 'text-content');
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain-content-detected'));
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testNonCachedNotLocal() {
|
||||||
|
$storage = new TemporaryNoLocal([]);
|
||||||
|
$storage->mkdir('foo');
|
||||||
|
$storage->file_put_contents('foo/bar.txt', 'text-content');
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain-path-detected'));
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testFallback() {
|
||||||
|
$storage = new Temporary([]);
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'unknown');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'application/octet-stream'));
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testFromCacheCached() {
|
||||||
|
$storage = new Temporary([]);
|
||||||
|
$storage->mkdir('foo');
|
||||||
|
$storage->file_put_contents('foo/bar.txt', 'asd');
|
||||||
|
$storage->getScanner()->scan('');
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain'));
|
||||||
|
|
||||||
|
$storage->getCache()->clear();
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain'));
|
||||||
|
|
||||||
|
$newCheck = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$newCheck->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
$this->assertTrue($newCheck->executeCheck('is', 'text/plain-path-detected'));
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testExistsCached() {
|
||||||
|
$storage = new TemporaryNoLocal([]);
|
||||||
|
$storage->mkdir('foo');
|
||||||
|
$storage->file_put_contents('foo/bar.txt', 'text-content');
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain-content-detected'));
|
||||||
|
$storage->unlink('foo/bar.txt');
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain-content-detected'));
|
||||||
|
|
||||||
|
$newCheck = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$newCheck->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
$this->assertTrue($newCheck->executeCheck('is', 'text/plain-path-detected'));
|
||||||
|
}
|
||||||
|
|
||||||
|
public function testNonExistsNotCached() {
|
||||||
|
$storage = new TemporaryNoLocal([]);
|
||||||
|
|
||||||
|
$check = new FileMimeType($this->l10n, $this->request, $this->mimeDetector);
|
||||||
|
$check->setFileInfo($storage, 'foo/bar.txt');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain-path-detected'));
|
||||||
|
|
||||||
|
$storage->mkdir('foo');
|
||||||
|
$storage->file_put_contents('foo/bar.txt', 'text-content');
|
||||||
|
|
||||||
|
$this->assertTrue($check->executeCheck('is', 'text/plain-content-detected'));
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue