public function testSave() { $em = $this->getMockBuilder('Doctrine\\ORM\\EntityManager')->disableOriginalConstructor()->getMock(); $em->expects($this->exactly(2))->method('persist'); $em->expects($this->once())->method('flush'); $registry = $this->getMock('Doctrine\\Common\\Persistence\\ManagerRegistry'); $registry->expects($this->any())->method('getManagerForClass')->will($this->returnValue($em)); $orderManager = new OrderManager('Sonata\\Test\\OrderBundle\\Entity\\Order', $registry); $order = $this->getMock('Sonata\\Test\\OrderBundle\\Entity\\Order'); $order->expects($this->once())->method('getCustomer'); $orderManager->save($order); }