137 lines
3.8 KiB
PHP
137 lines
3.8 KiB
PHP
<?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();
|
|
}
|
|
|
|
/**
|
|
* @group DDC-2173
|
|
*/
|
|
public function testPreAndOnFlushCalledAlways()
|
|
{
|
|
$listener = new OnFlushCalledListener();
|
|
$this->_em->getEventManager()->addEventListener(Events::onFlush, $listener);
|
|
$this->_em->getEventManager()->addEventListener(Events::preFlush, $listener);
|
|
$this->_em->getEventManager()->addEventListener(Events::postFlush, $listener);
|
|
|
|
$this->_em->flush();
|
|
|
|
$this->assertEquals(1, $listener->preFlush);
|
|
$this->assertEquals(1, $listener->onFlush);
|
|
|
|
$this->_em->flush();
|
|
|
|
$this->assertEquals(2, $listener->preFlush);
|
|
$this->assertEquals(2, $listener->onFlush);
|
|
}
|
|
}
|
|
|
|
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);
|
|
}*/
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
class OnFlushCalledListener
|
|
{
|
|
public $preFlush = 0;
|
|
public $onFlush = 0;
|
|
public $postFlush = 0;
|
|
|
|
public function preFlush($args)
|
|
{
|
|
$this->preFlush++;
|
|
}
|
|
|
|
public function onFlush($args)
|
|
{
|
|
$this->onFlush++;
|
|
}
|
|
|
|
public function postFlush($args)
|
|
{
|
|
$this->postFlush++;
|
|
}
|
|
}
|
|
|