silverstripe-framework/tests/php/Forms/HTMLEditor/HTMLEditorFieldToolbarTest.php

<?php

namespace SilverStripe\Forms\Tests\HTMLEditor;

use SilverStripe;
use SilverStripe\Assets\File;
use SilverStripe\Assets\Image;
use SilverStripe\Assets\Tests\Storage\AssetStoreTest\TestAssetStore;
use SilverStripe\Control\HTTPResponse_Exception;
use SilverStripe\Dev\SapphireTest;
use SilverStripe\Forms\HTMLEditor\HTMLEditorField_Toolbar;
use SilverStripe\Forms\Tests\HTMLEditor\HTMLEditorFieldToolbarTest\Toolbar;

class HTMLEditorFieldToolbarTest extends SapphireTest
{

	protected static $fixture_file = 'HTMLEditorFieldToolbarTest.yml';

	/**
	 * @return Toolbar
	 */
	protected function getToolbar()
	{
		return new Toolbar(null, '/');
	}

	public function setUp()
	{
		parent::setUp();

		HTMLEditorField_Toolbar::config()->update('fileurl_scheme_whitelist', array('http'));
		HTMLEditorField_Toolbar::config()->update('fileurl_domain_whitelist', array('example.com'));

		// Filesystem mock
		TestAssetStore::activate(__CLASS__);

		// Load up files
		/** @var File $file1 */
		$file1 = $this->objFromFixture(File::class, 'example_file');
		$file1->setFromString(str_repeat('x', 1000), $file1->Name);
		$file1->write();

		/** @var Image $image1 */
		$image1 = $this->objFromFixture(Image::class, 'example_image');
		$image1->setFromLocalFile(
			__DIR__ . '/HTMLEditorFieldTest/images/example.jpg',
			'folder/subfolder/example.jpg'
		);
		$image1->write();
	}

	public function testValidLocalReference()
	{
		/** @var File $exampleFile */
		$exampleFile = $this->objFromFixture(File::class, 'example_file');
		$expectedUrl = $exampleFile->AbsoluteLink();
		HTMLEditorField_Toolbar::config()->update('fileurl_domain_whitelist', array(
			'example.com',
			strtolower(parse_url($expectedUrl, PHP_URL_HOST))
		));

		list($file, $url) = $this->getToolbar()->viewfile_getRemoteFileByURL($exampleFile->AbsoluteLink());
		$this->assertEquals($expectedUrl, $url);
	}

	public function testValidScheme()
	{
		list($file, $url) = $this->getToolbar()->viewfile_getRemoteFileByURL('http://example.com/test.pdf');
		$this->assertEquals($url, 'http://example.com/test.pdf');
	}

	/** @expectedException SilverStripe\Control\HTTPResponse_Exception */
	public function testInvalidScheme()
	{
		list($file, $url) = $this->getToolbar()->viewfile_getRemoteFileByURL('nosuchscheme://example.com/test.pdf');
	}

	public function testValidDomain()
	{
		list($file, $url) = $this->getToolbar()->viewfile_getRemoteFileByURL('http://example.com/test.pdf');
		$this->assertEquals($url, 'http://example.com/test.pdf');
	}

	/** @expectedException SilverStripe\Control\HTTPResponse_Exception */
	public function testInvalidDomain()
	{
		list($file, $url) = $this->getToolbar()->viewfile_getRemoteFileByURL('http://evil.com/test.pdf');
	}

}