protected function setUp() { parent::setUp(); $this->attributeSetCollectionFactoryMock = $this->getMockBuilder(CollectionFactory::class)->setMethods(['create'])->disableOriginalConstructor()->getMock(); $this->attributeSetCollectionMock = $this->getMockBuilder(Collection::class)->disableOriginalConstructor()->getMock(); $this->urlBuilderMock = $this->getMockBuilder(UrlInterface::class)->disableOriginalConstructor()->getMock(); $this->productResourceMock = $this->getMockBuilder(ProductResource::class)->disableOriginalConstructor()->getMock(); $this->attributeSetCollectionFactoryMock->expects($this->any())->method('create')->willReturn($this->attributeSetCollectionMock); $this->productMock->expects($this->any())->method('getResource')->willReturn($this->productResourceMock); $this->attributeSetCollectionMock->expects($this->any())->method('setEntityTypeFilter')->willReturnSelf(); $this->attributeSetCollectionMock->expects($this->any())->method('addFieldToSelect')->willReturnSelf(); }
/** * @return $this */ protected function _initAttributeSets() { $attributeSetOne = $this->getMockBuilder('\\Magento\\Eav\\Model\\Entity\\Attribute\\Set')->disableOriginalConstructor()->getMock(); $attributeSetOne->expects($this->any())->method('getAttributeSetName')->willReturn('attributeSet1'); $attributeSetOne->expects($this->any())->method('getId')->willReturn('1'); $attributeSetTwo = $this->getMockBuilder('\\Magento\\Eav\\Model\\Entity\\Attribute\\Set')->disableOriginalConstructor()->getMock(); $attributeSetTwo->expects($this->any())->method('getAttributeSetName')->willReturn('attributeSet2'); $attributeSetTwo->expects($this->any())->method('getId')->willReturn('2'); $attributeSetCol = [$attributeSetOne, $attributeSetTwo]; $collection = $this->getMockBuilder('\\Magento\\Eav\\Model\\ResourceModel\\Entity\\Attribute\\Set\\Collection')->disableOriginalConstructor()->getMock(); $collection->expects($this->once())->method('setEntityTypeFilter')->with(self::ENTITY_TYPE_ID)->willReturn($attributeSetCol); $this->_setColFactory->expects($this->once())->method('create')->willReturn($collection); return $this; }