[2.0] Fixed issue with self-referential one-to-many associations not being...

[2.0] Fixed issue with self-referential one-to-many associations not being persisted correctly when IDENTITY key generation was used. Included now passing OneToManySelfReferentialTest.

lib/Doctrine/ORM/Persisters/JoinedSubclassPersister.php

@@ -187,8 +187,8 @@ class JoinedSubclassPersister extends StandardEntityPersister
         $this->_prepareData($entity, $updateData);
 
         $id = array_combine(
-        $this->_class->getIdentifierFieldNames(),
-        $this->_em->getUnitOfWork()->getEntityIdentifier($entity)
+            $this->_class->getIdentifierFieldNames(),
+            $this->_em->getUnitOfWork()->getEntityIdentifier($entity)
         );
 
         foreach ($updateData as $tableName => $data) {
@@ -205,15 +205,15 @@ class JoinedSubclassPersister extends StandardEntityPersister
     public function delete($entity)
     {
         $id = array_combine(
-        $this->_class->getIdentifierFieldNames(),
-        $this->_em->getUnitOfWork()->getEntityIdentifier($entity)
+            $this->_class->getIdentifierFieldNames(),
+            $this->_em->getUnitOfWork()->getEntityIdentifier($entity)
         );
 
         // If the database platform supports FKs, just
         // delete the row from the root table. Cascades do the rest.
         if ($this->_conn->getDatabasePlatform()->supportsForeignKeyConstraints()) {
             $this->_conn->delete($this->_em->getClassMetadata($this->_class->rootEntityName)
-            ->primaryTable['name'], $id);
+                    ->primaryTable['name'], $id);
         } else {
             // Delete the parent tables, starting from this class' table up to the root table
             $this->_conn->delete($this->_class->primaryTable['name'], $id);

lib/Doctrine/ORM/Persisters/StandardEntityPersister.php

@@ -179,16 +179,18 @@ class StandardEntityPersister
     {
         $updateData = array();
         $this->_prepareData($entity, $updateData);
-        $id = array_combine($this->_class->getIdentifierFieldNames(),
-        $this->_em->getUnitOfWork()->getEntityIdentifier($entity));
+        $id = array_combine(
+            $this->_class->getIdentifierFieldNames(),
+            $this->_em->getUnitOfWork()->getEntityIdentifier($entity)
+        );
         $tableName = $this->_class->primaryTable['name'];
-        
+
         if ($this->_evm->hasListeners(Events::preUpdate)) {
             $this->_preUpdate($entity);
         }
-        
+
         $this->_conn->update($tableName, $updateData[$tableName], $id);
-        
+
         if ($this->_evm->hasListeners(Events::postUpdate)) {
             $this->_postUpdate($entity);
         }
@@ -239,9 +241,10 @@ class StandardEntityPersister
     }
 
     /**
-     * Prepares the data changeset of an entity for database insertion.
-     * The array that is passed as the second parameter is filled with
-     * <columnName> => <value> pairs, grouped by table name, during this preparation.
+     * Prepares the data changeset of an entity for database insertion (INSERT/UPDATE).
+     * 
+     * During this preparation the array that is passed as the second parameter is filled with
+     * <columnName> => <value> pairs, grouped by table name.
      *
      * Example:
      * <code>
@@ -256,7 +259,7 @@ class StandardEntityPersister
      *
      * @param object $entity
      * @param array $result The reference to the data array.
-     * @param boolean $isInsert
+     * @param boolean $isInsert Whether the preparation is for an INSERT (or UPDATE, if FALSE).
      */
     protected function _prepareData($entity, array &$result, $isInsert = false)
     {
@@ -276,37 +279,43 @@ class StandardEntityPersister
                     continue;
                 }
 
-                // Special case: One-one self-referencing of the same class.
-                if ($newVal !== null && $assocMapping->sourceEntityName == $assocMapping->targetEntityName) {
+                // Special case: One-one self-referencing of the same class with IDENTITY type key generation.
+                if ($this->_class->isIdGeneratorIdentity() && $newVal !== null &&
+                        $assocMapping->sourceEntityName == $assocMapping->targetEntityName) {
                     $oid = spl_object_hash($newVal);
                     $isScheduledForInsert = $uow->isRegisteredNew($newVal);
                     if (isset($this->_queuedInserts[$oid]) || $isScheduledForInsert) {
                         // The associated entity $newVal is not yet persisted, so we must
-                        // set $newVal = null, in order to insert a null value and update later.
+                        // set $newVal = null, in order to insert a null value and schedule an
+                        // extra update on the UnitOfWork.
+                        $uow->scheduleExtraUpdate($entity, array(
+                            $field => array(null, $newVal)
+                        ));
                         $newVal = null;
                     } else if ($isInsert && ! $isScheduledForInsert && $uow->getEntityState($newVal) == UnitOfWork::STATE_MANAGED) {
-                        // $newVal is already fully persisted
-                        // Clear changeset of $newVal, so that only the identifier is updated.
-                        // Not sure this is really rock-solid here but it seems to work.
-                        $uow->clearEntityChangeSet($oid);
-                        $uow->propertyChanged($newVal, $field, $entity, $entity);
+                        // $newVal is already fully persisted.
+                        // Schedule an extra update for it, so that the foreign key(s) are properly set.
+                        $uow->scheduleExtraUpdate($newVal, array(
+                            $field => array(null, $entity)
+                        ));
                     }
                 }
-
+                
                 foreach ($assocMapping->sourceToTargetKeyColumns as $sourceColumn => $targetColumn) {
-                    $otherClass = $this->_em->getClassMetadata($assocMapping->targetEntityName);
                     if ($newVal === null) {
                         $result[$this->getOwningTable($field)][$sourceColumn] = null;
                     } else {
+                        $otherClass = $this->_em->getClassMetadata($assocMapping->targetEntityName);
                         $result[$this->getOwningTable($field)][$sourceColumn] =
-                        $otherClass->reflFields[$otherClass->fieldNames[$targetColumn]]->getValue($newVal);
+                                $otherClass->reflFields[$otherClass->fieldNames[$targetColumn]]
+                                ->getValue($newVal);
                     }
                 }
             } else if ($newVal === null) {
                 $result[$this->getOwningTable($field)][$columnName] = null;
             } else {
                 $result[$this->getOwningTable($field)][$columnName] = Type::getType(
-                        $this->_class->fieldMappings[$field]['type'])->convertToDatabaseValue($newVal, $platform);
+                $this->_class->fieldMappings[$field]['type'])->convertToDatabaseValue($newVal, $platform);
             }
         }
     }
