AddStockStatusToSelectTest.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\StoreManagerInterface;
use Magento\Store\Model\Website;
use Magento\TestFramework\Helper\Bootstrap;
use PHPUnit\Framework\TestCase;

class AddStockStatusToSelectTest extends TestCase
{
    private $stockStatus;

    private $website;

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

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

    public function testAddStockStatusToSelect(
        string $websiteCode,
        int $expectedIsSalableCount,
        int $expectedNotSalableCount
    ) {
        $actualIsSalableCount = $actualNotSalableCount = 0;

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

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

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

    public function addStockStatusToSelectDataProvider(): array
    {
        return [
            ['eu_website', 2, 3],
            ['us_website', 1, 4],
            ['global_website', 3, 2],
        ];
    }

    public function testAddStockStatusToSelectWithEmptyWebsiteCode()
    {
        $collection = Bootstrap::getObjectManager()->create(Collection::class);
        $this->stockStatus->addStockStatusToSelect($collection->getSelect(), $this->website);
    }

    public function testAddStockStatusToSelectWithNotExistedWebsiteCode()
    {
        $collection = Bootstrap::getObjectManager()->create(Collection::class);
        $this->website->setCode('not_existed_code');
        $this->stockStatus->addStockStatusToSelect($collection->getSelect(), $this->website);
    }
}