MCP-996: Update Profile Generator Tool with MSI

Updated StoresFixture.php so that other Fixtures can see which websites were added.
Added FixtureRegistry so that we can have Fixtures from other modules outside of magento2ce

The new Stocks fixture is in commit the MSI repository

Author: JacobBrownAustin

setup/performance-toolkit/profiles/ce/small.xml

@@ -11,6 +11,8 @@
         <websites>1</websites> <!-- Number of websites to generate -->
         <store_groups>1</store_groups> <!--Number of stores-->
         <store_views>1</store_views> <!-- Number of store views -->
+        <sources>1</sources> <!-- Number of sources to generate -->
+        <stocks>1</stocks> <!-- Number of stocks to generate -->
         <assign_entities_to_all_websites>0</assign_entities_to_all_websites> <!-- Whether to assign all products per each website -->
         <simple_products>800</simple_products> <!-- Simple products count -->
         <configurable_products> <!-- Configurable product -->

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

@@ -44,14 +44,14 @@ class FixtureModel
     /**
      * List of fixtures applied to the application
      *
-     * @var \Magento\Setup\Fixtures\Fixture[]
+     * @var Fixture[]
      */
     protected $fixtures = [];
 
     /**
      * List of fixtures indexed by class names
      *
-     * @var \Magento\Setup\Fixtures\Fixture[]
+     * @var Fixture[]
      */
     private $fixturesByNames = [];
 
@@ -109,35 +109,52 @@ public function reindex(OutputInterface $output)
     public function loadFixtures()
     {
         $files = glob(__DIR__ . DIRECTORY_SEPARATOR . self::FIXTURE_PATTERN, GLOB_NOSORT);
-
         foreach ($files as $file) {
             $file = basename($file, '.php');
-            /** @var \Magento\Setup\Fixtures\Fixture $fixture */
+            /** @var Fixture $fixture */
             $type = 'Magento\Setup\Fixtures' . '\\' . $file;
             $fixture = $this->getObjectManager()->create(
                 $type,
                 [
                     'fixtureModel' => $this,
                 ]
             );
-
-            if (isset($this->fixtures[$fixture->getPriority()])) {
-                throw new \InvalidArgumentException(
-                    sprintf('Duplicate priority %d in fixture %s', $fixture->getPriority(), $type)
-                );
-            }
-
-            if ($fixture->getPriority() >= 0) {
-                $this->fixtures[$fixture->getPriority()] = $fixture;
-            }
-
-            $this->fixturesByNames[get_class($fixture)] = $fixture;
+            $this->loadFixture($fixture);
+        }
+        foreach($this->getFixturesFromRegistry() as $fixture) {
+            $this->loadFixture($fixture);
         }
-
         ksort($this->fixtures);
         return $this;
     }
 
+    private function getFixturesFromRegistry() : array
+    {
+        $fixtureRegistry = $this->getObjectManager()->create(FixtureRegistry::class);
+        $fixtures = [];
+        foreach ($fixtureRegistry->getFixtures() as $fixtureClassName) {
+            $fixtures[] = $this->getObjectManager()->create(
+                $fixtureClassName,
+                ['fixtureModel' => $this]
+            );
+        }
+        return $fixtures;
+    }
+
+    private function loadFixture(Fixture $fixture)
+    {
+        $fixtureClassName = get_class($fixture);
+        if (isset($this->fixtures[$fixture->getPriority()])) {
+            throw new \InvalidArgumentException(
+                sprintf('Duplicate priority %d in fixture %s', $fixture->getPriority(), $fixtureClassName)
+            );
+        }
+        if ($fixture->getPriority() >= 0) {
+            $this->fixtures[$fixture->getPriority()] = $fixture;
+        }
+        $this->fixturesByNames[$fixtureClassName] = $fixture;
+    }
+
     /**
      * Get param labels
      *
@@ -162,7 +179,7 @@ public function getFixtures()
     /**
      * Returns fixture by name
      * @param $name string
-     * @return \Magento\Setup\Fixtures\Fixture
+     * @return Fixture
      * @throws \Magento\Setup\Exception
      */
     public function getFixtureByName($name)

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

@@ -0,0 +1,40 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+/**
+ * Registry for fixtures
+ */
+namespace Magento\Setup\Fixtures;
+
+class FixtureRegistry
+{
+
+    /**
+     * List of fixtures applied to the application
+     *
+     * @var Fixture[]
+     */
+    private $fixtures = [];
+
+    /**
+     * @param array $fixtures
+     */
+    public function __construct(array $fixtures = [])
+    {
+        $this->fixtures = $fixtures;
+    }
+
+    /**
+     * Get fixtures
+     *
+     * @return string[]
+     */
+    public function getFixtures() :array
+    {
+        return $this->fixtures;
+    }
+
+}

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

@@ -132,6 +132,12 @@ class StoresFixture extends Fixture
      */
     private $storeViewIds;
 
+
+    /**
+     * @var string[]
+     */
+    private $websiteCodes = [];
+
     /**
      * @var ManagerInterface
      */
@@ -183,7 +189,6 @@ public function execute()
         $this->storeGroupsCount = $this->fixtureModel->getValue('store_groups', self::DEFAULT_STORE_COUNT);
         $this->storesCount = $this->fixtureModel->getValue('store_views', self::DEFAULT_STORE_VIEW_COUNT);
         $this->singleRootCategory = (bool)$this->fixtureModel->getValue('assign_entities_to_all_websites', false);
-
         if ($this->websitesCount <= self::DEFAULT_WEBSITE_COUNT
             && $this->storeGroupsCount <= self::DEFAULT_STORE_COUNT
             && $this->storesCount <= self::DEFAULT_STORE_VIEW_COUNT
@@ -207,7 +212,6 @@ public function execute()
         $this->defaultStoreGroupId = $this->defaultStoreGroup->getId();
         $this->defaultStoreView = $this->storeManager->getDefaultStoreView();
         $this->storeViewIds = array_keys($this->storeManager->getStores());
-
         $this->generateWebsites();
         $this->generateStoreGroups();
         $this->generateStoreViews();
@@ -220,7 +224,6 @@ public function execute()
     private function generateWebsites()
     {
         $existedWebsitesCount = count($this->websiteIds) + self::DEFAULT_WEBSITE_COUNT;
-
         while ($existedWebsitesCount <= $this->websitesCount) {
             $website = clone $this->defaultWebsite;
             $websiteCode = sprintf('website_%d', $existedWebsitesCount);
@@ -236,6 +239,7 @@ private function generateWebsites()
             $website->save();
             $this->websiteIds[] = $website->getId();
             $existedWebsitesCount++;
+            $this->websiteCodes[] = $websiteCode;
         }
     }
 
@@ -380,4 +384,9 @@ private function getDefaultCategoryId()
         }
         return $this->defaultParentCategoryId;
     }
+
+    public function getWebsiteCodes()
+    {
+        return $this->websiteCodes;
+    }
 }