Merge pull request magento#24 from magento-nord/develop

[NORD] Bug fixes

Author: Klymenko, Volodymyr(vklymenko)

app/code/Magento/GroupedProductSampleData/Model/Product.php

@@ -6,6 +6,8 @@
 namespace Magento\GroupedProductSampleData\Model;
 
 use Magento\Framework\Setup\SampleData\Context as SampleDataContext;
+use Magento\Catalog\Model\Product\Initialization\Helper\ProductLinks;
+use Magento\Framework\App\ObjectManager;
 
 /**
  * Setup grouped product
@@ -17,6 +19,11 @@ class Product extends \Magento\CatalogSampleData\Model\Product
      */
     protected $productType = \Magento\GroupedProduct\Model\Product\Type\Grouped::TYPE_CODE;
 
+    /**
+     * @var \Magento\Catalog\Model\Product\Initialization\Helper\ProductLinks
+     */
+    private $productLinksHelper;
+
     /**
      * Product constructor.
      * @param SampleDataContext $sampleDataContext
@@ -48,4 +55,34 @@ public function __construct(
             $eavConfig
         );
     }
+
+    /**
+     * @param \Magento\Catalog\Model\Product $product
+     * @param array $data
+     * @return $this
+     */
+    protected function prepareProduct($product, $data)
+    {
+        $this->getProductLinksHelper()->initializeLinks($product, $data['grouped_link_data']);
+        $product->unsetData('grouped_link_data');
+        return $this;
+    }
+
+    /**
+     * Get product links helper
+     *
+     * @deprecated
+     * @return \Magento\Catalog\Model\Product\Initialization\Helper\ProductLinks
+     */
+    private function getProductLinksHelper()
+    {
+
+        if (!($this->productLinksHelper)) {
+            return ObjectManager::getInstance()->get(
+                '\Magento\Catalog\Model\Product\Initialization\Helper\ProductLinks'
+            );
+        } else {
+            return $this->productLinksHelper;
+        }
+    }
 }