public function testAfterLoad() { $productId = 2; $this->stockItemMock->expects($this->once())->method('getIsInStock')->will($this->returnValue(1)); $this->stockItemMock->expects($this->once())->method('getQty')->will($this->returnValue(5)); $store = $this->getMock('Magento\\Store\\Model\\Store', ['getWebsiteId', '__wakeup'], [], '', false); $store->expects($this->once())->method('getWebsiteId')->will($this->returnValue(10)); $object = new \Magento\Framework\DataObject(['id' => $productId, 'store' => $store]); $this->model->afterLoad($object); $data = $object->getData(); $this->assertEquals(1, $data[self::ATTRIBUTE_NAME]['is_in_stock']); $this->assertEquals(5, $data[self::ATTRIBUTE_NAME]['qty']); }
public function testAfterLoad() { $productId = 2; $stockItemDo = $this->getMockBuilder('Magento\\CatalogInventory\\Service\\V1\\Data\\StockItem')->disableOriginalConstructor()->getMock(); $this->stockItemService->expects($this->once())->method('getStockItem')->with($productId)->will($this->returnValue($stockItemDo)); $stockItemDo->expects($this->once())->method('getIsInStock')->will($this->returnValue(1)); $stockItemDo->expects($this->once())->method('getQty')->will($this->returnValue(5)); $object = new \Magento\Framework\Object(['id' => $productId]); $this->model->afterLoad($object); $data = $object->getData(); $this->assertEquals(1, $data[self::ATTRIBUTE_NAME]['is_in_stock']); $this->assertEquals(5, $data[self::ATTRIBUTE_NAME]['qty']); }