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();
}
