2008-04-06 10:20:13 +02:00
|
|
|
<?php
|
2015-09-15 04:52:02 +02:00
|
|
|
|
2016-10-14 03:30:05 +02:00
|
|
|
namespace SilverStripe\Assets\Tests;
|
|
|
|
|
2016-08-19 00:51:35 +02:00
|
|
|
use SilverStripe\Assets\File;
|
2016-10-14 03:30:05 +02:00
|
|
|
use SilverStripe\Assets\Upload;
|
|
|
|
use SilverStripe\Assets\Tests\Storage\AssetStoreTest\TestAssetStore;
|
2016-08-19 00:51:35 +02:00
|
|
|
use SilverStripe\Core\Config\Config;
|
|
|
|
use SilverStripe\Dev\SapphireTest;
|
2016-10-14 03:30:05 +02:00
|
|
|
use SilverStripe\ORM\Versioning\Versioned;
|
2016-06-15 06:03:16 +02:00
|
|
|
|
2016-12-16 05:34:21 +01:00
|
|
|
class UploadTest extends SapphireTest
|
|
|
|
{
|
|
|
|
|
|
|
|
protected $usesDatabase = true;
|
|
|
|
|
|
|
|
public function setUp()
|
|
|
|
{
|
|
|
|
parent::setUp();
|
|
|
|
Versioned::set_stage(Versioned::DRAFT);
|
|
|
|
TestAssetStore::activate('UploadTest');
|
|
|
|
}
|
|
|
|
|
|
|
|
public function tearDown()
|
|
|
|
{
|
|
|
|
TestAssetStore::reset();
|
|
|
|
parent::tearDown();
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUpload()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.txt';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
2016-12-29 09:21:00 +01:00
|
|
|
for ($i = 0; $i < 10000; $i++) {
|
2016-12-16 05:34:21 +01:00
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'txt',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u1 = new Upload();
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$u1->loadIntoFile($tmpFile);
|
|
|
|
$file1 = $u1->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'Uploads/UploadTest-testUpload.txt',
|
|
|
|
$file1->getFilename()
|
|
|
|
);
|
|
|
|
$this->assertEquals(
|
|
|
|
BASE_PATH . '/assets/UploadTest/.protected/Uploads/315ae4c3d4/UploadTest-testUpload.txt',
|
|
|
|
TestAssetStore::getLocalPath($file1)
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file1),
|
|
|
|
'File upload to standard directory in /assets'
|
|
|
|
);
|
|
|
|
|
|
|
|
// test upload into custom folder
|
|
|
|
$customFolder = 'UploadTest-testUpload';
|
|
|
|
$u2 = new Upload();
|
|
|
|
$u2->loadIntoFile($tmpFile, null, $customFolder);
|
|
|
|
$file2 = $u2->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload/UploadTest-testUpload.txt',
|
|
|
|
$file2->getFilename()
|
|
|
|
);
|
|
|
|
$this->assertEquals(
|
|
|
|
BASE_PATH . '/assets/UploadTest/.protected/UploadTest-testUpload/315ae4c3d4/UploadTest-testUpload.txt',
|
|
|
|
TestAssetStore::getLocalPath($file2)
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file2),
|
|
|
|
'File upload to custom directory in /assets'
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testAllowedFilesize()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.txt';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
2016-12-29 09:21:00 +01:00
|
|
|
for ($i = 0; $i < 10000; $i++) {
|
2016-12-16 05:34:21 +01:00
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'txt',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u1 = new Upload();
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
|
|
|
|
$v->setAllowedMaxFileSize(array('txt' => 10));
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$result = $u1->loadIntoFile($tmpFile);
|
|
|
|
$this->assertFalse($result, 'Load failed because size was too big');
|
|
|
|
|
|
|
|
$v->setAllowedMaxFileSize(array('[document]' => 10));
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$result = $u1->loadIntoFile($tmpFile);
|
|
|
|
$this->assertFalse($result, 'Load failed because size was too big');
|
|
|
|
|
|
|
|
$v->setAllowedMaxFileSize(array('txt' => 200000));
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$result = $u1->loadIntoFile($tmpFile);
|
|
|
|
$this->assertTrue($result, 'Load failed with setting max file size');
|
|
|
|
|
|
|
|
// check max file size set by app category
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.jpg';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent . $tmpFileContent);
|
|
|
|
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'image/jpeg',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'jpg',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v->setAllowedMaxFileSize(array('[image]' => '40k'));
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$result = $u1->loadIntoFile($tmpFile);
|
|
|
|
$this->assertTrue($result, 'Load failed with setting max file size');
|
|
|
|
|
|
|
|
$v->setAllowedMaxFileSize(array('[image]' => '1k'));
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$result = $u1->loadIntoFile($tmpFile);
|
|
|
|
$this->assertFalse($result, 'Load failed because size was too big');
|
|
|
|
|
|
|
|
$v->setAllowedMaxFileSize(array('[image]' => 1000));
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$result = $u1->loadIntoFile($tmpFile);
|
|
|
|
$this->assertFalse($result, 'Load failed because size was too big');
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testPHPUploadErrors()
|
|
|
|
{
|
|
|
|
$configMaxFileSizes = ['*' => '1k'];
|
|
|
|
Config::inst()->update(
|
|
|
|
'SilverStripe\\Assets\\Upload_Validator',
|
|
|
|
'default_max_file_size',
|
|
|
|
$configMaxFileSizes
|
|
|
|
);
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'myfile.jpg';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<100;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// Build file
|
|
|
|
$upload = new Upload();
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => '',
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test ok
|
|
|
|
$this->assertTrue($upload->validate($tmpFile));
|
|
|
|
|
|
|
|
// Test zero size file
|
|
|
|
$upload->clearErrors();
|
|
|
|
$tmpFile['size'] = 0;
|
|
|
|
$this->assertFalse($upload->validate($tmpFile));
|
|
|
|
$this->assertContains(
|
|
|
|
_t('File.NOFILESIZE', 'Filesize is zero bytes.'),
|
|
|
|
$upload->getErrors()
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test file too large
|
|
|
|
$upload->clearErrors();
|
|
|
|
$tmpFile['error'] = UPLOAD_ERR_INI_SIZE;
|
|
|
|
$this->assertFalse($upload->validate($tmpFile));
|
|
|
|
$this->assertContains(
|
|
|
|
_t(
|
|
|
|
'File.TOOLARGE',
|
|
|
|
'Filesize is too large, maximum {size} allowed',
|
|
|
|
'Argument 1: Filesize (e.g. 1MB)',
|
|
|
|
array('size' => '1 KB')
|
|
|
|
),
|
|
|
|
$upload->getErrors()
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test form size
|
|
|
|
$upload->clearErrors();
|
|
|
|
$tmpFile['error'] = UPLOAD_ERR_FORM_SIZE;
|
|
|
|
$this->assertFalse($upload->validate($tmpFile));
|
|
|
|
$this->assertContains(
|
|
|
|
_t(
|
|
|
|
'File.TOOLARGE',
|
|
|
|
'Filesize is too large, maximum {size} allowed',
|
|
|
|
'Argument 1: Filesize (e.g. 1MB)',
|
|
|
|
array('size' => '1 KB')
|
|
|
|
),
|
|
|
|
$upload->getErrors()
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test no file
|
|
|
|
$upload->clearErrors();
|
|
|
|
$tmpFile['error'] = UPLOAD_ERR_NO_FILE;
|
|
|
|
$this->assertFalse($upload->validate($tmpFile));
|
|
|
|
$this->assertContains(
|
|
|
|
_t('File.NOVALIDUPLOAD', 'File is not a valid upload'),
|
|
|
|
$upload->getErrors()
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testGetAllowedMaxFileSize()
|
|
|
|
{
|
|
|
|
Config::nest();
|
|
|
|
|
|
|
|
// Check the max file size uses the config values
|
|
|
|
$configMaxFileSizes = array(
|
|
|
|
'[image]' => '1k',
|
|
|
|
'txt' => 1000
|
|
|
|
);
|
|
|
|
Config::inst()->update('SilverStripe\\Assets\\Upload_Validator', 'default_max_file_size', $configMaxFileSizes);
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('[image]');
|
|
|
|
$this->assertEquals(1024, $retrievedSize, 'Max file size check on default values failed (config category set check)');
|
|
|
|
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('txt');
|
|
|
|
$this->assertEquals(1000, $retrievedSize, 'Max file size check on default values failed (config extension set check)');
|
|
|
|
|
|
|
|
// Check instance values for max file size
|
|
|
|
$maxFileSizes = array(
|
|
|
|
'[document]' => 2000,
|
|
|
|
'txt' => '4k'
|
|
|
|
);
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedMaxFileSize($maxFileSizes);
|
|
|
|
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('[document]');
|
|
|
|
$this->assertEquals(2000, $retrievedSize, 'Max file size check on instance values failed (instance category set check)');
|
|
|
|
|
|
|
|
// Check that the instance values overwrote the default values
|
|
|
|
// ie. The max file size will not exist for [image]
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('[image]');
|
|
|
|
$this->assertFalse($retrievedSize, 'Max file size check on instance values failed (config overridden check)');
|
|
|
|
|
|
|
|
// Check a category that has not been set before
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('[archive]');
|
|
|
|
$this->assertFalse($retrievedSize, 'Max file size check on instance values failed (category not set check)');
|
|
|
|
|
|
|
|
// Check a file extension that has not been set before
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('mp3');
|
|
|
|
$this->assertFalse($retrievedSize, 'Max file size check on instance values failed (extension not set check)');
|
|
|
|
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('txt');
|
|
|
|
$this->assertEquals(4096, $retrievedSize, 'Max file size check on instance values failed (instance extension set check)');
|
|
|
|
|
|
|
|
// Check a wildcard max file size against a file with an extension
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedMaxFileSize(2000);
|
|
|
|
|
|
|
|
$retrievedSize = $v->getAllowedMaxFileSize('.jpg');
|
|
|
|
$this->assertEquals(2000, $retrievedSize, 'Max file size check on instance values failed (wildcard max file size)');
|
|
|
|
|
|
|
|
Config::unnest();
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testAllowedSizeOnFileWithNoExtension()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<10000;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => '',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedMaxFileSize(array('' => 10));
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u1 = new Upload();
|
|
|
|
$u1->setValidator($v);
|
|
|
|
$result = $u1->loadIntoFile($tmpFile);
|
|
|
|
|
|
|
|
$this->assertFalse($result, 'Load failed because size was too big');
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUploadDoesNotAllowUnknownExtension()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.php';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<10000;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'php',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedExtensions(array('txt'));
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$result = $u->loadIntoFile($tmpFile);
|
|
|
|
|
|
|
|
$this->assertFalse($result, 'Load failed because extension was not accepted');
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUploadAcceptsAllowedExtension()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.txt';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<10000;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'txt',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedExtensions(array('txt'));
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file = $u->getFile();
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file),
|
|
|
|
'File upload to custom directory in /assets'
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUploadDeniesNoExtensionFilesIfNoEmptyStringSetForValidatorExtensions()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<10000;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => '',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedExtensions(array('txt'));
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$result = $u->loadIntoFile($tmpFile);
|
|
|
|
|
|
|
|
$this->assertFalse($result, 'Load failed because extension was not accepted');
|
|
|
|
$this->assertEquals(1, count($u->getErrors()), 'There is a single error of the file extension');
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUploadTarGzFileTwiceAppendsNumber()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.tar.gz';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<10000;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'tar.gz',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file = $u->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload.tar.gz',
|
|
|
|
$file->Name,
|
|
|
|
'File has a name without a number because it\'s not a duplicate'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
|
|
|
|
$u = new Upload();
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file2 = $u->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload-v2.tar.gz',
|
|
|
|
$file2->Name,
|
|
|
|
'File receives a number attached to the end before the extension'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file2),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
$this->assertGreaterThan(
|
|
|
|
$file->ID,
|
|
|
|
$file2->ID,
|
|
|
|
'File database record is not the same'
|
|
|
|
);
|
|
|
|
|
|
|
|
$u = new Upload();
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file3 = $u->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload-v3.tar.gz',
|
|
|
|
$file3->Name,
|
|
|
|
'File receives a number attached to the end before the extension'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file3),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
$this->assertGreaterThan(
|
|
|
|
$file2->ID,
|
|
|
|
$file3->ID,
|
|
|
|
'File database record is not the same'
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testUploadFileWithNoExtensionTwiceAppendsNumber()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<10000;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'txt',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedExtensions(array(''));
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file = $u->getFile();
|
|
|
|
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload',
|
|
|
|
$file->Name,
|
|
|
|
'File is uploaded without extension'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file2 = $u->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload-v2',
|
|
|
|
$file2->Name,
|
|
|
|
'File receives a number attached to the end'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file2),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
$this->assertGreaterThan(
|
|
|
|
$file->ID,
|
|
|
|
$file2->ID,
|
|
|
|
'File database record is not the same'
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testReplaceFile()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i=0; $i<10000;
|
|
|
|
$i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'txt',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
$v->setAllowedExtensions(array(''));
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file = $u->getFile();
|
|
|
|
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload',
|
|
|
|
$file->Name,
|
|
|
|
'File is uploaded without extension'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->setReplaceFile(true);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file2 = $u->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload',
|
|
|
|
$file2->Name,
|
|
|
|
'File does not receive new name'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file2),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
$this->assertEquals(
|
|
|
|
$file->ID,
|
|
|
|
$file2->ID,
|
|
|
|
'File database record is the same'
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testReplaceFileWithLoadIntoFile()
|
|
|
|
{
|
|
|
|
// create tmp file
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.txt';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i = 0; $i < 10000; $i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'txt',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
$file = $u->getFile();
|
|
|
|
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload.txt',
|
|
|
|
$file->Name,
|
|
|
|
'File is uploaded without extension'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
|
|
|
|
// replace=true
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->setReplaceFile(true);
|
|
|
|
$u->loadIntoFile($tmpFile, new File());
|
|
|
|
$file2 = $u->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload.txt',
|
|
|
|
$file2->Name,
|
|
|
|
'File does not receive new name'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file2),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
$this->assertEquals(
|
|
|
|
$file->ID,
|
|
|
|
$file2->ID,
|
|
|
|
'File database record is the same'
|
|
|
|
);
|
|
|
|
|
|
|
|
// replace=false
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->setReplaceFile(false);
|
|
|
|
$u->loadIntoFile($tmpFile, new File());
|
|
|
|
$file3 = $u->getFile();
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-testUpload-v2.txt',
|
|
|
|
$file3->Name,
|
|
|
|
'File does receive new name'
|
|
|
|
);
|
|
|
|
$this->assertFileExists(
|
|
|
|
TestAssetStore::getLocalPath($file3),
|
|
|
|
'File exists'
|
|
|
|
);
|
|
|
|
$this->assertGreaterThan(
|
|
|
|
$file2->ID,
|
|
|
|
$file3->ID,
|
|
|
|
'File database record is not the same'
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testDeleteResampledImagesOnUpload()
|
|
|
|
{
|
|
|
|
$tmpFileName = 'UploadTest-testUpload.jpg';
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
|
|
|
|
$uploadImage = function () use ($tmpFileName, $tmpFilePath) {
|
|
|
|
copy(__DIR__ . '/GDTest/images/test_jpg.jpg', $tmpFilePath);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'jpg',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setReplaceFile(true);
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
return $u->getFile();
|
|
|
|
};
|
|
|
|
|
|
|
|
// Image upload and generate a resampled image
|
|
|
|
$image = $uploadImage();
|
|
|
|
$resampled = $image->ResizedImage(123, 456);
|
|
|
|
$resampledPath = TestAssetStore::getLocalPath($resampled);
|
|
|
|
$this->assertFileExists($resampledPath);
|
|
|
|
|
|
|
|
// Re-upload the image, overwriting the original
|
|
|
|
// Resampled images should removed when their parent file is overwritten
|
|
|
|
$image = $uploadImage();
|
|
|
|
$this->assertFileExists($resampledPath);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testFileVersioningWithAnExistingFile()
|
|
|
|
{
|
|
|
|
$upload = function ($tmpFileName) {
|
|
|
|
// create tmp file
|
|
|
|
$tmpFilePath = TEMP_FOLDER . '/' . $tmpFileName;
|
|
|
|
$tmpFileContent = '';
|
|
|
|
for ($i = 0; $i < 10000; $i++) {
|
|
|
|
$tmpFileContent .= '0';
|
|
|
|
}
|
|
|
|
file_put_contents($tmpFilePath, $tmpFileContent);
|
|
|
|
|
|
|
|
// emulates the $_FILES array
|
|
|
|
$tmpFile = array(
|
|
|
|
'name' => $tmpFileName,
|
|
|
|
'type' => 'text/plaintext',
|
|
|
|
'size' => filesize($tmpFilePath),
|
|
|
|
'tmp_name' => $tmpFilePath,
|
|
|
|
'extension' => 'jpg',
|
|
|
|
'error' => UPLOAD_ERR_OK,
|
|
|
|
);
|
|
|
|
|
|
|
|
$v = new UploadTest\Validator();
|
|
|
|
|
|
|
|
// test upload into default folder
|
|
|
|
$u = new Upload();
|
|
|
|
$u->setReplaceFile(false);
|
|
|
|
$u->setValidator($v);
|
|
|
|
$u->loadIntoFile($tmpFile);
|
|
|
|
return $u->getFile();
|
|
|
|
};
|
|
|
|
|
|
|
|
// test empty file version prefix
|
|
|
|
Config::inst()->update('SilverStripe\\Assets\\Storage\\DefaultAssetNameGenerator', 'version_prefix', '');
|
|
|
|
|
|
|
|
$file1 = $upload('UploadTest-IMG001.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-IMG001.jpg',
|
|
|
|
$file1->Name,
|
|
|
|
'File does not receive new name'
|
|
|
|
);
|
|
|
|
|
|
|
|
$file2 = $upload('UploadTest-IMG001.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-IMG002.jpg',
|
|
|
|
$file2->Name,
|
|
|
|
'File does receive new name'
|
|
|
|
);
|
|
|
|
|
|
|
|
$file3 = $upload('UploadTest-IMG002.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-IMG003.jpg',
|
|
|
|
$file3->Name,
|
|
|
|
'File does receive new name'
|
|
|
|
);
|
|
|
|
|
|
|
|
$file4 = $upload('UploadTest-IMG3.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest-IMG3.jpg',
|
|
|
|
$file4->Name,
|
|
|
|
'File does not receive new name'
|
|
|
|
);
|
|
|
|
|
|
|
|
$file1->delete();
|
|
|
|
$file2->delete();
|
|
|
|
$file3->delete();
|
|
|
|
$file4->delete();
|
|
|
|
|
|
|
|
// test '-v' file version prefix
|
|
|
|
Config::inst()->update('SilverStripe\\Assets\\Storage\\DefaultAssetNameGenerator', 'version_prefix', '-v');
|
|
|
|
|
|
|
|
$file1 = $upload('UploadTest2-IMG001.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest2-IMG001.jpg',
|
|
|
|
$file1->Name,
|
|
|
|
'File does not receive new name'
|
|
|
|
);
|
|
|
|
|
|
|
|
$file2 = $upload('UploadTest2-IMG001.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest2-IMG001-v2.jpg',
|
|
|
|
$file2->Name,
|
|
|
|
'File does receive new name'
|
|
|
|
);
|
|
|
|
|
|
|
|
$file3 = $upload('UploadTest2-IMG001.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest2-IMG001-v3.jpg',
|
|
|
|
$file3->Name,
|
|
|
|
'File does receive new name'
|
|
|
|
);
|
|
|
|
|
|
|
|
$file4 = $upload('UploadTest2-IMG001-v3.jpg');
|
|
|
|
$this->assertEquals(
|
|
|
|
'UploadTest2-IMG001-v4.jpg',
|
|
|
|
$file4->Name,
|
|
|
|
'File does receive new name'
|
|
|
|
);
|
|
|
|
}
|
2008-04-06 10:20:13 +02:00
|
|
|
}
|