/** * @expectedException \Magento\Framework\Exception\NoSuchEntityException * @expectedExceptionMessage Product with SKU linkedProduct is not linked to product with SKU product */ public function testDeleteWithNoSuchEntityException() { $entityMock = $this->getMock('\\Magento\\Catalog\\Model\\ProductLink\\Link', [], [], '', false); $productMock = $this->getMock('\\Magento\\Catalog\\Model\\Product', [], [], '', false); $linkedProductMock = $this->getMock('\\Magento\\Catalog\\Model\\Product', [], [], '', false); $this->productRepositoryMock->expects($this->exactly(2))->method('get')->will($this->returnValueMap([['product', false, null, false, $productMock], ['linkedProduct', false, null, false, $linkedProductMock]])); $entityMock->expects($this->exactly(2))->method('getLinkedProductSku')->willReturn('linkedProduct'); $entityMock->expects($this->exactly(2))->method('getSku')->willReturn('product'); $entityMock->expects($this->once())->method('getLinkType')->willReturn('linkType'); $this->model->delete($entityMock); }