AddStockDataToCollectionOnDefaultStockTest.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\TestFramework\Helper\Bootstrap;
use PHPUnit\Framework\TestCase;

class AddStockDataToCollectionOnDefaultStockTest extends TestCase
{
    private $stockStatus;

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

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

    public function testAddStockDataToCollection(int $expectedSize, bool $isFilterInStock)
    {
        $collection = Bootstrap::getObjectManager()->create(Collection::class);
        $collection = $this->stockStatus->addStockDataToCollection($collection, $isFilterInStock);

        self::assertEquals($expectedSize, $collection->getSize());
    }

    public function addStockDataToCollectionDataProvider(): array
    {
        return [
            [3, true],
            [5, false],
        ];
    }
}