@@ -404,7 +413,7 @@ class StandardEntityPersister
         return 'SELECT ' . $columnList . ' FROM ' . $this->_class->getTableName()
                 . ' WHERE ' . $conditionSql;
     }
-    
+
     /**
      * Dispatches the preInsert event for the given entity.
      *
@@ -417,7 +426,7 @@ class StandardEntityPersister
         );
         $this->_evm->dispatchEvent(Events::preInsert, $eventArgs);
     }
-    
+
     /**
      * Dispatches the postInsert event for the given entity.
      *
@@ -427,7 +436,7 @@ class StandardEntityPersister
     {
         $this->_evm->dispatchEvent(Events::postInsert);
     }
-    
+
     /**
      * Dispatches the preUpdate event for the given entity.
      *
@@ -440,7 +449,7 @@ class StandardEntityPersister
         );
         $this->_evm->dispatchEvent(Events::preUpdate, $eventArgs);
     }
-    
+
     /**
      * Dispatches the postUpdate event for the given entity.
      *

tests/Doctrine/Tests/Models/ECommerce/ECommerceCart.php

@@ -2,6 +2,8 @@
 
 namespace Doctrine\Tests\Models\ECommerce;
 
+use Doctrine\Common\Collections\Collection;
+
 /**
  * ECommerceCart
  * Represents a typical cart of a shopping application.
@@ -40,7 +42,7 @@ class ECommerceCart
 
     public function __construct()
     {
-        $this->products = new \Doctrine\Common\Collections\Collection;
+        $this->products = new Collection;
     }
     
     public function getId() {

tests/Doctrine/Tests/Models/ECommerce/ECommerceCategory.php

@@ -2,6 +2,8 @@
 
 namespace Doctrine\Tests\Models\ECommerce;
 
+use Doctrine\Common\Collections\Collection;
+
 /**
  * ECommerceCategory
  * Represents a tag applied on particular products.
@@ -42,8 +44,8 @@ class ECommerceCategory
 
     public function __construct()
     {
-        $this->products = new \Doctrine\Common\Collections\Collection();
-        $this->children = new \Doctrine\Common\Collections\Collection();
+        $this->products = new Collection();
+        $this->children = new Collection();
     }
 
     public function getId()

tests/Doctrine/Tests/Models/ECommerce/ECommerceProduct.php

@@ -2,6 +2,8 @@
 
 namespace Doctrine\Tests\Models\ECommerce;
 
+use Doctrine\Common\Collections\Collection;
+
 /**
  * ECommerceProduct
  * Represents a type of product of a shopping application.
@@ -45,8 +47,8 @@ class ECommerceProduct
 
     public function __construct()
     {
-        $this->features = new \Doctrine\Common\Collections\Collection;
-        $this->categories = new \Doctrine\Common\Collections\Collection;
+        $this->features = new Collection;
+        $this->categories = new Collection;
     }
 
     public function getId()

tests/Doctrine/Tests/Models/ECommerce/ECommerceShipping.php

@@ -13,8 +13,7 @@ namespace Doctrine\Tests\Models\ECommerce;
 class ECommerceShipping
 {
     /**
-     * @Column(type="integer")
-     * @Id
+     * @Id @Column(type="integer")
      * @GeneratedValue(strategy="AUTO")
      */
     private $id;

