mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-10-22 14:05:37 +02:00
125 lines
3.2 KiB
PHP
125 lines
3.2 KiB
PHP
<?php
|
|
use SilverStripe\Assets\Storage\AssetStore;
|
|
use SilverStripe\ORM\DataObject;
|
|
use SilverStripe\Core\Config\Config;
|
|
use SilverStripe\Dev\SapphireTest;
|
|
use SilverStripe\Dev\TestOnly;
|
|
|
|
|
|
|
|
|
|
/**
|
|
* Description of DBFileTest
|
|
*
|
|
* @author dmooyman
|
|
*/
|
|
class DBFileTest extends SapphireTest {
|
|
|
|
protected $extraDataObjects = array(
|
|
'DBFileTest_Object',
|
|
'DBFileTest_Subclass'
|
|
);
|
|
|
|
protected $usesDatabase = true;
|
|
|
|
public function setUp() {
|
|
parent::setUp();
|
|
|
|
// Set backend
|
|
AssetStoreTest_SpyStore::activate('DBFileTest');
|
|
Config::inst()->update('SilverStripe\\Control\\Director', 'alternate_base_url', '/mysite/');
|
|
}
|
|
|
|
public function tearDown() {
|
|
AssetStoreTest_SpyStore::reset('DBFileTest');
|
|
parent::tearDown();
|
|
}
|
|
|
|
/**
|
|
* Test that images in a DBFile are rendered properly
|
|
*/
|
|
public function testRender() {
|
|
$obj = new DBFileTest_Object();
|
|
|
|
// Test image tag
|
|
$fish = realpath(__DIR__ .'/../model/testimages/test-image-high-quality.jpg');
|
|
$this->assertFileExists($fish);
|
|
$obj->MyFile->setFromLocalFile($fish, 'awesome-fish.jpg');
|
|
$this->assertEquals(
|
|
'<img src="/mysite/assets/DBFileTest/a870de278b/awesome-fish.jpg" alt="awesome-fish.jpg" />',
|
|
trim($obj->MyFile->forTemplate())
|
|
);
|
|
|
|
// Test download tag
|
|
$obj->MyFile->setFromString('puppies', 'subdir/puppy-document.txt');
|
|
$this->assertEquals(
|
|
'<a href="/mysite/assets/DBFileTest/subdir/2a17a9cb4b/puppy-document.txt" title="puppy-document.txt" download="puppy-document.txt"/>',
|
|
trim($obj->MyFile->forTemplate())
|
|
);
|
|
}
|
|
|
|
public function testValidation() {
|
|
$obj = new DBFileTest_ImageOnly();
|
|
|
|
// Test from image
|
|
$fish = realpath(__DIR__ .'/../model/testimages/test-image-high-quality.jpg');
|
|
$this->assertFileExists($fish);
|
|
$obj->MyFile->setFromLocalFile($fish, 'awesome-fish.jpg');
|
|
|
|
// This should fail
|
|
$this->setExpectedException('SilverStripe\\ORM\\ValidationException');
|
|
$obj->MyFile->setFromString('puppies', 'subdir/puppy-document.txt');
|
|
}
|
|
|
|
public function testPermission() {
|
|
$obj = new DBFileTest_Object();
|
|
|
|
// Test from image
|
|
$fish = realpath(__DIR__ .'/../model/testimages/test-image-high-quality.jpg');
|
|
$this->assertFileExists($fish);
|
|
$obj->MyFile->setFromLocalFile($fish, 'private/awesome-fish.jpg', null, null, array(
|
|
'visibility' => AssetStore::VISIBILITY_PROTECTED
|
|
));
|
|
|
|
// Test various file permissions work on DBFile
|
|
$this->assertFalse($obj->MyFile->canViewFile());
|
|
$obj->MyFile->getURL();
|
|
$this->assertTrue($obj->MyFile->canViewFile());
|
|
$obj->MyFile->revokeFile();
|
|
$this->assertFalse($obj->MyFile->canViewFile());
|
|
$obj->MyFile->getURL(false);
|
|
$this->assertFalse($obj->MyFile->canViewFile());
|
|
$obj->MyFile->grantFile();
|
|
$this->assertTrue($obj->MyFile->canViewFile());
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* @property DBFile $MyFile
|
|
*/
|
|
class DBFileTest_Object extends DataObject implements TestOnly {
|
|
private static $db = array(
|
|
"MyFile" => "DBFile"
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @property DBFile $AnotherFile
|
|
*/
|
|
class DBFileTest_Subclass extends DBFileTest_Object implements TestOnly {
|
|
private static $db = array(
|
|
"AnotherFile" => "DBFile"
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @property DBFile $MyFile
|
|
*/
|
|
class DBFileTest_ImageOnly extends DataObject implements TestOnly {
|
|
private static $db = array(
|
|
"MyFile" => "DBFile('image/supported')"
|
|
);
|
|
}
|
|
|