silverstripe-framework/tests/php/ORM/UnsavedRelationListTest.php

<?php

namespace SilverStripe\ORM\Tests;

use SilverStripe\Dev\SapphireTest;
use SilverStripe\ORM\RelationList;

class UnsavedRelationListTest extends SapphireTest
{
    protected static $fixture_file = 'UnsavedRelationListTest.yml';

    protected $extraDataObjects = [
        UnsavedRelationListTest\TestObject::class
    ];

    public function testReturnedList()
    {
        $object = new UnsavedRelationListTest\TestObject();
        $children = $object->Children();
        $siblings = $object->Siblings();
        $this->assertEquals(
            $children,
            $object->Children(),
            'Returned UnsavedRelationList should be the same.'
        );
        $this->assertEquals(
            $siblings,
            $object->Siblings(),
            'Returned UnsavedRelationList should be the same.'
        );

        $object->write();
        $this->assertInstanceOf(RelationList::class, $object->Children());
        $this->assertNotEquals(
            $children,
            $object->Children(),
            'Return should be a RelationList after first write'
        );
        $this->assertInstanceOf(RelationList::class, $object->Siblings());
        $this->assertNotEquals(
            $siblings,
            $object->Siblings(),
            'Return should be a RelationList after first write'
        );
    }

    public function testHasManyExisting()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $children = $object->Children();
        $children->add($this->objFromFixture(UnsavedRelationListTest\TestObject::class, 'ObjectA'));
        $children->add($this->objFromFixture(UnsavedRelationListTest\TestObject::class, 'ObjectB'));
        $children->add($this->objFromFixture(UnsavedRelationListTest\TestObject::class, 'ObjectC'));

        $children = $object->Children();

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $children
        );

        $object->write();

        $this->assertNotEquals($children, $object->Children());

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $object->Children()
        );
    }

    public function testManyManyExisting()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $Siblings = $object->Siblings();
        $Siblings->add($this->objFromFixture(UnsavedRelationListTest\TestObject::class, 'ObjectA'));
        $Siblings->add($this->objFromFixture(UnsavedRelationListTest\TestObject::class, 'ObjectB'));
        $Siblings->add($this->objFromFixture(UnsavedRelationListTest\TestObject::class, 'ObjectC'));

        $siblings = $object->Siblings();

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $siblings
        );

        $object->write();

        $this->assertNotEquals($siblings, $object->Siblings());

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $object->Siblings()
        );
    }

    public function testHasManyNew()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $children = $object->Children();
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'A')));
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'B')));
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'C')));

        $children = $object->Children();

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $children
        );

        $object->write();

        $this->assertNotEquals($children, $object->Children());

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $object->Children()
        );
    }

    public function testHasManyPolymorphic()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $children = $object->RelatedObjects();
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'A')));
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'B')));
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'C')));

        $children = $object->RelatedObjects();

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $children
        );

        $object->write();

        $this->assertNotEquals($children, $object->RelatedObjects());

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $object->RelatedObjects()
        );
    }

    public function testManyManyNew()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $Siblings = $object->Siblings();
        $Siblings->add(new UnsavedRelationListTest\TestObject(array('Name' => 'A')));
        $Siblings->add(new UnsavedRelationListTest\TestObject(array('Name' => 'B')));
        $Siblings->add(new UnsavedRelationListTest\TestObject(array('Name' => 'C')));

        $siblings = $object->Siblings();

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $siblings
        );

        $object->write();

        $this->assertNotEquals($siblings, $object->Siblings());

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $object->Siblings()
        );
    }

    public function testManyManyExtraFields()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $Siblings = $object->Siblings();
        $Siblings->add(new UnsavedRelationListTest\TestObject(array('Name' => 'A')), array('Number' => 1));
        $Siblings->add(new UnsavedRelationListTest\TestObject(array('Name' => 'B')), array('Number' => 2));
        $Siblings->add(new UnsavedRelationListTest\TestObject(array('Name' => 'C')), array('Number' => 3));

        $siblings = $object->Siblings();

        $this->assertDOSEquals(
            array(
            array('Name' => 'A', 'Number' => 1),
            array('Name' => 'B', 'Number' => 2),
            array('Name' => 'C', 'Number' => 3)
            ),
            $siblings
        );

        $object->write();

        $this->assertNotEquals($siblings, $object->Siblings());

        $this->assertDOSEquals(
            array(
            array('Name' => 'A', 'Number' => 1),
            array('Name' => 'B', 'Number' => 2),
            array('Name' => 'C', 'Number' => 3)
            ),
            $object->Siblings()
        );
    }

    public function testGetIDList()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $children = $object->Children();
        $this->assertEquals($children->getIDList(), array());
        $children->add($child1 = new UnsavedRelationListTest\TestObject(array('Name' => 'A')));
        $children->add($child2 = new UnsavedRelationListTest\TestObject(array('Name' => 'B')));
        $children->add($child3 = new UnsavedRelationListTest\TestObject(array('Name' => 'C')));
        $children->add($child1);

        $this->assertEquals($children->getIDList(), array());

        $child1->write();
        $this->assertEquals(
            $children->getIDList(),
            array(
            $child1->ID => $child1->ID
            )
        );

        $child2->write();
        $child3->write();
        $this->assertEquals(
            $children->getIDList(),
            array(
            $child1->ID => $child1->ID,
            $child2->ID => $child2->ID,
            $child3->ID => $child3->ID
            )
        );
    }

    public function testColumn()
    {
        $object = new UnsavedRelationListTest\TestObject();

        $children = $object->Children();
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'A')));
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'B')));
        $children->add(new UnsavedRelationListTest\TestObject(array('Name' => 'C')));

        $children = $object->Children();

        $this->assertDOSEquals(
            array(
            array('Name' => 'A'),
            array('Name' => 'B'),
            array('Name' => 'C')
            ),
            $children
        );

        $this->assertEquals(
            $children->column('Name'),
            array(
            'A',
            'B',
            'C'
            )
        );
    }
}