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']); }
/** * @param int $productId * @return float */ public function getProductStockQty($productId) { return $this->stockItemService->getStockItem($productId)->getQty(); }