stock_website_sales_channels_rollback.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

use Magento\InventoryApi\Api\StockRepositoryInterface;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterfaceFactory;
use Magento\TestFramework\Helper\Bootstrap;

$stockRepository = Bootstrap::getObjectManager()->get(StockRepositoryInterface::class);
$salesChannelFactory = Bootstrap::getObjectManager()->get(SalesChannelInterfaceFactory::class);

$websiteCodes = ['eu_website', 'us_website', 'global_website'];
$defaultStock = $stockRepository->get(1);
$extensionAttributes = $defaultStock->getExtensionAttributes();
$salesChannels = $extensionAttributes->getSalesChannels();

// reassign on Default Stock because website can't exists without link to any Stock
foreach ($websiteCodes as $websiteCode) {
    $salesChannel = $salesChannelFactory->create();
    $salesChannel->setCode($websiteCode);
    $salesChannel->setType(SalesChannelInterface::TYPE_WEBSITE);
    $salesChannels[] = $salesChannel;
}
$extensionAttributes->setSalesChannels($salesChannels);
$stockRepository->save($defaultStock);