protected function getProductOptionExtensionMock() { $productOptionExtension = $this->getMockBuilder('Magento\\Catalog\\Api\\Data\\ProductOptionExtensionInterface')->setMethods(['setData'])->getMockForAbstractClass(); $productOptionExtension->expects($this->any())->method('setData')->with(key($this->productOptionData), current($this->productOptionData))->willReturnSelf(); $this->extensionFactory->expects($this->any())->method('create')->willReturn($productOptionExtension); }