public function setUp()
 {
     $this->listener = new DoctrineListener();
     $this->eventDispatcher = $this->getMock('Symfony\\Component\\EventDispatcher\\EventDispatcherInterface');
     $this->listener->setEventDispatcher($this->eventDispatcher);
     $this->assertEquals($this->eventDispatcher, $this->listener->getEventDispatcher());
 }
 public function testPostFlush()
 {
     $args = $this->getMockBuilder('Doctrine\\ORM\\Event\\PostFlushEventArgs')->disableOriginalConstructor()->getMock();
     $entityManager = $this->getMockBuilder('Doctrine\\ORM\\EntityManager')->disableOriginalConstructor()->getMock();
     $args->expects($this->once())->method('getEntityManager')->will($this->returnValue($entityManager));
     $this->entityPool->expects($this->once())->method('persistAndFlush')->with($entityManager);
     $this->listener->postFlush($args);
 }