/** * Cover isReportEntityType(). * * @dataProvider isReportEntityTypeExceptionDataProvider * @expectedException \Magento\Framework\Exception\LocalizedException */ public function testIsReportEntityTypeException($entity, $getEntitiesResult, $getEntityResult, $expectedResult) { $importMock = $this->getMockBuilder('\\Magento\\ImportExport\\Model\\Import')->disableOriginalConstructor()->setMethods(['getEntity', '_getEntityAdapter', 'getEntityTypeCode', 'isNeedToLogInHistory'])->getMock(); $importMock->expects($this->any())->method('_getEntityAdapter')->willReturnSelf(); $importMock->expects($this->any())->method('getEntityTypeCode')->willReturn('catalog_product'); $this->_importConfig->expects($this->any())->method('getEntities')->willReturn($getEntitiesResult); $this->_entityFactory->expects($this->any())->method('create')->willReturn(''); $this->setPropertyValue($importMock, '_importConfig', $this->_importConfig); $this->setPropertyValue($importMock, '_entityFactory', $this->_entityFactory); $importMock->expects($this->any())->method('getEntity')->willReturn($getEntityResult); $actualResult = $importMock->isReportEntityType($entity); $this->assertEquals($expectedResult, $actualResult); }