silverstripe-dms/tests/ShortCodeRelationFinderTest.php

<?php
class ShortCodeRelationFinderTest extends SapphireTest
{

    public static $fixture_file = array(
        'dms/tests/dmstest.yml'
    );

    public function testFindInRate()
    {
        $d1 = $this->objFromFixture('DMSDocument', 'd1');
        $d2 = $this->objFromFixture('DMSDocument', 'd2');

        $page1 = new SiteTree();
        $page1->Content = 'Condition:  <a title="document test 1" href="[dms_document_link,id='.$d1->ID.']">';
        $page1ID = $page1->write();

        $page2 = new SiteTree();
        $page2->Content = 'Condition:  <a title="document test 2" href="[dms_document_link,id='.$d2->ID.']">';
        $page2ID = $page2->write();

        $page3 = new SiteTree();
        $page3->Content = 'Condition:  <a title="document test 1" href="[dms_document_link,id='.$d1->ID.']">';
        $page3ID = $page3->write();

        $finder = new ShortCodeRelationFinder();

        $ids = $finder->findPageIDs('UnknownShortcode');
        $this->assertEquals(0, count($ids));

        $ids = $finder->findPageIDs($d1->ID);
        $this->assertNotContains($page2ID, $ids);
        $this->assertContains($page1ID, $ids);
        $this->assertContains($page3ID, $ids);
    }
}