useModelSet('cms'); parent::setUp(); } public function testIssue() { $test = $this; $user = new CmsUser(); $uow = $this->_em->getUnitOfWork(); $user->name = 'Marco'; $user->username = 'ocramius'; $this->_em->persist($user); $uow->scheduleExtraUpdate($user, array('name' => 'changed name')); $listener = $this->getMock('stdClass', array(Events::postFlush)); $listener ->expects($this->once()) ->method(Events::postFlush) ->will($this->returnCallback(function () use ($uow, $test) { $test->assertAttributeEmpty('extraUpdates', $uow, 'ExtraUpdates are reset before postFlush'); })); $this->_em->getEventManager()->addEventListener(Events::postFlush, $listener); $this->_em->flush(); } }