Sonata\OrderBundle\Entity\OrderManager::save PHP Method

    public function save($order, $andFlush = true)
    {
        $this->getEntityManager()->persist($order->getCustomer());
        parent::save($order, $andFlush);
    }

 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);
 }