1
0
mirror of synced 2024-12-16 08:06:02 +03:00
doctrine2/tests/Doctrine/Tests/ORM/Functional/OneToOneSelfReferentialAssociationTest.php

155 lines
5.2 KiB
PHP
Raw Normal View History

<?php
namespace Doctrine\Tests\ORM\Functional;
use Doctrine\Tests\Models\ECommerce\ECommerceCustomer;
use Doctrine\ORM\Mapping\AssociationMapping;
require_once __DIR__ . '/../../TestInit.php';
/**
* Tests a self referential one-to-one association mapping (without inheritance).
* Relation is defined as the mentor that a customer choose. The mentor could
* help only one other customer, while a customer can choose only one mentor
* for receiving support.
* Inverse side is not present.
*/
class OneToOneSelfReferentialAssociationTest extends \Doctrine\Tests\OrmFunctionalTestCase
{
private $customer;
private $mentor;
protected function setUp()
{
$this->useModelSet('ecommerce');
parent::setUp();
$this->customer = new ECommerceCustomer();
$this->customer->setName('Anakin Skywalker');
$this->mentor = new ECommerceCustomer();
$this->mentor->setName('Obi-wan Kenobi');
}
public function testSavesAOneToOneAssociationWithCascadeSaveSet() {
$this->customer->setMentor($this->mentor);
2009-07-19 20:54:53 +04:00
$this->_em->persist($this->customer);
$this->_em->flush();
$this->assertForeignKeyIs($this->mentor->getId());
}
public function testRemovesOneToOneAssociation()
{
$this->customer->setMentor($this->mentor);
2009-07-19 20:54:53 +04:00
$this->_em->persist($this->customer);
$this->customer->removeMentor();
$this->_em->flush();
$this->assertForeignKeyIs(null);
}
public function testEagerLoadsAssociation()
{
$this->_createFixture();
$query = $this->_em->createQuery('select c, m from Doctrine\Tests\Models\ECommerce\ECommerceCustomer c left join c.mentor m order by c.id asc');
$result = $query->getResult();
$customer = $result[0];
$this->assertLoadingOfAssociation($customer);
}
/**
* @group mine
* @return unknown_type
*/
public function testLazyLoadsAssociation()
{
$this->_createFixture();
$metadata = $this->_em->getClassMetadata('Doctrine\Tests\Models\ECommerce\ECommerceCustomer');
$metadata->getAssociationMapping('mentor')->fetchMode = AssociationMapping::FETCH_LAZY;
$query = $this->_em->createQuery("select c from Doctrine\Tests\Models\ECommerce\ECommerceCustomer c where c.name='Luke Skywalker'");
$result = $query->getResult();
$customer = $result[0];
$this->assertLoadingOfAssociation($customer);
}
2009-11-15 14:42:05 +03:00
public function testMultiSelfReference()
{
try {
$this->_schemaTool->createSchema(array(
$this->_em->getClassMetadata('Doctrine\Tests\ORM\Functional\MultiSelfReference')
));
} catch (\Exception $e) {
// Swallow all exceptions. We do not test the schema tool here.
}
$entity1 = new MultiSelfReference();
$this->_em->persist($entity1);
$entity1->setOther1($entity2 = new MultiSelfReference);
$entity1->setOther2($entity3 = new MultiSelfReference);
$this->_em->flush();
$this->_em->clear();
$entity2 = $this->_em->find(get_class($entity1), $entity1->getId());
$this->assertTrue($entity2->getOther1() instanceof MultiSelfReference);
$this->assertTrue($entity2->getOther2() instanceof MultiSelfReference);
$this->assertNull($entity2->getOther1()->getOther1());
$this->assertNull($entity2->getOther1()->getOther2());
$this->assertNull($entity2->getOther2()->getOther1());
$this->assertNull($entity2->getOther2()->getOther2());
}
public function assertLoadingOfAssociation($customer)
{
$this->assertTrue($customer->getMentor() instanceof ECommerceCustomer);
$this->assertEquals('Obi-wan Kenobi', $customer->getMentor()->getName());
}
public function assertForeignKeyIs($value) {
$foreignKey = $this->_em->getConnection()->executeQuery('SELECT mentor_id FROM ecommerce_customers WHERE id=?', array($this->customer->getId()))->fetchColumn();
$this->assertEquals($value, $foreignKey);
}
private function _createFixture()
{
$customer = new ECommerceCustomer;
$customer->setName('Luke Skywalker');
$mentor = new ECommerceCustomer;
$mentor->setName('Obi-wan Kenobi');
$customer->setMentor($mentor);
$this->_em->persist($customer);
$this->_em->flush();
$this->_em->clear();
}
}
2009-11-15 14:42:05 +03:00
/**
* @Entity
*/
class MultiSelfReference {
/** @Id @GeneratedValue(strategy="AUTO") @Column(type="integer") */
private $id;
/**
* @OneToOne(targetEntity="MultiSelfReference", cascade={"persist"})
* @JoinColumn(name="other1", referencedColumnName="id")
*/
private $other1;
/**
* @OneToOne(targetEntity="MultiSelfReference", cascade={"persist"})
* @JoinColumn(name="other2", referencedColumnName="id")
*/
private $other2;
public function getId() {return $this->id;}
public function setOther1($other1) {$this->other1 = $other1;}
public function getOther1() {return $this->other1;}
public function setOther2($other2) {$this->other2 = $other2;}
public function getOther2() {return $this->other2;}
}