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

Author: Yaroslav Onischenko

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

@@ -6,6 +6,10 @@
 namespace Magento\BundleSampleData\Model;
 
 use Magento\Framework\Setup\SampleData\Context as SampleDataContext;
+use Magento\Bundle\Api\Data\OptionInterfaceFactory as OptionFactory;
+use Magento\Bundle\Api\Data\LinkInterfaceFactory as LinkFactory;
+use Magento\Catalog\Api\ProductRepositoryInterface as ProductRepository;
+use \Magento\Framework\App\ObjectManager;
 
 /**
  * Setup bundle product
@@ -17,6 +21,11 @@ class Product extends \Magento\CatalogSampleData\Model\Product
      */
     protected $productType = \Magento\Catalog\Model\Product\Type::TYPE_BUNDLE;
 
+    /**
+     * @var OptionFactory
+     */
+    private $optionFactory;
+
     /**
      * Product constructor.
      * @param SampleDataContext $sampleDataContext
@@ -48,6 +57,16 @@ public function __construct(
         );
     }
 
+    /**
+     * @var LinkFactory
+     */
+    private $linkFactory;
+
+    /**
+     * @var ProductRepository
+     */
+    private $productRepository;
+
     /**
      * @inheritdoc
      */
@@ -57,7 +76,88 @@ protected function prepareProduct($product, $data)
             ->setCanSaveConfigurableAttributes(true)
             ->setCanSaveBundleSelections(true)
             ->setPriceType(0);
+        $bundleOptionsData = $product->getBundleOptionsData();
+        $options = [];
+        foreach ($bundleOptionsData as $key => $optionData) {
+            $option = $this->getOptionFactory()->create(['data' => $optionData]);
+            $option->setSku($product->getSku());
+            $option->setOptionId(null);
+
+            $links = [];
+            $bundleLinks = $product->getBundleSelectionsData();
+            foreach ($bundleLinks[$key] as $linkData) {
+                $linkProduct = $this->getProductRepository()->getById($linkData['product_id']);
+                $link = $this->getLinkFactory()->create(['data' => $linkData]);
+                $link->setSku($linkProduct->getSku());
+                $link->setQty($linkData['selection_qty']);
+
+                if (array_key_exists('selection_can_change_qty', $linkData)) {
+                    $link->setCanChangeQuantity($linkData['selection_can_change_qty']);
+                }
+                $links[] = $link;
+            }
+            $option->setProductLinks($links);
+            $options[] = $option;
+        }
+
+        $extension = $product->getExtensionAttributes();
+        $extension->setBundleProductOptions($options);
+        $product->setExtensionAttributes($extension);
 
         return $this;
     }
+
+    /**
+     * Get option interface factory
+     *
+     * @deprecated
+     * @return \Magento\Bundle\Api\Data\OptionInterfaceFactory
+     */
+    private function getOptionFactory()
+    {
+
+        if (!($this->optionFactory)) {
+            return ObjectManager::getInstance()->get(
+                '\Magento\Bundle\Api\Data\OptionInterfaceFactory'
+            );
+        } else {
+            return $this->optionFactory;
+        }
+    }
+
+    /**
+     * Get bundle link interface factory
+     *
+     * @deprecated
+     * @return \Magento\Bundle\Api\Data\LinkInterfaceFactory
+     */
+    private function getLinkFactory()
+    {
+
+        if (!($this->linkFactory)) {
+            return ObjectManager::getInstance()->get(
+                '\Magento\Bundle\Api\Data\LinkInterfaceFactory'
+            );
+        } else {
+            return $this->linkFactory;
+        }
+    }
+
+    /**
+     * Get product repository
+     *
+     * @deprecated
+     * @return \Magento\Catalog\Api\ProductRepositoryInterface
+     */
+    private function getProductRepository()
+    {
+
+        if (!($this->productRepository)) {
+            return ObjectManager::getInstance()->get(
+                '\Magento\Catalog\Api\ProductRepositoryInterface'
+            );
+        } else {
+            return $this->productRepository;
+        }
+    }
 }

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

@@ -166,7 +166,7 @@ protected function installGallery($product)
     }
 
     /**
-     * @param \Magento\Framework\Model\AbstractModel $product
+     * @param \Magento\Catalog\Model\Product $product
      * @param array $data
      * @return $this
      * @SuppressWarnings(PHPMD.UnusedFormalParameter)