public function testAddInventoryDataToCollection()
{
$productCollection = $this->getMockBuilder('Magento\\Catalog\\Model\\Resource\\Product\\Collection')->disableOriginalConstructor()->getMock();
$this->event->expects($this->once())->method('getCollection')->will($this->returnValue($productCollection));
$this->stockHelper->expects($this->once())->method('addStockStatusToProducts')->with($productCollection)->will($this->returnSelf());
$this->eventObserver->expects($this->atLeastOnce())->method('getEvent')->will($this->returnValue($this->event));
$this->assertEquals($this->observer, $this->observer->addStockStatusToCollection($this->eventObserver));
}
public function testAddStockStatusToCollectionRequireStockItems()
{
$requireStockItems = true;
$productCollection = $this->getMockBuilder('Magento\\Catalog\\Model\\Resource\\Product\\Collection')->disableOriginalConstructor()->setMethods(['hasFlag'])->getMock();
$this->event->expects($this->once())->method('getCollection')->will($this->returnValue($productCollection));
$productCollection->expects($this->once())->method('hasFlag')->with('require_stock_items')->will($this->returnValue($requireStockItems));
$stock = $this->getMockBuilder('Magento\\CatalogInventory\\Model\\Stock')->disableOriginalConstructor()->getMock();
$this->stockFactory->expects($this->once())->method('create')->will($this->returnValue($stock));
$stock->expects($this->once())->method('addItemsToProducts')->with($productCollection)->will($this->returnSelf());
$this->assertEquals($this->model, $this->model->addStockStatusToCollection($this->eventObserver));
}
