silverstripe-cms/tests/tasks/MigrateSiteTreeLinkingTaskTest.php

<?php
/**
 * @package cms
 * @subpackage tests
 */
class MigrateSiteTreeLinkingTaskTest extends SapphireTest {

	protected static $fixture_file = 'MigrateSiteTreeLinkingTaskTest.yml';

	protected static $use_draft_site = true;

	public function testLinkingMigration() {
		ob_start();

		$task = new MigrateSiteTreeLinkingTask();
		$task->run(null);

		$this->assertEquals (
			"Rewrote 9 link(s) on 5 page(s) to use shortcodes.\n",
			ob_get_contents(),
			'Rewritten links are correctly reported'
		);
		ob_end_clean();

		$homeID   = $this->idFromFixture('SiteTree', 'home');
		$aboutID  = $this->idFromFixture('SiteTree', 'about');
		$staffID  = $this->idFromFixture('SiteTree', 'staff');
		$actionID = $this->idFromFixture('SiteTree', 'action');
		$hashID   = $this->idFromFixture('SiteTree', 'hash_link');

		$homeContent = sprintf (
			'<a href="[sitetree_link,id=%d]">About</a><a href="[sitetree_link,id=%d]">Staff</a><a href="http://silverstripe.org/">External Link</a><a name="anchor"></a>',
			$aboutID,
			$staffID
		);
		$aboutContent = sprintf (
			'<a href="[sitetree_link,id=%d]">Home</a><a href="[sitetree_link,id=%d]">Staff</a><a name="second-anchor"></a>',
			$homeID,
			$staffID
		);
		$staffContent = sprintf (
			'<a href="[sitetree_link,id=%d]">Home</a><a href="[sitetree_link,id=%d]">About</a>',
			$homeID,
			$aboutID
		);
		$actionContent = sprintf (
			'<a href="[sitetree_link,id=%d]SearchForm">Search Form</a>', $homeID
		);
		$hashLinkContent = sprintf (
			'<a href="[sitetree_link,id=%d]#anchor">Home</a><a href="[sitetree_link,id=%d]#second-anchor">About</a>',
			$homeID,
			$aboutID
		);

		$this->assertEquals (
			$homeContent,
			DataObject::get_by_id('SiteTree', $homeID)->Content,
			'HTML URLSegment links are rewritten.'
		);
		$this->assertEquals (
			$aboutContent,
			DataObject::get_by_id('SiteTree', $aboutID)->Content
		);
		$this->assertEquals (
			$staffContent,
			DataObject::get_by_id('SiteTree', $staffID)->Content
		);
		$this->assertEquals (
			$actionContent,
			DataObject::get_by_id('SiteTree', $actionID)->Content,
			'Links to actions on pages are rewritten correctly.'
		);
		$this->assertEquals (
			$hashLinkContent,
			DataObject::get_by_id('SiteTree', $hashID)->Content,
			'Hash/anchor links are correctly handled.'
		);
	}

}