MAGETWO-61556: Fixture generation (Configurables, Attribute Set)

Author: nikshostko

setup/src/Magento/Setup/Fixtures/AttributeSetsFixture.php

@@ -16,12 +16,27 @@ class AttributeSetsFixture extends Fixture
      */
     protected $priority = 25;
 
+    /**
+     * Cache for attribute IDs.
+     *
+     * @var array
+     */
+    private $attributeIdsCache = [];
+
+    /**
+     * Quantity of unique attributes to generate. Zero means infinity.
+     *
+     * @var int
+     */
+    protected $uniqueAttributesQuantity = 0;
+
     /**
      * {@inheritdoc}
      */
     public function execute()
     {
-        $attributeSets = $this->fixtureModel->getValue('attribute_sets', null);
+        $this->populateUniqueAttributesQuantity();
+        $attributeSets = $this->getAttributeSetsFixtureValue();
         if ($attributeSets === null) {
             return;
         }
@@ -71,37 +86,81 @@ public function execute()
             $attributesData = array_key_exists(0, $attributeSetData['attributes']['attribute'])
                 ? $attributeSetData['attributes']['attribute'] : [$attributeSetData['attributes']['attribute']];
             foreach ($attributesData as $attributeData) {
-                //Create Attribute
-                /** @var  \Magento\Catalog\Api\Data\ProductAttributeInterfaceFactory $attributeFactory */
-                $attributeFactory = $this->fixtureModel->getObjectManager()->create(
-                    \Magento\Catalog\Api\Data\ProductAttributeInterfaceFactory::class
-                );
-
-                $optionsData = array_key_exists(0, $attributeData['options']['option'])
-                    ? $attributeData['options']['option'] : [$attributeData['options']['option']];
-                $options = [];
-                foreach ($optionsData as $optionData) {
-                    /** @var \Magento\Eav\Api\Data\AttributeOptionInterfaceFactory $optionFactory */
-                    $optionFactory = $this->fixtureModel->getObjectManager()->create(
-                        \Magento\Eav\Api\Data\AttributeOptionInterfaceFactory::class
+                if ($this->uniqueAttributesQuantity === 0
+                    || (count($this->attributeIdsCache) < $this->uniqueAttributesQuantity)) {
+                    //Create Attribute
+                    /** @var  \Magento\Catalog\Api\Data\ProductAttributeInterfaceFactory $attributeFactory */
+                    $attributeFactory = $this->fixtureModel->getObjectManager()->create(
+                        \Magento\Catalog\Api\Data\ProductAttributeInterfaceFactory::class
                     );
-                    $option = $optionFactory->create(['data' => $optionData]);
-                    $options[] = $option;
-                }
 
-                $attribute = $attributeFactory->create(['data' => $attributeData]);
-                $attribute->setOptions($options);
+                    $optionsData = array_key_exists(0, $attributeData['options']['option'])
+                        ? $attributeData['options']['option'] : [$attributeData['options']['option']];
+                    $options = [];
+                    foreach ($optionsData as $optionData) {
+                        /** @var \Magento\Eav\Api\Data\AttributeOptionInterfaceFactory $optionFactory */
+                        $optionFactory = $this->fixtureModel->getObjectManager()->create(
+                            \Magento\Eav\Api\Data\AttributeOptionInterfaceFactory::class
+                        );
+                        $option = $optionFactory->create(['data' => $optionData]);
+                        $options[] = $option;
+                    }
 
-                $result = $attributeRepository->save($attribute);
-                $attributeId = $result->getAttributeId();
+                    $attribute = $attributeFactory->create(['data' => $attributeData]);
+                    $attribute->setOptions($options);
 
+                    $result = $attributeRepository->save($attribute);
+                    $attributeId = $result->getAttributeId();
+                    $this->fillAttributeIdsCache($attributeId);
+                } else {
+                    $attributeId = $this->getAttributeIdFromCache();
+                }
                 //Associate Attribute to Attribute Set
                 $sortOrder = 3;
+
                 $attributeManagement->assign($attributeSetId, $attributeGroupId, $attributeId, $sortOrder);
             }
         }
     }
 
+    /**
+     * Get attribute ID from cache.
+     *
+     * @return int
+     */
+    private function getAttributeIdFromCache()
+    {
+        $attributeId = next($this->attributeIdsCache);
+        if ($attributeId === false) {
+            $attributeId = reset($this->attributeIdsCache);
+        }
+
+        return $attributeId;
+    }
+
+    /**
+     * Fill attribute IDs cache.
+     *
+     * @param int $attributeId
+     * @return void
+     */
+    private function fillAttributeIdsCache($attributeId)
+    {
+        if ($this->uniqueAttributesQuantity !== 0) {
+            $this->attributeIdsCache[] = $attributeId;
+        }
+    }
+
+    /**
+     * Populate quantity of unique attributes to generate.
+     *
+     * @return void
+     */
+    protected function populateUniqueAttributesQuantity()
+    {
+        $this->uniqueAttributesQuantity = $this->fixtureModel->getValue('unique_attributes_quantity', 0);
+    }
+
     /**
      * {@inheritdoc}
      */
@@ -119,4 +178,14 @@ public function introduceParamLabels()
             'attribute_sets' => 'Attribute Sets'
         ];
     }
+
+    /**
+     * Get attribute sets fixture value.
+     *
+     * @return array|null
+     */
+    protected function getAttributeSetsFixtureValue()
+    {
+        return $this->fixtureModel->getValue('attribute_sets', null);
+    }
 }