silverstripe-framework/tests/php/ORM/HasManyListTest.php
2018-09-27 17:09:24 +02:00

117 lines
4.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
{
protected static $fixture_file = [
'DataObjectTest.yml', // Borrow the model from DataObjectTest
'HasManyListTest.yml',
];
public static $extra_data_objects = [
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([], $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(
['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(['Bob', 'Joe'], $team1->Comments()->sort('Name')->column('Name'));
$this->assertEquals(['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(['Bob', 'Joe'], $team1->Comments()->sort('Name')->column('Name'));
$this->assertEquals(['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(['Bob'], $team1->Comments()->sort('Name')->column('Name'));
$this->assertEquals([], $team2->Comments()->sort('Name')->column('Name'));
$this->assertEmpty($team1comment->TeamID);
$this->assertEmpty($team2comment->TeamID);
}
public function testDefaultSortIsUsedOnList()
{
/** @var Company $company */
$company = $this->objFromFixture(Company::class, 'silverstripe');
$this->assertListEquals([
['Make' => 'Ferrari'],
['Make' => 'Jaguar'],
['Make' => 'Lamborghini'],
], $company->CompanyCars());
}
public function testCanBeSortedDescending()
{
/** @var Company $company */
$company = $this->objFromFixture(Company::class, 'silverstripe');
$this->assertListEquals([
['Make' => 'Lamborghini'],
['Make' => 'Jaguar'],
['Make' => 'Ferrari'],
], $company->CompanyCars()->sort('"Make" DESC'));
}
public function testSortByModel()
{
/** @var Company $company */
$company = $this->objFromFixture(Company::class, 'silverstripe');
$this->assertListEquals([
['Model' => 'Countach'],
['Model' => 'E Type'],
['Model' => 'F40'],
], $company->CompanyCars()->sort('"Model" ASC'));
}
}