public function testPostUpdate_WithStdClass() { $container = $this->getMock('\\Symfony\\Component\\DependencyInjection\\Container'); $pageListener = new PageListener(new \Symfony\Component\HttpFoundation\Session\Session(), $container); // entity $entity = $this->getMock('\\StdClass'); $entity->expects($this->never())->method('getContentRoute'); // em $em = $this->getMockBuilder('\\Doctrine\\ORM\\EntityManager')->disableOriginalConstructor()->getMock(); $uow = $this->getMockBuilder('\\Doctrine\\ORM\\UnitOfWork')->disableOriginalConstructor()->getMock(); $em->expects($this->once())->method('getUnitOfWork')->will($this->returnValue($uow)); $uow->expects($this->once())->method('getScheduledEntityUpdates')->will($this->returnValue(array($entity))); $em->expects($this->never())->method('persist'); $em->expects($this->never())->method('flush'); $args = $this->getMockBuilder('\\Doctrine\\ORM\\Event\\OnFlushEventArgs')->disableOriginalConstructor()->getMock(); // args methods: // getObjectManager $args->expects($this->once())->method('getEntityManager')->will($this->returnValue($em)); $pageListener->onFlush($args); }