MSI: integration prototype (write scenario)

Author: Valeriy Nayda

app/code/Magento/CatalogInventory/Model/ResourceModel/Stock/Item.php

@@ -125,6 +125,35 @@ protected function _afterSave(AbstractModel $object)
         if ($this->processIndexEvents) {
             $this->stockIndexerProcessor->reindexRow($object->getProductId());
         }
+
+        // start prototype code
+        $sourceItemFactory = \Magento\Framework\App\ObjectManager::getInstance()
+            ->get(\Magento\InventoryApi\Api\Data\SourceItemInterfaceFactory::class);
+        /** @var \Magento\InventoryApi\Api\SourceItemsSaveInterface $sourceItemsSave */
+        $sourceItemsSave = \Magento\Framework\App\ObjectManager::getInstance()
+            ->get(\Magento\InventoryApi\Api\SourceItemsSaveInterface::class);
+        /** @var \Magento\Catalog\Api\ProductRepositoryInterface $productRepository */
+        $productRepository = \Magento\Framework\App\ObjectManager::getInstance()
+            ->get(\Magento\Catalog\Api\ProductRepositoryInterface::class);
+        /** @var \Magento\InventoryCatalog\Api\DefaultSourceProviderInterface $defaultSourceProvider */
+        $defaultSourceProvider = \Magento\Framework\App\ObjectManager::getInstance()
+            ->get(\Magento\InventoryCatalog\Api\DefaultSourceProviderInterface::class);
+
+        $sourceId = $defaultSourceProvider->getId();
+        $product = $productRepository->getById($object->getProductId());
+        $sku = $product->getSku();
+        $qty = (float)$object->getQty();
+        $status = (bool)$object->getIsInStock();
+
+        /** @var \Magento\InventoryApi\Api\Data\SourceItemInterface $sourceItem */
+        $sourceItem = $sourceItemFactory->create();
+        $sourceItem->setSourceId($sourceId);
+        $sourceItem->setSku($sku);
+        $sourceItem->setQuantity($qty);
+        $sourceItem->setStatus($status);
+        $sourceItemsSave->execute([$sourceItem]);
+        // end prototype code
+
         return $this;
     }
 

app/code/Magento/Inventory/Observer/ProcessSourceItemsObserver.php

@@ -39,6 +39,11 @@ public function __construct(SourceItemsProcessor $sourceItemsProcessor)
      */
     public function execute(EventObserver $observer)
     {
+        // start prototype code
+        // Work like as in single stock mode
+        return;
+        // end prototype code
+
         /** @var ProductInterface $product */
         $product = $observer->getEvent()->getProduct();
         /** @var Save $controller */

dev/tests/integration/testsuite/Magento/CatalogInventory/Model/StockItemSave/StockItemDataChecker.php

@@ -74,6 +74,35 @@ public function checkStockItemData($sku, array $expectedData)
         $productLoadedByModel = $this->productFactory->create();
         $productLoadedByModel->load($product->getId());
         $this->doCheckStockItemData($product, $expectedData);
+
+        // start prototype code
+        /** @var \Magento\InventoryApi\Api\SourceItemRepositoryInterface $sourceItemRepository */
+        $sourceItemRepository = \Magento\Framework\App\ObjectManager::getInstance()
+            ->get(\Magento\InventoryApi\Api\SourceItemRepositoryInterface::class);
+        /** @var \Magento\Framework\Api\SearchCriteriaBuilder $searchCriteriaBuilder */
+        $searchCriteriaBuilder = \Magento\Framework\App\ObjectManager::getInstance()
+            ->get(\Magento\Framework\Api\SearchCriteriaBuilder::class);
+        /** @var \Magento\InventoryCatalog\Api\DefaultSourceProviderInterface $defaultSourceProvider */
+        $defaultSourceProvider = \Magento\Framework\App\ObjectManager::getInstance()
+            ->get(\Magento\InventoryCatalog\Api\DefaultSourceProviderInterface::class);
+
+        $searchCriteria = $searchCriteriaBuilder
+            ->addFilter(\Magento\InventoryApi\Api\Data\SourceItemInterface::SOURCE_ID, $defaultSourceProvider->getId())
+            ->addFilter(\Magento\InventoryApi\Api\Data\SourceItemInterface::SKU, $sku)
+            ->create();
+        $sourceItems = $sourceItemRepository->getList($searchCriteria)->getItems();
+        \PHPUnit\Framework\Assert::assertCount(1, $sourceItems);
+
+        $sourceItem = reset($sourceItems);
+        \PHPUnit\Framework\Assert::assertEquals(
+            $expectedData[StockItemInterface::QTY],
+            $sourceItem->getQuantity()
+        );
+        \PHPUnit\Framework\Assert::assertEquals(
+            $expectedData[StockItemInterface::IS_IN_STOCK],
+            (int)$sourceItem->getStatus()
+        );
+        // end prototype code
     }
 
     /**