MAGETWO-90999: Bundle Product not appears in catalog after import

- Added bundle product entity relations to import

Author: viktym

app/code/Magento/BundleImportExport/Model/Import/Product/Type/Bundle.php

@@ -8,11 +8,12 @@
  */
 namespace Magento\BundleImportExport\Model\Import\Product\Type;
 
-use \Magento\Framework\App\ObjectManager;
-use \Magento\Bundle\Model\Product\Price as BundlePrice;
-use \Magento\Catalog\Model\Product\Type\AbstractType;
-use \Magento\CatalogImportExport\Model\Import\Product;
-use \Magento\Store\Model\StoreManagerInterface;
+use Magento\Framework\App\ObjectManager;
+use Magento\Bundle\Model\Product\Price as BundlePrice;
+use Magento\Catalog\Model\Product\Type\AbstractType;
+use Magento\CatalogImportExport\Model\Import\Product;
+use Magento\Store\Model\StoreManagerInterface;
+use Magento\Bundle\Model\ResourceModel\Bundle as BundleResourceModel;
 
 /**
  * Class Bundle
@@ -148,6 +149,11 @@ class Bundle extends \Magento\CatalogImportExport\Model\Import\Product\Type\Abst
      */
     private $storeCodeToId = [];
 
+    /**
+     * @var BundleResourceModel
+     */
+    private $bundleResourceModel;
+
     /**
      * @param \Magento\Eav\Model\ResourceModel\Entity\Attribute\Set\CollectionFactory $attrSetColFac
      * @param \Magento\Catalog\Model\ResourceModel\Product\Attribute\CollectionFactory $prodAttrColFac
@@ -156,8 +162,8 @@ class Bundle extends \Magento\CatalogImportExport\Model\Import\Product\Type\Abst
      * @param \Magento\Framework\EntityManager\MetadataPool|null $metadataPool
      * @param Bundle\RelationsDataSaver|null $relationsDataSaver
      * @param StoreManagerInterface $storeManager
+     * @param BundleResourceModel $bundleResourceModel
      * @throws \Magento\Framework\Exception\LocalizedException
-     * @throws \RuntimeException
      */
     public function __construct(
         \Magento\Eav\Model\ResourceModel\Entity\Attribute\Set\CollectionFactory $attrSetColFac,
@@ -166,13 +172,16 @@ public function __construct(
         array $params,
         \Magento\Framework\EntityManager\MetadataPool $metadataPool = null,
         Bundle\RelationsDataSaver $relationsDataSaver = null,
-        StoreManagerInterface $storeManager = null
+        StoreManagerInterface $storeManager = null,
+        BundleResourceModel $bundleResourceModel = null
     ) {
         parent::__construct($attrSetColFac, $prodAttrColFac, $resource, $params, $metadataPool);
         $this->relationsDataSaver = $relationsDataSaver
             ?: ObjectManager::getInstance()->get(Bundle\RelationsDataSaver::class);
         $this->storeManager = $storeManager
             ?: ObjectManager::getInstance()->get(StoreManagerInterface::class);
+        $this->bundleResourceModel = $bundleResourceModel
+            ?: ObjectManager::getInstance()->get(BundleResourceModel::class);
     }
 
     /**
@@ -389,13 +398,15 @@ public function saveData()
                     if ($this->_type != $productData['type_id']) {
                         continue;
                     }
-                    $this->parseSelections($rowData, $productData[$this->getProductEntityLinkField()]);
+                    $productId = $productData[$this->getProductEntityLinkField()];
+                    $this->parseSelections($rowData, $productId);
                 }
                 if (!empty($this->_cachedOptions)) {
                     $this->retrieveProducsByCachedSkus();
                     $this->populateExistingOptions();
                     $this->insertOptions();
                     $this->insertSelections();
+                    $this->insertProductRelations();
                     $this->clear();
                 }
             }
@@ -643,6 +654,29 @@ protected function insertSelections()
         return $this;
     }
 
+    /**
+     * Insert product relations.
+     *
+     * @return void
+     */
+    private function insertProductRelations()
+    {
+        foreach ($this->_cachedOptions as $productId => $options) {
+            $childIds = [];
+            foreach ($options as $option) {
+                foreach ($option['selections'] as $selection) {
+                    if (isset($this->_cachedSkuToProducts[$selection['sku']])) {
+                        $childIds[] = $this->_cachedSkuToProducts[$selection['sku']];
+                    }
+                }
+            }
+            if (!empty($childIds)) {
+                $childIds = array_unique($childIds);
+                $this->bundleResourceModel->saveProductRelations($productId, $childIds);
+            }
+        }
+    }
+
     /**
      * Initialize attributes parameters for all attributes' sets.
      *

dev/tests/integration/testsuite/Magento/BundleImportExport/Model/Import/Product/Type/BundleTest.php

@@ -5,12 +5,12 @@
  */
 namespace Magento\BundleImportExport\Model\Import\Product\Type;
 
-use Magento\Framework\App\Bootstrap;
 use Magento\Framework\App\Filesystem\DirectoryList;
-use Magento\ImportExport\Model\Import;
+use Magento\Framework\Indexer\IndexerRegistry;
 
 /**
  * @magentoAppArea adminhtml
+ * @SuppressWarnings(PHPMD.CouplingBetweenObjects)
  */
 class BundleTest extends \PHPUnit\Framework\TestCase
 {
@@ -49,7 +49,7 @@ protected function setUp()
 
     /**
      * @magentoAppArea adminhtml
-     * @magentoDbIsolation enabled
+     * @magentoDbIsolation disabled
      * @magentoAppIsolation enabled
      */
     public function testBundleImport()
@@ -80,12 +80,17 @@ public function testBundleImport()
         $this->assertTrue($errors->getErrorsCount() == 0);
         $this->model->importData();
 
+        $indexerRegistry = $this->objectManager->get(IndexerRegistry::class);
+        $indexerRegistry->get('cataloginventory_stock')->reindexAll();
+
+        /** @var \Magento\Catalog\Model\ResourceModel\Product $resource */
         $resource = $this->objectManager->get(\Magento\Catalog\Model\ResourceModel\Product::class);
         $productId = $resource->getIdBySku(self::TEST_PRODUCT_NAME);
         $this->assertTrue(is_numeric($productId));
+
         /** @var \Magento\Catalog\Model\Product $product */
-        $product = $this->objectManager->create(\Magento\Catalog\Model\Product::class);
-        $product->load($productId);
+        $productRepository = $this->objectManager->get(\Magento\Catalog\Api\ProductRepositoryInterface::class);
+        $product = $productRepository->get(self::TEST_PRODUCT_NAME, false, null, true);
 
         $this->assertFalse($product->isObjectNew());
         $this->assertEquals(self::TEST_PRODUCT_NAME, $product->getName());
@@ -106,6 +111,17 @@ public function testBundleImport()
                 $this->assertEquals($optionSku, $productLink->getData('sku'));
             }
         }
+
+        /** @var \Magento\CatalogInventory\Model\Stock\Item $stockItem */
+        $stockItem = $product->getExtensionAttributes()->getStockItem();
+        $this->assertTrue(
+            $stockItem->getIsInStock(),
+            'Imported bundle product should be in stock'
+        );
+        $this->assertTrue(
+            $product->isSalable(),
+            'Imported bundle product should be available for sale'
+        );
     }
 
     /**