update('DMS', 'shortcode_handler_key', 'dms_document_link'); $d1 = $this->objFromFixture('DMSDocument', 'd1'); $d2 = $this->objFromFixture('DMSDocument', 'd2'); $page1 = new SiteTree(); $page1->Content = 'Condition: '; $page1ID = $page1->write(); $page2 = new SiteTree(); $page2->Content = 'Condition: '; $page2ID = $page2->write(); $page3 = new SiteTree(); $page3->Content = 'Condition: '; $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); } }