6
6
namespace Magento \GroupedProductSampleData \Model ;
7
7
8
8
use Magento \Framework \Setup \SampleData \Context as SampleDataContext ;
9
+ use Magento \Catalog \Model \Product \Initialization \Helper \ProductLinks ;
9
10
10
11
/**
11
12
* Setup grouped product
@@ -17,6 +18,11 @@ class Product extends \Magento\CatalogSampleData\Model\Product
17
18
*/
18
19
protected $ productType = \Magento \GroupedProduct \Model \Product \Type \Grouped::TYPE_CODE ;
19
20
21
+ /**
22
+ * @var \Magento\Catalog\Model\Product\Initialization\Helper\ProductLinks
23
+ */
24
+ protected $ productLinksHelper ;
25
+
20
26
/**
21
27
* Product constructor.
22
28
* @param SampleDataContext $sampleDataContext
@@ -27,6 +33,7 @@ class Product extends \Magento\CatalogSampleData\Model\Product
27
33
* @param \Magento\CatalogSampleData\Model\Product\Gallery $gallery
28
34
* @param \Magento\Store\Model\StoreManagerInterface $storeManager
29
35
* @param \Magento\Eav\Model\Config $eavConfig
36
+ * @param \Magento\Catalog\Model\Product\Initialization\Helper\ProductLinks $productLinksHelper
30
37
*/
31
38
public function __construct (
32
39
SampleDataContext $ sampleDataContext ,
@@ -36,7 +43,8 @@ public function __construct(
36
43
\Magento \Framework \Setup \SampleData \FixtureManager $ fixtureManager ,
37
44
\Magento \CatalogSampleData \Model \Product \Gallery $ gallery ,
38
45
\Magento \Store \Model \StoreManagerInterface $ storeManager ,
39
- \Magento \Eav \Model \Config $ eavConfig
46
+ \Magento \Eav \Model \Config $ eavConfig ,
47
+ \Magento \Catalog \Model \Product \Initialization \Helper \ProductLinks $ productLinksHelper
40
48
) {
41
49
parent ::__construct (
42
50
$ sampleDataContext ,
@@ -47,5 +55,18 @@ public function __construct(
47
55
$ storeManager ,
48
56
$ eavConfig
49
57
);
58
+ $ this ->productLinksHelper = $ productLinksHelper ;
59
+ }
60
+
61
+ /**
62
+ * @param \Magento\Catalog\Model\Product $product
63
+ * @param array $data
64
+ * @return $this
65
+ */
66
+ protected function prepareProduct ($ product , $ data )
67
+ {
68
+ $ this ->productLinksHelper ->initializeLinks ($ product , $ data ['grouped_link_data ' ]);
69
+ $ product ->unsetData ('grouped_link_data ' );
70
+ return $ this ;
50
71
}
51
72
}
0 commit comments