AddStockStatusToSelectOnDefaultStockTest.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryCatalog\Test\Integration\CatalogInventory\Model\ResourceModel\Stock\Status;

use Magento\Catalog\Model\ResourceModel\Product\Collection;
use Magento\CatalogInventory\Model\ResourceModel\Stock\Status as StockStatus;
use Magento\Store\Model\Website;
use Magento\TestFramework\Helper\Bootstrap;
use PHPUnit\Framework\TestCase;

class AddStockStatusToSelectOnDefaultStockTest extends TestCase
{
    private $stockStatus;

    private $website;

    protected function setUp()
    {
        parent::setUp();

        $this->stockStatus = Bootstrap::getObjectManager()->get(StockStatus::class);
        $this->website = Bootstrap::getObjectManager()->get(Website::class);
    }

    public function testAddStockStatusToSelect()
    {
        $actualIsSalableCount = $actualNotSalableCount = 0;
        $expectedIsSalableCount = 3;
        $expectedNotSalableCount = 2;

        $collection = Bootstrap::getObjectManager()->create(Collection::class);
        $this->website->setCode('base');
        $this->stockStatus->addStockStatusToSelect($collection->getSelect(), $this->website);

        foreach ($collection as $item) {
            $item->getIsSalable() === true ? $actualIsSalableCount++ : $actualNotSalableCount++;
        }

        self::assertEquals($expectedIsSalableCount, $actualIsSalableCount);
        self::assertEquals($expectedNotSalableCount, $actualNotSalableCount);
        self::assertEquals($expectedNotSalableCount + $expectedIsSalableCount, $collection->getSize());
    }
}