Cherry pick unit test from PR doctrine#5570 (Fix PrePersist EventListener when using merge instead of persist)

bilouwan · alexgurrola · commit d3f8eccce67e · 2017-04-13T15:46:57.000-07:00
diff --git a/tests/Doctrine/Tests/Models/Company/CompanyContractListener.php b/tests/Doctrine/Tests/Models/Company/CompanyContractListener.php
@@ -4,19 +4,23 @@
 
 class CompanyContractListener
 {
+    const PRE_PERSIST = 0;
+
     public $postPersistCalls;
     public $prePersistCalls;
-    
+
     public $postUpdateCalls;
     public $preUpdateCalls;
-    
+
     public $postRemoveCalls;
     public $preRemoveCalls;
 
     public $preFlushCalls;
-    
+
     public $postLoadCalls;
-    
+
+    public $snapshots = [];
+
     /**
      * @PostPersist
      */
@@ -30,6 +34,7 @@ public function postPersistHandler(CompanyContract $contract)
      */
     public function prePersistHandler(CompanyContract $contract)
     {
+        $this->snapshots[self::PRE_PERSIST][] = $this->takeSnapshot($contract);
         $this->prePersistCalls[] = func_get_args();
     }
 
@@ -81,4 +86,20 @@ public function postLoadHandler(CompanyContract $contract)
         $this->postLoadCalls[] = func_get_args();
     }
 
+    public function takeSnapshot(CompanyContract $contract)
+    {
+        $snapshot = [];
+        $reflexion = new \ReflectionClass($contract);
+        foreach ($reflexion->getProperties() as $property) {
+            $property->setAccessible(true);
+            $value = $property->getValue($contract);
+            if (is_object($value) || is_array($value)) {
+                continue;
+            }
+            $snapshot[$property->getName()] = $property->getValue($contract);
+        }
+
+        return $snapshot;
+    }
+
 }
diff --git a/tests/Doctrine/Tests/ORM/Functional/EntityListenersOnMergeTest.php b/tests/Doctrine/Tests/ORM/Functional/EntityListenersOnMergeTest.php
@@ -2,24 +2,39 @@
 
 namespace Doctrine\Tests\ORM\Functional;
 
+use Doctrine\Tests\Models\Company\CompanyContractListener;
+use Doctrine\Tests\Models\Company\CompanyFixContract;
 use Doctrine\Tests\Models\DDC3597\DDC3597Image;
 use Doctrine\Tests\Models\DDC3597\DDC3597Media;
 use Doctrine\Tests\Models\DDC3597\DDC3597Root;
 
 /**
+ * @group DDC-1955
  */
 class EntityListenersOnMergeTest extends \Doctrine\Tests\OrmFunctionalTestCase
 {
+
+    /**
+     * @var \Doctrine\Tests\Models\Company\CompanyContractListener
+     */
+    private $listener;
+
     protected function setUp()
     {
+        $this->useModelSet('company');
         parent::setUp();
+
         $this->_schemaTool->createSchema(
             [
                 $this->_em->getClassMetadata(DDC3597Root::class),
                 $this->_em->getClassMetadata(DDC3597Media::class),
                 $this->_em->getClassMetadata(DDC3597Image::class),
             ]
         );
+
+        $this->listener = $this->_em->getConfiguration()
+            ->getEntityListenerResolver()
+            ->resolve('Doctrine\Tests\Models\Company\CompanyContractListener');
     }
 
     protected function tearDown()
@@ -47,4 +62,35 @@ public function testMergeNewEntityLifecyleEventsModificationsShouldBeKept()
         $this->assertNotNull($imageEntity->getCreatedAt());
         $this->assertNotNull($imageEntity->getUpdatedAt());
     }
+
+    public function testPrePersistListeners()
+    {
+        $fix = new CompanyFixContract();
+        $fix->setFixPrice(2000);
+
+        $this->listener->prePersistCalls = [];
+
+        $fix = $this->_em->merge($fix);
+        $this->_em->flush();
+
+        $this->assertCount(1, $this->listener->prePersistCalls);
+
+        $this->assertSame($fix, $this->listener->prePersistCalls[0][0]);
+
+        $this->assertInstanceOf(
+            'Doctrine\Tests\Models\Company\CompanyFixContract',
+            $this->listener->prePersistCalls[0][0]
+        );
+
+        $this->assertInstanceOf(
+            'Doctrine\ORM\Event\LifecycleEventArgs',
+            $this->listener->prePersistCalls[0][1]
+        );
+
+        $this->assertArrayHasKey('fixPrice', $this->listener->snapshots[CompanyContractListener::PRE_PERSIST][0]);
+        $this->assertEquals(
+            $fix->getFixPrice(),
+            $this->listener->snapshots[CompanyContractListener::PRE_PERSIST][0]['fixPrice']
+        );
+    }
 }
