mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-08-27 13:06:04 +02:00
Check only when the parent has changed - hierarchy traversal is expensive operation, so we do it only when it is needed.
195 lines
7.1 KiB
PHP
195 lines
7.1 KiB
PHP
<?php
|
|
|
|
class HierarchyTest extends SapphireTest {
|
|
|
|
static $fixture_file = 'HierarchyTest.yml';
|
|
|
|
protected $requiredExtensions = array(
|
|
'HierarchyTest_Object' => array('Hierarchy', 'Versioned')
|
|
);
|
|
|
|
protected $extraDataObjects = array(
|
|
'HierarchyTest_Object'
|
|
);
|
|
|
|
/**
|
|
* Test the Hierarchy prevents infinite loops.
|
|
*/
|
|
function testPreventLoop() {
|
|
$obj2 = $this->objFromFixture('HierarchyTest_Object', 'obj2');
|
|
$obj2aa = $this->objFromFixture('HierarchyTest_Object', 'obj2aa');
|
|
|
|
$obj2->ParentID = $obj2aa->ID;
|
|
try {
|
|
$obj2->write();
|
|
}
|
|
catch (ValidationException $e) {
|
|
$this->assertContains('Infinite loop found within the "HierarchyTest_Object" hierarchy', $e->getMessage());
|
|
return;
|
|
}
|
|
|
|
$this->fail('Failed to prevent infinite loop in hierarchy.');
|
|
}
|
|
|
|
/**
|
|
* Test Hierarchy::AllHistoricalChildren().
|
|
*/
|
|
function testAllHistoricalChildren() {
|
|
// Delete some objs
|
|
$this->objFromFixture('HierarchyTest_Object', 'obj2b')->delete();
|
|
$this->objFromFixture('HierarchyTest_Object', 'obj3a')->delete();
|
|
$this->objFromFixture('HierarchyTest_Object', 'obj3')->delete();
|
|
|
|
// Check that obj1-3 appear at the top level of the AllHistoricalChildren tree
|
|
$this->assertEquals(array("Obj 1", "Obj 2", "Obj 3"),
|
|
singleton('HierarchyTest_Object')->AllHistoricalChildren()->column('Title'));
|
|
|
|
// Check numHistoricalChildren
|
|
$this->assertEquals(3, singleton('HierarchyTest_Object')->numHistoricalChildren());
|
|
|
|
// Check that both obj 2 children are returned
|
|
$obj2 = $this->objFromFixture('HierarchyTest_Object', 'obj2');
|
|
$this->assertEquals(array("Obj 2a", "Obj 2b"),
|
|
$obj2->AllHistoricalChildren()->column('Title'));
|
|
|
|
// Check numHistoricalChildren
|
|
$this->assertEquals(2, $obj2->numHistoricalChildren());
|
|
|
|
|
|
// Obj 3 has been deleted; let's bring it back from the grave
|
|
$obj3 = Versioned::get_including_deleted("HierarchyTest_Object", "\"Title\" = 'Obj 3'")->First();
|
|
|
|
// Check that both obj 3 children are returned
|
|
$this->assertEquals(array("Obj 3a", "Obj 3b"),
|
|
$obj3->AllHistoricalChildren()->column('Title'));
|
|
|
|
// Check numHistoricalChildren
|
|
$this->assertEquals(2, $obj3->numHistoricalChildren());
|
|
|
|
}
|
|
|
|
/**
|
|
* Test that you can call Hierarchy::markExpanded/Unexpanded/Open() on a obj, and that
|
|
* calling Hierarchy::isMarked() on a different instance of that object will return true.
|
|
*/
|
|
function testItemMarkingIsntRestrictedToSpecificInstance() {
|
|
// Mark a few objs
|
|
$this->objFromFixture('HierarchyTest_Object', 'obj2')->markExpanded();
|
|
$this->objFromFixture('HierarchyTest_Object', 'obj2a')->markExpanded();
|
|
$this->objFromFixture('HierarchyTest_Object', 'obj2b')->markExpanded();
|
|
$this->objFromFixture('HierarchyTest_Object', 'obj3')->markUnexpanded();
|
|
|
|
// Query some objs in a different context and check their m
|
|
$objs = DataObject::get("HierarchyTest_Object", '', '"ID" ASC');
|
|
$marked = $expanded = array();
|
|
foreach($objs as $obj) {
|
|
if($obj->isMarked()) $marked[] = $obj->Title;
|
|
if($obj->isExpanded()) $expanded[] = $obj->Title;
|
|
}
|
|
|
|
$this->assertEquals(array('Obj 2', 'Obj 3', 'Obj 2a', 'Obj 2b'), $marked);
|
|
$this->assertEquals(array('Obj 2', 'Obj 2a', 'Obj 2b'), $expanded);
|
|
}
|
|
|
|
function testNumChildren() {
|
|
$this->assertEquals($this->objFromFixture('HierarchyTest_Object', 'obj1')->numChildren(), 0);
|
|
$this->assertEquals($this->objFromFixture('HierarchyTest_Object', 'obj2')->numChildren(), 2);
|
|
$this->assertEquals($this->objFromFixture('HierarchyTest_Object', 'obj3')->numChildren(), 2);
|
|
$this->assertEquals($this->objFromFixture('HierarchyTest_Object', 'obj2a')->numChildren(), 2);
|
|
$this->assertEquals($this->objFromFixture('HierarchyTest_Object', 'obj2b')->numChildren(), 0);
|
|
$this->assertEquals($this->objFromFixture('HierarchyTest_Object', 'obj3a')->numChildren(), 2);
|
|
$this->assertEquals($this->objFromFixture('HierarchyTest_Object', 'obj3b')->numChildren(), 0);
|
|
|
|
$obj1 = $this->objFromFixture('HierarchyTest_Object', 'obj1');
|
|
$this->assertEquals($obj1->numChildren(), 0);
|
|
$obj1Child1 = new HierarchyTest_Object();
|
|
$obj1Child1->ParentID = $obj1->ID;
|
|
$obj1Child1->write();
|
|
$this->assertEquals($obj1->numChildren(false), 1,
|
|
'numChildren() caching can be disabled through method parameter'
|
|
);
|
|
$obj1Child2 = new HierarchyTest_Object();
|
|
$obj1Child2->ParentID = $obj1->ID;
|
|
$obj1Child2->write();
|
|
$obj1->flushCache();
|
|
$this->assertEquals($obj1->numChildren(), 2,
|
|
'numChildren() caching can be disabled by flushCache()'
|
|
);
|
|
}
|
|
|
|
function testLoadDescendantIDListIntoArray() {
|
|
$obj2 = $this->objFromFixture('HierarchyTest_Object', 'obj2');
|
|
$obj2a = $this->objFromFixture('HierarchyTest_Object', 'obj2a');
|
|
$obj2b = $this->objFromFixture('HierarchyTest_Object', 'obj2b');
|
|
$obj2aa = $this->objFromFixture('HierarchyTest_Object', 'obj2aa');
|
|
$obj2ab = $this->objFromFixture('HierarchyTest_Object', 'obj2ab');
|
|
|
|
$obj2IdList = $obj2->getDescendantIDList();
|
|
$obj2aIdList = $obj2a->getDescendantIDList();
|
|
|
|
$this->assertContains($obj2a->ID, $obj2IdList);
|
|
$this->assertContains($obj2b->ID, $obj2IdList);
|
|
$this->assertContains($obj2aa->ID, $obj2IdList);
|
|
$this->assertContains($obj2ab->ID, $obj2IdList);
|
|
$this->assertEquals(4, count($obj2IdList));
|
|
|
|
$this->assertContains($obj2aa->ID, $obj2aIdList);
|
|
$this->assertContains($obj2ab->ID, $obj2aIdList);
|
|
$this->assertEquals(2, count($obj2aIdList));
|
|
}
|
|
|
|
/**
|
|
* The "only deleted from stage" argument to liveChildren() should exclude
|
|
* any page that has been moved to another location on the stage site
|
|
*/
|
|
function testLiveChildrenOnlyDeletedFromStage() {
|
|
$obj1 = $this->objFromFixture('HierarchyTest_Object', 'obj1');
|
|
$obj2 = $this->objFromFixture('HierarchyTest_Object', 'obj2');
|
|
$obj2a = $this->objFromFixture('HierarchyTest_Object', 'obj2a');
|
|
$obj2b = $this->objFromFixture('HierarchyTest_Object', 'obj2b');
|
|
|
|
// Get a published set of objects for our fixture
|
|
$obj1->publish("Stage", "Live");
|
|
$obj2->publish("Stage", "Live");
|
|
$obj2a->publish("Stage", "Live");
|
|
$obj2b->publish("Stage", "Live");
|
|
|
|
// Then delete 2a from stage and move 2b to a sub-node of 1.
|
|
$obj2a->delete();
|
|
$obj2b->ParentID = $obj1->ID;
|
|
$obj2b->write();
|
|
|
|
// Get live children, excluding pages that have been moved on the stage site
|
|
$children = $obj2->liveChildren(true, true)->column("Title");
|
|
|
|
// 2a has been deleted from stage and should be shown
|
|
$this->assertContains("Obj 2a", $children);
|
|
|
|
// 2b has merely been moved to a different parent and so shouldn't be shown
|
|
$this->assertNotContains("Obj 2b", $children);
|
|
}
|
|
|
|
function testBreadcrumbs() {
|
|
$obj1 = $this->objFromFixture('HierarchyTest_Object', 'obj1');
|
|
$obj2 = $this->objFromFixture('HierarchyTest_Object', 'obj2');
|
|
$obj2a = $this->objFromFixture('HierarchyTest_Object', 'obj2a');
|
|
$obj2aa = $this->objFromFixture('HierarchyTest_Object', 'obj2aa');
|
|
|
|
$this->assertEquals('Obj 1', $obj1->getBreadcrumbs());
|
|
$this->assertEquals('Obj 2 » Obj 2a', $obj2a->getBreadcrumbs());
|
|
$this->assertEquals('Obj 2 » Obj 2a » Obj 2aa', $obj2aa->getBreadcrumbs());
|
|
}
|
|
|
|
}
|
|
|
|
class HierarchyTest_Object extends DataObject implements TestOnly {
|
|
static $db = array(
|
|
'Title' => 'Varchar'
|
|
);
|
|
|
|
static $extensions = array(
|
|
'Hierarchy',
|
|
"Versioned('Stage', 'Live')",
|
|
);
|
|
}
|