stock_source_links_rollback.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

use Magento\InventoryApi\Api\Data\StockSourceLinkInterface;
use Magento\InventoryApi\Api\Data\StockSourceLinkInterfaceFactory;
use Magento\InventoryApi\Api\StockSourceLinksDeleteInterface;
use Magento\TestFramework\Helper\Bootstrap;

$stockSourceLinkFactory = Bootstrap::getObjectManager()->get(StockSourceLinkInterfaceFactory::class);
$stockSourceLinksDelete = Bootstrap::getObjectManager()->get(StockSourceLinksDeleteInterface::class);

$linksData = [
    10 => ['eu-1', 'eu-2', 'eu-3', 'eu-disabled'],
    20 => ['us-1'],
    30 => ['eu-1', 'eu-2', 'eu-3', 'eu-disabled', 'us-1']
];

$links = [];

foreach ($linksData as $stockID => $sourceCodes) {
    foreach ($sourceCodes as $sourceCode) {
        $link = $stockSourceLinkFactory->create();

        $link->setStockId($stockID);
        $link->setSourceCode($sourceCode);

        $links[] = $link;
    }
}

$stockSourceLinksDelete->execute($links);