Merge remote-tracking branch 'origin/MAGETWO-50415' into develop

Author: Yaroslav Onischenko

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

@@ -5,7 +5,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
@@ -18,34 +19,37 @@ class Product extends \Magento\CatalogSampleData\Model\Product
     protected $productType = \Magento\GroupedProduct\Model\Product\Type\Grouped::TYPE_CODE;
 
     /**
-     * Product constructor.
-     * @param SampleDataContext $sampleDataContext
-     * @param \Magento\Catalog\Model\ProductFactory $productFactory
-     * @param \Magento\Catalog\Model\ConfigFactory $catalogConfig
-     * @param \Magento\GroupedProductSampleData\Model\Product\Converter $converter
-     * @param \Magento\Framework\Setup\SampleData\FixtureManager $fixtureManager
-     * @param \Magento\CatalogSampleData\Model\Product\Gallery $gallery
-     * @param \Magento\Store\Model\StoreManagerInterface $storeManager
-     * @param \Magento\Eav\Model\Config $eavConfig
+     * @var \Magento\Catalog\Model\Product\Initialization\Helper\ProductLinks
      */
-    public function __construct(
-        SampleDataContext $sampleDataContext,
-        \Magento\Catalog\Model\ProductFactory $productFactory,
-        \Magento\Catalog\Model\ConfigFactory $catalogConfig,
-        \Magento\GroupedProductSampleData\Model\Product\Converter $converter,
-        \Magento\Framework\Setup\SampleData\FixtureManager $fixtureManager,
-        \Magento\CatalogSampleData\Model\Product\Gallery $gallery,
-        \Magento\Store\Model\StoreManagerInterface $storeManager,
-        \Magento\Eav\Model\Config $eavConfig
-    ) {
-        parent::__construct(
-            $sampleDataContext,
-            $productFactory,
-            $catalogConfig,
-            $converter,
-            $gallery,
-            $storeManager,
-            $eavConfig
-        );
+    private $productLinksHelper;
+
+    /**
+     * @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;
+        }
     }
 }