MAGETWO-50450: Bundle product has no products in sample data

Author: Leonid Poluyanov

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

@@ -6,6 +6,9 @@
 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;
 
 /**
  * Setup bundle product
@@ -17,6 +20,21 @@ class Product extends \Magento\CatalogSampleData\Model\Product
      */
     protected $productType = \Magento\Catalog\Model\Product\Type::TYPE_BUNDLE;
 
+    /**
+     * @var OptionFactory
+     */
+    protected $optionFactory;
+
+    /**
+     * @var LinkFactory
+     */
+    protected $linkFactory;
+
+    /**
+     * @var ProductRepository
+     */
+    protected $productRepository;
+
     /**
      * Product constructor.
      * @param SampleDataContext $sampleDataContext
@@ -34,7 +52,10 @@ public function __construct(
         \Magento\BundleSampleData\Model\Product\Converter $converter,
         \Magento\CatalogSampleData\Model\Product\Gallery $gallery,
         \Magento\Store\Model\StoreManagerInterface $storeManager,
-        \Magento\Eav\Model\Config $eavConfig
+        \Magento\Eav\Model\Config $eavConfig,
+        OptionFactory $optionFactory,
+        LinkFactory $linkFactory,
+        ProductRepository $productRepository
     ) {
         $this->eavConfig = $eavConfig;
         parent::__construct(
@@ -46,6 +67,9 @@ public function __construct(
             $storeManager,
             $eavConfig
         );
+        $this->optionFactory = $optionFactory;
+        $this->linkFactory = $linkFactory;
+        $this->productRepository = $productRepository;
     }
 
     /**
@@ -57,6 +81,34 @@ protected function prepareProduct($product, $data)
             ->setCanSaveConfigurableAttributes(true)
             ->setCanSaveBundleSelections(true)
             ->setPriceType(0);
+        $bundleOptionsData = $product->getBundleOptionsData();
+        $options = [];
+        foreach ($bundleOptionsData as $key => $optionData) {
+
+            $option = $this->optionFactory->create(['data' => $optionData]);
+            $option->setSku($product->getSku());
+            $option->setOptionId(null);
+
+            $links = [];
+            $bundleLinks = $product->getBundleSelectionsData();
+            foreach ($bundleLinks[$key] as $linkData) {
+                $linkProduct = $this->productRepository->getById($linkData['product_id']);
+                $link = $this->linkFactory->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;
     }

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)