public function testGetMediaGalleryAttributeIdIfNotSetYet() { // reset possible existing id $this->setPropertyValue($this->importProduct, '_mediaGalleryAttributeId', null); $expectedId = '100'; $attribute = $this->getMockBuilder('\\Magento\\Eav\\Model\\Entity\\Attribute\\AbstractAttribute')->disableOriginalConstructor()->setMethods(['getId'])->getMockForAbstractClass(); $attribute->expects($this->once())->method('getId')->willReturn($expectedId); $resource = $this->getMockBuilder('\\Magento\\CatalogImportExport\\Model\\Import\\Proxy\\Product\\Resource')->disableOriginalConstructor()->setMethods(['getAttribute'])->getMock(); $resource->expects($this->once())->method('getAttribute')->willReturn($attribute); $this->_resourceFactory->expects($this->once())->method('create')->willReturn($resource); $result = $this->importProduct->getMediaGalleryAttributeId(); $this->assertEquals($expectedId, $result); }