allFixtureIDs('Folder'); foreach($folderIDs as $folderID) { $folder = DataObject::get_by_id('Folder', $folderID); $filePath = ASSETS_PATH . '/FileTest/' . $folder->getFilename(); SS_Filesystem::makeFolder($filePath); } // Create a test files for each of the fixture references $fileIDs = $this->allFixtureIDs('File'); foreach($fileIDs as $fileID) { $file = DataObject::get_by_id('File', $fileID); $root = ASSETS_PATH . '/FileTest/'; if($folder = $file->Parent()) { $root .= $folder->getFilename(); } $path = $root . substr($file->getHash(), 0, 10) . '/' . basename($file->getFilename()); SS_Filesystem::makeFolder(dirname($path)); $fh = fopen($path, "w+"); fwrite($fh, str_repeat('x', 1000000)); fclose($fh); } // Conditional fixture creation in case the 'cms' module is installed if(class_exists('ErrorPage')) { $page = new ErrorPage(array( 'Title' => 'Page not Found', 'ErrorCode' => 404 )); $page->write(); $page->publish('Stage', 'Live'); } } public function tearDown() { AssetStoreTest_SpyStore::reset(); parent::tearDown(); } public function testLinkShortcodeHandler() { $testFile = $this->objFromFixture('File', 'asdf'); $parser = new ShortcodeParser(); $parser->register('file_link', array('File', 'handle_shortcode')); $fileShortcode = sprintf('[file_link,id=%d]', $testFile->ID); $fileEnclosed = sprintf('[file_link,id=%d]Example Content[/file_link]', $testFile->ID); $fileShortcodeExpected = $testFile->Link(); $fileEnclosedExpected = sprintf( 'Example Content', $testFile->Link()); $this->assertEquals($fileShortcodeExpected, $parser->parse($fileShortcode), 'Test that simple linking works.'); $this->assertEquals($fileEnclosedExpected, $parser->parse($fileEnclosed), 'Test enclosed content is linked.'); $testFile->delete(); $fileShortcode = '[file_link,id="-1"]'; $fileEnclosed = '[file_link,id="-1"]Example Content[/file_link]'; $this->assertEquals('', $parser->parse('[file_link]'), 'Test that invalid ID attributes are not parsed.'); $this->assertEquals('', $parser->parse('[file_link,id="text"]')); $this->assertEquals('', $parser->parse('[file_link]Example Content[/file_link]')); if(class_exists('ErrorPage')) { $errorPage = ErrorPage::get()->filter('ErrorCode', 404)->First(); $this->assertEquals( $errorPage->Link(), $parser->parse($fileShortcode), 'Test link to 404 page if no suitable matches.' ); $this->assertEquals( sprintf('Example Content', $errorPage->Link()), $parser->parse($fileEnclosed) ); } else { $this->assertEquals('', $parser->parse($fileShortcode), 'Short code is removed if file record is not present.'); $this->assertEquals('', $parser->parse($fileEnclosed)); } } public function testCreateWithFilenameWithSubfolder() { // Note: We can't use fixtures/setUp() for this, as we want to create the db record manually. // Creating the folder is necessary to avoid having "Filename" overwritten by setName()/setRelativePath(), // because the parent folders don't exist in the database $folder = Folder::find_or_make('/FileTest/'); $testfilePath = BASE_PATH . '/assets/FileTest/CreateWithFilenameHasCorrectPath.txt'; // Important: No leading slash $fh = fopen($testfilePath, "w"); fwrite($fh, str_repeat('x',1000000)); fclose($fh); $file = new File(); $file->setFromLocalFile($testfilePath); $file->ParentID = $folder->ID; $file->write(); $this->assertEquals( 'CreateWithFilenameHasCorrectPath.txt', $file->Name, '"Name" property is automatically set from "Filename"' ); $this->assertEquals( 'FileTest/CreateWithFilenameHasCorrectPath.txt', $file->Filename, '"Filename" property remains unchanged' ); // TODO This should be auto-detected, see File->updateFilesystem() // $this->assertInstanceOf('Folder', $file->Parent(), 'Parent folder is created in database'); // $this->assertFileExists($file->Parent()->getURL(), 'Parent folder is created on filesystem'); // $this->assertEquals('FileTest', $file->Parent()->Name); // $this->assertInstanceOf('Folder', $file->Parent()->Parent(), 'Grandparent folder is created in database'); // $this->assertFileExists($file->Parent()->Parent()->getURL(), // 'Grandparent folder is created on filesystem'); // $this->assertEquals('assets', $file->Parent()->Parent()->Name); } public function testGetExtension() { $this->assertEquals('', File::get_file_extension('myfile'), 'No extension'); $this->assertEquals('txt', File::get_file_extension('myfile.txt'), 'Simple extension'); $this->assertEquals('gz', File::get_file_extension('myfile.tar.gz'), 'Double-barrelled extension only returns last bit'); } public function testValidateExtension() { Session::set('loggedInAs', null); $orig = Config::inst()->get('File', 'allowed_extensions'); Config::inst()->remove('File', 'allowed_extensions'); Config::inst()->update('File', 'allowed_extensions', array('txt')); $file = $this->objFromFixture('File', 'asdf'); // Invalid ext $file->Name = 'asdf.php'; $v = $file->validate(); $this->assertFalse($v->valid()); $this->assertContains('Extension is not allowed', $v->message()); // Valid ext $file->Name = 'asdf.txt'; $v = $file->validate(); $this->assertTrue($v->valid()); // Capital extension is valid as well $file->Name = 'asdf.TXT'; $v = $file->validate(); $this->assertTrue($v->valid()); Config::inst()->remove('File', 'allowed_extensions'); Config::inst()->update('File', 'allowed_extensions', $orig); } public function testAppCategory() { // Test various categories $this->assertEquals('image', File::get_app_category('jpg')); $this->assertEquals('image', File::get_app_category('JPG')); $this->assertEquals('image', File::get_app_category('JPEG')); $this->assertEquals('image', File::get_app_category('png')); $this->assertEquals('image', File::get_app_category('tif')); $this->assertEquals('document', File::get_app_category('pdf')); $this->assertEquals('video', File::get_app_category('mov')); $this->assertEquals('audio', File::get_app_category('OGG')); } public function testGetCategoryExtensions() { // Test specific categories $images = array( 'alpha', 'als', 'bmp', 'cel', 'gif', 'ico', 'icon', 'jpeg', 'jpg', 'pcx', 'png', 'ps', 'tif', 'tiff' ); $this->assertEquals($images, File::get_category_extensions('image')); $this->assertEquals(array('gif', 'jpeg', 'jpg', 'png'), File::get_category_extensions('image/supported')); $this->assertEquals($images, File::get_category_extensions(array('image', 'image/supported'))); $this->assertEquals( array('fla', 'gif', 'jpeg', 'jpg', 'png', 'swf'), File::get_category_extensions(array('flash', 'image/supported')) ); // Test other categories have at least one item $this->assertNotEmpty(File::get_category_extensions('archive')); $this->assertNotEmpty(File::get_category_extensions('audio')); $this->assertNotEmpty(File::get_category_extensions('document')); $this->assertNotEmpty(File::get_category_extensions('flash')); $this->assertNotEmpty(File::get_category_extensions('video')); } /** * @dataProvider allowedExtensions * @param string $extension */ public function testAllFilesHaveCategory($extension) { $this->assertNotEmpty( File::get_app_category($extension), "Assert that extension {$extension} has a valid category" ); } /** * Gets the list of all extensions for testing * * @return array */ public function allowedExtensions() { $args = array(); foreach(array_filter(File::config()->allowed_extensions) as $ext) { $args[] = array($ext); } return $args; } public function testSetNameChangesFilesystemOnWrite() { $file = $this->objFromFixture('File', 'asdf'); $oldPath = AssetStoreTest_SpyStore::getLocalPath($file); $newPath = str_replace('FileTest.txt', 'renamed.txt', $oldPath); // Before write() $file->Name = 'renamed.txt'; $this->assertFileExists($oldPath, 'Old path is still present'); $this->assertFileNotExists($newPath, 'New path is updated in memory, not written before write() is called'); $file->write(); // After write() $this->assertFileExists($oldPath, 'Old path is left after write()'); $this->assertFileExists($newPath, 'New path is created after write()'); } public function testSetParentIDChangesFilesystemOnWrite() { $file = $this->objFromFixture('File', 'asdf'); $subfolder = $this->objFromFixture('Folder', 'subfolder'); $oldPath = AssetStoreTest_SpyStore::getLocalPath($file); $newPath = str_replace('assets/FileTest/', 'assets/FileTest/FileTest-subfolder/', $oldPath); // set ParentID $file->ParentID = $subfolder->ID; // Before write() $this->assertFileExists($oldPath, 'Old path is still present'); $this->assertFileNotExists($newPath, 'New path is updated in memory, not written before write() is called'); $this->assertEquals($oldPath, AssetStoreTest_SpyStore::getLocalPath($file), 'URL is not updated until write is called'); $file->write(); // After write() $this->assertFileExists($oldPath, 'Old path is left after write()'); $this->assertFileExists($newPath, 'New path is created after write()'); $this->assertEquals($newPath, AssetStoreTest_SpyStore::getLocalPath($file), 'URL is updated after write is called'); } /** * @see http://open.silverstripe.org/ticket/5693 * * @expectedException ValidationException */ public function testSetNameWithInvalidExtensionDoesntChangeFilesystem() { $orig = Config::inst()->get('File', 'allowed_extensions'); Config::inst()->remove('File', 'allowed_extensions'); Config::inst()->update('File', 'allowed_extensions', array('txt')); $file = $this->objFromFixture('File', 'asdf'); $oldPath = $file->getURL(); $file->Name = 'renamed.php'; // evil extension try { $file->write(); } catch(ValidationException $e) { Config::inst()->remove('File', 'allowed_extensions'); Config::inst()->update('File', 'allowed_extensions', $orig); throw $e; } } public function testGetURL() { $rootfile = $this->objFromFixture('File', 'asdf'); $this->assertEquals('/assets/FileTest/55b443b601/FileTest.txt', $rootfile->getURL()); } public function testGetAbsoluteURL() { $rootfile = $this->objFromFixture('File', 'asdf'); $this->assertEquals( Director::absoluteBaseURL() . 'assets/FileTest/55b443b601/FileTest.txt', $rootfile->getAbsoluteURL() ); } public function testNameAndTitleGeneration() { // When name is assigned, title is automatically assigned $file = $this->objFromFixture('Image', 'setfromname'); $this->assertEquals('FileTest', $file->Title); } public function testSizeAndAbsoluteSizeParameters() { $file = $this->objFromFixture('File', 'asdf'); /* AbsoluteSize will give the integer number */ $this->assertEquals(1000000, $file->AbsoluteSize); /* Size will give a humanised number */ $this->assertEquals('977 KB', $file->Size); } public function testFileType() { $file = $this->objFromFixture('Image', 'gif'); $this->assertEquals("GIF image - good for diagrams", $file->FileType); $file = $this->objFromFixture('File', 'pdf'); $this->assertEquals("Adobe Acrobat PDF file", $file->FileType); $file = $this->objFromFixture('Image', 'gifupper'); $this->assertEquals("GIF image - good for diagrams", $file->FileType); /* Only a few file types are given special descriptions; the rest are unknown */ $file = $this->objFromFixture('File', 'asdf'); $this->assertEquals("unknown", $file->FileType); } /** * Test the File::format_size() method */ public function testFormatSize() { $this->assertEquals("1000 bytes", File::format_size(1000)); $this->assertEquals("1023 bytes", File::format_size(1023)); $this->assertEquals("1 KB", File::format_size(1025)); $this->assertEquals("9.8 KB", File::format_size(10000)); $this->assertEquals("49 KB", File::format_size(50000)); $this->assertEquals("977 KB", File::format_size(1000000)); $this->assertEquals("1 MB", File::format_size(1024*1024)); $this->assertEquals("954 MB", File::format_size(1000000000)); $this->assertEquals("1 GB", File::format_size(1024*1024*1024)); $this->assertEquals("9.3 GB", File::format_size(10000000000)); // It use any denomination higher than GB. It also doesn't overflow with >32 bit integers $this->assertEquals("93132.3 GB", File::format_size(100000000000000)); } public function testDeleteFile() { $file = $this->objFromFixture('File', 'asdf'); $fileID = $file->ID; $filePath = AssetStoreTest_SpyStore::getLocalPath($file); $file->delete(); // File is deleted $this->assertFileNotExists($filePath); $this->assertEmpty(DataObject::get_by_id('File', $fileID)); } public function testRenameFolder() { $newTitle = "FileTest-folder-renamed"; //rename a folder's title $folderID = $this->objFromFixture("Folder","folder2")->ID; $folder = DataObject::get_by_id('Folder',$folderID); $folder->Title = $newTitle; $folder->write(); //get folder again and see if the filename has changed $folder = DataObject::get_by_id('Folder',$folderID); $this->assertEquals( $newTitle . "/", $folder->Filename, "Folder Filename updated after rename of Title" ); //rename a folder's name $newTitle2 = "FileTest-folder-renamed2"; $folder->Name = $newTitle2; $folder->write(); //get folder again and see if the Title has changed $folder = DataObject::get_by_id('Folder',$folderID); $this->assertEquals($folder->Title, $newTitle2, "Folder Title updated after rename of Name"); //rename a folder's Filename $newTitle3 = "FileTest-folder-renamed3"; $folder->Filename = $newTitle3; $folder->write(); //get folder again and see if the Title has changed $folder = DataObject::get_by_id('Folder',$folderID); $this->assertEquals($folder->Title, $newTitle3, "Folder Title updated after rename of Filename"); } public function testSetsOwnerOnFirstWrite() { Session::set('loggedInAs', null); $member1 = new Member(); $member1->write(); $member2 = new Member(); $member2->write(); $file1 = new File(); $file1->write(); $this->assertEquals(0, $file1->OwnerID, 'Owner not written when no user is logged in'); $member1->logIn(); $file2 = new File(); $file2->write(); $this->assertEquals($member1->ID, $file2->OwnerID, 'Owner written when user is logged in'); $member2->logIn(); $file2->forceChange(); $file2->write(); $this->assertEquals($member1->ID, $file2->OwnerID, 'Owner not overwritten on existing files'); } public function testCanEdit() { $file = $this->objFromFixture('Image', 'gif'); // Test anonymous permissions Session::set('loggedInAs', null); $this->assertFalse($file->canEdit(), "Anonymous users can't edit files"); // Test permissionless user $this->objFromFixture('Member', 'frontend')->logIn(); $this->assertFalse($file->canEdit(), "Permissionless users can't edit files"); // Test global CMS section users $this->objFromFixture('Member', 'cms')->logIn(); $this->assertTrue($file->canEdit(), "Users with all CMS section access can edit files"); // Test cms access users without file access $this->objFromFixture('Member', 'security')->logIn(); $this->assertFalse($file->canEdit(), "Security CMS users can't edit files"); // Test asset-admin user $this->objFromFixture('Member', 'assetadmin')->logIn(); $this->assertTrue($file->canEdit(), "Asset admin users can edit files"); // Test admin $this->objFromFixture('Member', 'admin')->logIn(); $this->assertTrue($file->canEdit(), "Admins can edit files"); } public function testJoinPaths() { $this->assertEquals('name/file.jpg', File::join_paths('/name', 'file.jpg')); $this->assertEquals('name/file.jpg', File::join_paths('name', 'file.jpg')); $this->assertEquals('name/file.jpg', File::join_paths('/name', '/file.jpg')); $this->assertEquals('name/file.jpg', File::join_paths('name/', '/', 'file.jpg')); $this->assertEquals('file.jpg', File::join_paths('/', '/', 'file.jpg')); $this->assertEquals('', File::join_paths('/', '/')); } } class FileTest_MyCustomFile extends File implements TestOnly { }