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