From 49103096746173059ef41d6ceba020f328214266 Mon Sep 17 00:00:00 2001
From: piccoloprincipe <piccoloprincipe@625475ce-881a-0410-a577-b389adb331d8>
Date: Mon, 6 Jul 2009 12:42:14 +0000
Subject: [PATCH] added one-one self referential functional tests (closes
 #2276)

---
 .../Models/ECommerce/ECommerceCustomer.php    | 24 ++++++
 .../Tests/ORM/Functional/AllTests.php         |  1 +
 ...OneToOneSelfReferentialAssociationTest.php | 79 +++++++++++++++++++
 3 files changed, 104 insertions(+)
 create mode 100644 tests/Doctrine/Tests/ORM/Functional/OneToOneSelfReferentialAssociationTest.php

diff --git a/tests/Doctrine/Tests/Models/ECommerce/ECommerceCustomer.php b/tests/Doctrine/Tests/Models/ECommerce/ECommerceCustomer.php
index c5c8f19f1..a35b5acb8 100644
--- a/tests/Doctrine/Tests/Models/ECommerce/ECommerceCustomer.php
+++ b/tests/Doctrine/Tests/Models/ECommerce/ECommerceCustomer.php
@@ -28,6 +28,15 @@ class ECommerceCustomer
      * @OneToOne(targetEntity="ECommerceCart", mappedBy="customer", cascade={"save"})
      */
     private $cart;
+
+    /**
+     * Example of a one-one self referential association. A mentor can follow
+     * only one customer at the time, while a customer can choose only one
+     * mentor. Not properly appropriate but it works.
+     * @OneToOne(targetEntity="ECommerceCustomer", cascade={"save"})
+     * @JoinColumn(name="mentor_id", referencedColumnName="id")
+     */
+    private $mentor;
     
     public function getId() {
         return $this->id;
@@ -66,4 +75,19 @@ class ECommerceCustomer
             $cart->removeCustomer();
         }
     }
+
+    public function setMentor(ECommerceCustomer $mentor)
+    {
+        $this->mentor = $mentor;
+    }
+
+    public function removeMentor()
+    {
+        $this->mentor = null;
+    }
+
+    public function getMentor()
+    {
+        return $this->mentor;
+    }
 }
diff --git a/tests/Doctrine/Tests/ORM/Functional/AllTests.php b/tests/Doctrine/Tests/ORM/Functional/AllTests.php
index d792293ed..43fe7034f 100644
--- a/tests/Doctrine/Tests/ORM/Functional/AllTests.php
+++ b/tests/Doctrine/Tests/ORM/Functional/AllTests.php
@@ -31,6 +31,7 @@ class AllTests
         $suite->addTestSuite('Doctrine\Tests\ORM\Functional\OneToManyBidirectionalAssociationTest');
         $suite->addTestSuite('Doctrine\Tests\ORM\Functional\ManyToManyUnidirectionalAssociationTest');
         $suite->addTestSuite('Doctrine\Tests\ORM\Functional\ManyToManyBidirectionalAssociationTest');
+        $suite->addTestSuite('Doctrine\Tests\ORM\Functional\OneToOneSelfReferentialAssociationTest');
         $suite->addTestSuite('Doctrine\Tests\ORM\Functional\OneToManySelfReferentialAssociationTest');
         $suite->addTestSuite('Doctrine\Tests\ORM\Functional\ManyToManySelfReferentialAssociationTest');
 
diff --git a/tests/Doctrine/Tests/ORM/Functional/OneToOneSelfReferentialAssociationTest.php b/tests/Doctrine/Tests/ORM/Functional/OneToOneSelfReferentialAssociationTest.php
new file mode 100644
index 000000000..0c588d8a1
--- /dev/null
+++ b/tests/Doctrine/Tests/ORM/Functional/OneToOneSelfReferentialAssociationTest.php
@@ -0,0 +1,79 @@
+<?php
+
+namespace Doctrine\Tests\ORM\Functional;
+
+use Doctrine\Tests\Models\ECommerce\ECommerceCustomer;
+
+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);
+        $this->_em->save($this->customer);
+        
+        $this->assertForeignKeyIs($this->mentor->getId());
+    }
+
+    public function testRemovesOneToOneAssociation()
+    {
+        $this->customer->setMentor($this->mentor);
+        $this->_em->save($this->customer);
+        $this->customer->removeMentor();
+
+        $this->_em->flush();
+
+        $this->assertForeignKeyIs(null);
+    }
+
+    public function testEagerLoad()
+    {
+        $customer = new ECommerceCustomer;
+        $customer->setName('Luke Skywalker');
+        $mentor = new ECommerceCustomer;
+        $mentor->setName('Obi-wan Kenobi');
+        $customer->setMentor($mentor);
+        
+        $this->_em->save($customer);
+        
+        $this->_em->flush();
+        $this->_em->clear();
+
+        $query = $this->_em->createQuery('select c, m from Doctrine\Tests\Models\ECommerce\ECommerceCustomer c left join c.mentor m');
+        $result = $query->getResultList();
+        $customer = $result[0];
+        
+        $this->assertTrue($customer->getMentor() instanceof ECommerceCustomer);
+        $this->assertEquals('Obi-wan Kenobi', $customer->getMentor()->getName());
+    }
+    
+    /* TODO: not yet implemented
+    public function testLazyLoad() {
+        
+    }*/
+
+    public function assertForeignKeyIs($value) {
+        $foreignKey = $this->_em->getConnection()->execute('SELECT mentor_id FROM ecommerce_customers WHERE id=?', array($this->customer->getId()))->fetchColumn();
+        $this->assertEquals($value, $foreignKey);
+    }
+}