Rajout de doctrine/orm

[zf2.biz/application_blanche.git] / vendor / doctrine / orm / tests / Doctrine / Tests / ORM / Functional / FlushEventTest.php

diff --git a/vendor/doctrine/orm/tests/Doctrine/Tests/ORM/Functional/FlushEventTest.php b/vendor/doctrine/orm/tests/Doctrine/Tests/ORM/Functional/FlushEventTest.php
new file mode 100644 (file)
index 0000000..8167e85
--- /dev/null
+++ b/vendor/doctrine/orm/tests/Doctrine/Tests/ORM/Functional/FlushEventTest.php
@@ -0,0 +1,94 @@
+<?php
+
+namespace Doctrine\Tests\ORM\Functional;
+
+use Doctrine\Tests\Models\CMS\CmsUser;
+use Doctrine\Tests\Models\CMS\CmsPhonenumber;
+use Doctrine\ORM\Event\OnFlushEventArgs;
+use Doctrine\ORM\Events;
+
+require_once __DIR__ . '/../../TestInit.php';
+
+/**
+ * FlushEventTest
+ *
+ * @author robo
+ */
+class FlushEventTest extends \Doctrine\Tests\OrmFunctionalTestCase
+{
+    protected function setUp() {
+        $this->useModelSet('cms');
+        parent::setUp();
+    }
+
+    public function testPersistNewEntitiesOnPreFlush()
+    {
+        //$this->_em->getConnection()->getConfiguration()->setSQLLogger(new \Doctrine\DBAL\Logging\EchoSQLLogger);
+        $this->_em->getEventManager()->addEventListener(Events::onFlush, new OnFlushListener);
+
+        $user = new CmsUser;
+        $user->username = 'romanb';
+        $user->name = 'Roman';
+        $user->status = 'Dev';
+
+        $this->_em->persist($user);
+
+        $this->assertEquals(0, $user->phonenumbers->count());
+
+        $this->_em->flush();
+
+        $this->assertEquals(1, $user->phonenumbers->count());
+        $this->assertTrue($this->_em->contains($user->phonenumbers->get(0)));
+        $this->assertTrue($user->phonenumbers->get(0)->getUser() === $user);
+
+        $this->assertFalse($user->phonenumbers->isDirty());
+
+        // Can be used together with SQL Logging to check that a subsequent flush has
+        // nothing to do. This proofs the correctness of the changes that happened in onFlush.
+        //echo "SECOND FLUSH";
+        //$this->_em->flush();
+    }
+}
+
+class OnFlushListener
+{
+    public function onFlush(OnFlushEventArgs $args)
+    {
+        //echo "---preFlush".PHP_EOL;
+
+        $em = $args->getEntityManager();
+        $uow = $em->getUnitOfWork();
+
+        foreach ($uow->getScheduledEntityInsertions() as $entity) {
+
+            if ($entity instanceof CmsUser) {
+                // Adds a phonenumber to every newly persisted CmsUser ...
+
+                $phone = new CmsPhonenumber;
+                $phone->phonenumber = 12345;
+                // Update object model
+                $entity->addPhonenumber($phone);
+                // Invoke regular persist call
+                $em->persist($phone);
+                // Explicitly calculate the changeset since onFlush is raised
+                // after changeset calculation!
+                $uow->computeChangeSet($em->getClassMetadata(get_class($phone)), $phone);
+
+                // Take a snapshot because the UoW wont do this for us, because
+                // the UoW did not visit this collection.
+                // Alternatively we could provide an ->addVisitedCollection() method
+                // on the UoW.
+                $entity->getPhonenumbers()->takeSnapshot();
+            }
+
+            /*foreach ($uow->getEntityChangeSet($entity) as $field => $change) {
+                list ($old, $new) = $change;
+
+                var_dump($old);
+            }*/
+
+        }
+    }
+}
+
+