public function testGetDeleteMessageDatagrid()
 {
     $message = 'Payment term delete message for datagrid';
     $paymentTermId = 1;
     $this->deleteMessageTextGenerator->expects($this->once())->method('getDeleteMessageTextForDataGrid')->with($paymentTermId)->willReturn($message);
     $result = $this->deleteMessageTextExtension->getDeleteMessageDatagrid($paymentTermId);
     $this->assertEquals($message, $result);
 }
 public function testGetDeleteMessageTextForDataGrid()
 {
     $paymentTermId = 1;
     $paymentTermRepository = $this->getMockBuilder('OroB2B\\Bundle\\PaymentBundle\\Entity\\Repository\\PaymentTermRepository')->disableOriginalConstructor()->getMock();
     $paymentTermRepository->expects($this->once())->method('find')->with($paymentTermId)->willReturnCallback(function ($id) {
         return new PaymentTermStub($id);
     });
     $om = $this->getMock('\\Doctrine\\Common\\Persistence\\ObjectManager');
     $om->expects($this->once())->method('getRepository')->with($this->equalTo('OroB2BPaymentBundle:PaymentTerm'))->willReturn($paymentTermRepository);
     $this->managerRegistry->expects($this->once())->method('getManagerForClass')->with($this->equalTo('OroB2BPaymentBundle:PaymentTerm'))->willReturn($om);
     $message = $this->extension->getDeleteMessageTextForDataGrid($paymentTermId);
     $this->assertDeleteMessage($message, $paymentTermId, 0, 0);
 }
 /**
  * @param $paymentTermId
  * @return string
  */
 public function getDeleteMessageDatagrid($paymentTermId)
 {
     return $this->deleteMessageGenerator->getDeleteMessageTextForDataGrid($paymentTermId);
 }