mirror of
https://github.com/silverstripe/silverstripe-framework
synced 2024-09-06 01:31:22 +02:00
82 lines
3.2 KiB
PHP
82 lines
3.2 KiB
PHP
<?php
|
|
|
|
namespace SilverStripe\ORM\Tests;
|
|
|
|
use SilverStripe\Dev\SapphireTest;
|
|
use SilverStripe\ORM\Tests\DataObjectTest\Team;
|
|
use SilverStripe\ORM\Tests\HasManyListTest\Company;
|
|
use SilverStripe\ORM\Tests\HasManyListTest\CompanyCar;
|
|
use SilverStripe\ORM\Tests\HasManyListTest\Employee;
|
|
|
|
class HasManyListTest extends SapphireTest
|
|
{
|
|
|
|
// Borrow the model from DataObjectTest
|
|
protected static $fixture_file = [
|
|
'DataObjectTest.yml', // Borrow the model from DataObjectTest
|
|
'HasManyListTest.yml',
|
|
];
|
|
|
|
public static $extra_data_objects = array(
|
|
Company::class,
|
|
Employee::class,
|
|
CompanyCar::class,
|
|
);
|
|
|
|
public static function getExtraDataObjects()
|
|
{
|
|
return array_merge(
|
|
DataObjectTest::$extra_data_objects,
|
|
ManyManyListTest::$extra_data_objects,
|
|
static::$extra_data_objects
|
|
);
|
|
}
|
|
|
|
public function testRelationshipEmptyOnNewRecords()
|
|
{
|
|
// Relies on the fact that (unrelated) comments exist in the fixture file already
|
|
$newTeam = new Team(); // has_many Comments
|
|
$this->assertEquals(array(), $newTeam->Comments()->column('ID'));
|
|
}
|
|
|
|
/**
|
|
* Test that related objects can be removed from a relation
|
|
*/
|
|
public function testRemoveRelation()
|
|
{
|
|
|
|
// Check that expected teams exist
|
|
$list = 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::class, 'team1');
|
|
$team2 = $this->objFromFixture(DataObjectTest\Team::class, 'team2');
|
|
$this->assertEquals(array('Bob', 'Joe'), $team1->Comments()->sort('Name')->column('Name'));
|
|
$this->assertEquals(array('Phil'), $team2->Comments()->sort('Name')->column('Name'));
|
|
|
|
// Test that removing comments from unrelated team has no effect
|
|
$team1comment = $this->objFromFixture(DataObjectTest\TeamComment::class, 'comment1');
|
|
$team2comment = $this->objFromFixture(DataObjectTest\TeamComment::class, 'comment3');
|
|
$team1->Comments()->remove($team2comment);
|
|
$team2->Comments()->remove($team1comment);
|
|
$this->assertEquals(array('Bob', 'Joe'), $team1->Comments()->sort('Name')->column('Name'));
|
|
$this->assertEquals(array('Phil'), $team2->Comments()->sort('Name')->column('Name'));
|
|
$this->assertEquals($team1->ID, $team1comment->TeamID);
|
|
$this->assertEquals($team2->ID, $team2comment->TeamID);
|
|
|
|
// Test that removing items from the related team resets the has_one relations on the fan
|
|
$team1comment = $this->objFromFixture(DataObjectTest\TeamComment::class, 'comment1');
|
|
$team2comment = $this->objFromFixture(DataObjectTest\TeamComment::class, 'comment3');
|
|
$team1->Comments()->remove($team1comment);
|
|
$team2->Comments()->remove($team2comment);
|
|
$this->assertEquals(array('Bob'), $team1->Comments()->sort('Name')->column('Name'));
|
|
$this->assertEquals(array(), $team2->Comments()->sort('Name')->column('Name'));
|
|
$this->assertEmpty($team1comment->TeamID);
|
|
$this->assertEmpty($team2comment->TeamID);
|
|
}
|
|
}
|