silverstripe-framework/tests/model/PolymorphicHasManyListTest.php

121 lines
4.5 KiB
PHP

<?php
/**
* Tests the PolymorphicHasManyList class
*
* @see PolymorphicHasManyList
*
* @todo Complete
*
* @package framework
* @subpackage tests
*/
class PolymorphicHasManyListTest extends SapphireTest {
// Borrow the model from DataObjectTest
protected static $fixture_file = 'DataObjectTest.yml';
protected $extraDataObjects = array(
'DataObjectTest_Team',
'DataObjectTest_Fixture',
'DataObjectTest_SubTeam',
'OtherSubclassWithSameField',
'DataObjectTest_FieldlessTable',
'DataObjectTest_FieldlessSubTable',
'DataObjectTest_ValidatedObject',
'DataObjectTest_Player',
'DataObjectTest_TeamComment',
'DataObjectTest_EquipmentCompany',
'DataObjectTest_SubEquipmentCompany',
'DataObjectTest\NamespacedClass',
'DataObjectTest\RelationClass',
'DataObjectTest_ExtendedTeamComment',
'DataObjectTest_Company',
'DataObjectTest_Staff',
'DataObjectTest_CEO',
'DataObjectTest_Fan',
);
public function testRelationshipEmptyOnNewRecords() {
// Relies on the fact that (unrelated) comments exist in the fixture file already
$newTeam = new DataObjectTest_Team(); // has_many Comments
$this->assertEquals(array(), $newTeam->Fans()->column('ID'));
}
/**
* Test that DataList::relation works with PolymorphicHasManyList
*/
public function testFilterRelation() {
// Check that expected teams exist
$list = DataObjectTest_Team::get();
$this->assertEquals(
array('Subteam 1', 'Subteam 2', 'Subteam 3', 'Team 1', 'Team 2', 'Team 3'),
$list->sort('Title')->column('Title')
);
// Check that fan list exists
$fans = $list->relation('Fans');
$this->assertEquals(array('Damian', 'Mitch', 'Richard'), $fans->sort('Name')->column('Name'));
// Modify list of fans and retest
$team1 = $this->objFromFixture('DataObjectTest_Team', 'team1');
$subteam1 = $this->objFromFixture('DataObjectTest_SubTeam', 'subteam1');
$newFan1 = DataObjectTest_Fan::create();
$newFan1->Name = 'Bobby';
$newFan1->write();
$newFan2 = DataObjectTest_Fan::create();
$newFan2->Name = 'Mindy';
$newFan2->write();
$team1->Fans()->add($newFan1);
$subteam1->Fans()->add($newFan2);
$fans = DataObjectTest_Team::get()->relation('Fans');
$this->assertEquals(array('Bobby', 'Damian', 'Richard'), $team1->Fans()->sort('Name')->column('Name'));
$this->assertEquals(array('Mindy', 'Mitch'), $subteam1->Fans()->sort('Name')->column('Name'));
$this->assertEquals(array('Bobby', 'Damian', 'Mindy', 'Mitch', 'Richard'), $fans->sort('Name')->column('Name'));
}
/**
* Test that related objects can be removed from a relation
*/
public function testRemoveRelation() {
// Check that expected teams exist
$list = DataObjectTest_Team::get();
$this->assertEquals(
array('Subteam 1', 'Subteam 2', 'Subteam 3', 'Team 1', 'Team 2', 'Team 3'),
$list->sort('Title')->column('Title')
);
// Test that each team has the correct fans
$team1 = $this->objFromFixture('DataObjectTest_Team', 'team1');
$subteam1 = $this->objFromFixture('DataObjectTest_SubTeam', 'subteam1');
$this->assertEquals(array('Damian', 'Richard'), $team1->Fans()->sort('Name')->column('Name'));
$this->assertEquals(array('Mitch'), $subteam1->Fans()->sort('Name')->column('Name'));
// Test that removing items from unrelated team has no effect
$team1fan = $this->objFromFixture('DataObjectTest_Fan', 'fan1');
$subteam1fan = $this->objFromFixture('DataObjectTest_Fan', 'fan4');
$team1->Fans()->remove($subteam1fan);
$subteam1->Fans()->remove($team1fan);
$this->assertEquals(array('Damian', 'Richard'), $team1->Fans()->sort('Name')->column('Name'));
$this->assertEquals(array('Mitch'), $subteam1->Fans()->sort('Name')->column('Name'));
$this->assertEquals($team1->ID, $team1fan->FavouriteID);
$this->assertEquals('DataObjectTest_Team', $team1fan->FavouriteClass);
$this->assertEquals($subteam1->ID, $subteam1fan->FavouriteID);
$this->assertEquals('DataObjectTest_SubTeam', $subteam1fan->FavouriteClass);
// Test that removing items from the related team resets the has_one relations on the fan
$team1fan = $this->objFromFixture('DataObjectTest_Fan', 'fan1');
$subteam1fan = $this->objFromFixture('DataObjectTest_Fan', 'fan4');
$team1->Fans()->remove($team1fan);
$subteam1->Fans()->remove($subteam1fan);
$this->assertEquals(array('Richard'), $team1->Fans()->sort('Name')->column('Name'));
$this->assertEquals(array(), $subteam1->Fans()->sort('Name')->column('Name'));
$this->assertEmpty($team1fan->FavouriteID);
$this->assertEmpty($team1fan->FavouriteClass);
$this->assertEmpty($subteam1fan->FavouriteID);
$this->assertEmpty($subteam1fan->FavouriteClass);
}
}