tests/Doctrine/Tests/ORM/Functional/AbstractManyToManyAssociationTestCase.php

@@ -38,7 +38,9 @@ class AbstractManyToManyAssociationTestCase extends \Doctrine\Tests\OrmFunctiona
 
     public function assertCollectionEquals(Collection $first, Collection $second)
     {
-        if (count($first) != count($second)) {
+        return $first->forAll(function($k, $e) use($second) { return $second->contains($e); });
+        
+        /*if (count($first) != count($second)) {
             return false;
         }
         foreach ($first as $element) {
@@ -46,6 +48,6 @@ class AbstractManyToManyAssociationTestCase extends \Doctrine\Tests\OrmFunctiona
                 return false;
             }
         }
-        return true;
+        return true;*/
     }
 }

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\OneToManySelfReferentialAssociationTest');
 
         return $suite;
     }

tests/Doctrine/Tests/ORM/Functional/OneToManySelfReferentialAssociationTest.php

@@ -32,6 +32,8 @@ class OneToManySelfReferentialAssociationTest extends \Doctrine\Tests\OrmFunctio
         $this->parent->addChild($this->secondChild);
         $this->_em->save($this->parent);
         
+        $this->_em->flush();
+        
         $this->assertForeignKeyIs($this->parent->getId(), $this->firstChild);
         $this->assertForeignKeyIs($this->parent->getId(), $this->secondChild);
     }
@@ -39,6 +41,7 @@ class OneToManySelfReferentialAssociationTest extends \Doctrine\Tests\OrmFunctio
     public function testSavesAnEmptyCollection()
     {
         $this->_em->save($this->parent);
+        $this->_em->flush();
 
         $this->assertEquals(0, count($this->parent->getChildren()));
     }
@@ -46,6 +49,7 @@ class OneToManySelfReferentialAssociationTest extends \Doctrine\Tests\OrmFunctio
     public function testDoesNotSaveAnInverseSideSet() {
         $this->parent->brokenAddChild($this->firstChild);
         $this->_em->save($this->parent);
+        $this->_em->flush();
         
         $this->assertForeignKeyIs(null, $this->firstChild);
     }
@@ -80,10 +84,10 @@ class OneToManySelfReferentialAssociationTest extends \Doctrine\Tests\OrmFunctio
         
         $this->assertTrue($children[0] instanceof ECommerceCategory);
         $this->assertSame($parent, $children[0]->getParent());
-        $this->assertTrue(strstr($children[0]->getName(), ' books'));
+        $this->assertEquals(' books', strstr($children[0]->getName(), ' books'));
         $this->assertTrue($children[1] instanceof ECommerceCategory);
         $this->assertSame($parent, $children[1]->getParent());
-        $this->assertTrue(strstr($children[1]->getName(), ' books'));
+        $this->assertEquals(' books', strstr($children[1]->getName(), ' books'));
     }
     
     /* TODO: not yet implemented