silverstripe-cms/tests/php/Tasks/MigrateSiteTreeLinkingTaskTest.php

<?php

namespace SilverStripe\CMS\Tests\Tasks;

use SilverStripe\CMS\Model\SiteTree;
use SilverStripe\CMS\Tasks\MigrateSiteTreeLinkingTask;
use SilverStripe\Dev\SapphireTest;
use SilverStripe\ORM\DataObject;

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::class, 'home');
        $aboutID  = $this->idFromFixture(SiteTree::class, 'about');
        $staffID  = $this->idFromFixture(SiteTree::class, 'staff');
        $actionID = $this->idFromFixture(SiteTree::class, 'action');
        $hashID   = $this->idFromFixture(SiteTree::class, '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::class, $homeID)->Content,
            'HTML URLSegment links are rewritten.'
        );
        $this->assertEquals(
            $aboutContent,
            DataObject::get_by_id(SiteTree::class, $aboutID)->Content
        );
        $this->assertEquals(
            $staffContent,
            DataObject::get_by_id(SiteTree::class, $staffID)->Content
        );
        $this->assertEquals(
            $actionContent,
            DataObject::get_by_id(SiteTree::class, $actionID)->Content,
            'Links to actions on pages are rewritten correctly.'
        );
        $this->assertEquals(
            $hashLinkContent,
            DataObject::get_by_id(SiteTree::class, $hashID)->Content,
            'Hash/anchor links are correctly handled.'
        );
    }
}