StockItemImporterTest.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryImportExport\Test\Integration\Model;

use Magento\CatalogImportExport\Model\StockItemImporterInterface;
use Magento\Framework\Api\SearchCriteria;
use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\Framework\Api\SearchCriteriaBuilderFactory;
use Magento\InventoryApi\Api\Data\SourceItemInterface;
use Magento\InventoryApi\Api\SourceItemRepositoryInterface;
use Magento\InventoryCatalogApi\Api\DefaultSourceProviderInterface;
use Magento\InventoryCatalogApi\Model\GetProductIdsBySkusInterface;
use Magento\TestFramework\Helper\Bootstrap;
use PHPUnit\Framework\TestCase;

class StockItemImporterTest extends TestCase
{
    private $defaultSourceProvider;

    private $importer;

    private $searchCriteriaBuilderFactory;

    private $sourceItemRepository;

    private $productIdBySku;

    public function setUp()
    {
        $this->defaultSourceProvider = Bootstrap::getObjectManager()->get(
            DefaultSourceProviderInterface::class
        );
        $this->importer = Bootstrap::getObjectManager()->get(
            StockItemImporterInterface::class
        );
        $this->searchCriteriaBuilderFactory = Bootstrap::getObjectManager()->get(
            SearchCriteriaBuilderFactory::class
        );
        $this->sourceItemRepository = Bootstrap::getObjectManager()->get(
            SourceItemRepositoryInterface::class
        );
        $this->productIdBySku = Bootstrap::getObjectManager()->get(
            GetProductIdsBySkusInterface::class
        );
    }

    public function testSourceItemImportWithDefaultSource()
    {
        $productId = $this->productIdBySku->execute(['SKU-1'])['SKU-1'];
        $stockData = [
            'SKU-1' => [
                'qty' => 1,
                'is_in_stock' => SourceItemInterface::STATUS_IN_STOCK,
                'product_id' => $productId,
                'website_id' => 0,
                'stock_id' => 1,
            ]
        ];

        $this->importer->import($stockData);

        $compareData = $this->buildDataArray($this->getSourceItemList()->getItems());
        $expectedData = [
            SourceItemInterface::SKU => 'SKU-1',
            SourceItemInterface::QUANTITY => 1.0,
            SourceItemInterface::SOURCE_CODE => $this->defaultSourceProvider->getCode(),
            SourceItemInterface::STATUS => SourceItemInterface::STATUS_IN_STOCK,
        ];

        $this->assertArrayHasKey('SKU-1', $compareData);
        $this->assertSame($expectedData, $compareData['SKU-1']);
    }

    private function getSourceItemList()
    {
        $searchCriteriaBuilder = $this->searchCriteriaBuilderFactory->create();

        $searchCriteriaBuilder->addFilter(
            SourceItemInterface::SKU,
            'SKU-1'
        );

        $searchCriteriaBuilder->addFilter(
            SourceItemInterface::SOURCE_CODE,
            $this->defaultSourceProvider->getCode()
        );

        $searchCriteria = $searchCriteriaBuilder->create();

        return $this->sourceItemRepository->getList($searchCriteria);
    }

    private function buildDataArray(array $sourceItems)
    {
        $comparableArray = [];
        foreach ($sourceItems as $sourceItem) {
            $comparableArray[$sourceItem->getSku()] = [
                SourceItemInterface::SKU => $sourceItem->getSku(),
                SourceItemInterface::QUANTITY => $sourceItem->getQuantity(),
                SourceItemInterface::SOURCE_CODE => $sourceItem->getSourceCode(),
                SourceItemInterface::STATUS => $sourceItem->getStatus(),
            ];
        }

        return $comparableArray;
    }
}