products_virtual.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

use Magento\Catalog\Api\Data\ProductInterfaceFactory;
use Magento\Catalog\Api\Data\ProductWebsiteLinkInterface;
use Magento\Catalog\Api\Data\ProductWebsiteLinkInterfaceFactory;
use Magento\Catalog\Api\ProductRepositoryInterface;
use Magento\Catalog\Api\ProductWebsiteLinkRepositoryInterface;
use Magento\Catalog\Model\Product\Attribute\Source\Status;
use Magento\Catalog\Model\Product\Type;
use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\Framework\Module\Manager;
use Magento\InventoryApi\Api\Data\SourceItemInterface;
use Magento\InventoryApi\Api\SourceItemRepositoryInterface;
use Magento\InventoryApi\Api\SourceItemsDeleteInterface;
use Magento\InventoryCatalogApi\Api\DefaultSourceProviderInterface;
use Magento\Store\Api\WebsiteRepositoryInterface;
use Magento\TestFramework\Helper\Bootstrap;

$objectManager = Bootstrap::getObjectManager();
$productFactory = $objectManager->get(ProductInterfaceFactory::class);
$productRepository = $objectManager->get(ProductRepositoryInterface::class);
$productRepository->cleanCache();
$productWebsiteLinkRepository = $objectManager->get(ProductWebsiteLinkRepositoryInterface::class);
$productWebsiteLinkFactory = $objectManager->get(ProductWebsiteLinkInterfaceFactory::class);
$websiteRepository = $objectManager->get(WebsiteRepositoryInterface::class);
$websites = $websiteRepository->getList();

$productData = [
    'VIRT-1' => [
        'qty' => 33,
        'is_in_stock' => true,
        'manage_stock' => true
    ],
    'VIRT-2' => [
        'qty' => 30,
        'is_in_stock' => true,
        'manage_stock' => true
    ],
    'VIRT-3' => [
        'qty' => 2,
        'is_in_stock' => true,
        'manage_stock' => true
    ],
    'VIRT-4' => [
        'qty' => 6,
        'is_in_stock' => false,
        'manage_stock' => true
    ]
];

foreach ($productData as $sku => $stockData) {
    $product = $productFactory->create();
    $product->setTypeId(Type::TYPE_VIRTUAL)
        ->setAttributeSetId(4)
        ->setName('Virtual Product ' . $sku)
        ->setSku($sku)
        ->setPrice(10)
        ->setStockData($stockData)
        ->setStatus(Status::STATUS_ENABLED);
    $productRepository->save($product);

    foreach ($websites as $website) {
        if ($website->getCode() === 'admin') {
            continue;
        }

        $websiteLink = $productWebsiteLinkFactory->create();
        $websiteLink->setSku($sku);
        $websiteLink->setWebsiteId($website->getId());
        $productWebsiteLinkRepository->save($websiteLink);
    }
}

$moduleManager = Bootstrap::getObjectManager()->get(Manager::class);
// soft dependency in tests because we don't have possibility replace fixture from different modules
if ($moduleManager->isEnabled('Magento_InventoryCatalog')) {
    $searchCriteriaBuilder = Bootstrap::getObjectManager()->get(SearchCriteriaBuilder::class);
    $defaultSourceProvider = $objectManager->get(DefaultSourceProviderInterface::class);
    $sourceItemRepository = $objectManager->get(SourceItemRepositoryInterface::class);
    $sourceItemsDelete = $objectManager->get(SourceItemsDeleteInterface::class);

    // Unassign created product from default Source
    $searchCriteria = $searchCriteriaBuilder
        ->addFilter(SourceItemInterface::SKU, ['VIRT-1', 'VIRT-2', 'VIRT-3', 'VIRT-4'], 'in')
        ->addFilter(SourceItemInterface::SOURCE_CODE, $defaultSourceProvider->getCode())
        ->create();
    $sourceItems = $sourceItemRepository->getList($searchCriteria)->getItems();
    if (count($sourceItems)) {
        $sourceItemsDelete->execute($sourceItems);
    }
}