Pass environment to seed from manager.

cakephp · Aug 24, 2022 · d40b220 · d40b220
1 parent 18bf93e
commit d40b220
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 5 deletions.
diff --git a/src/Phinx/Migration/Manager.php b/src/Phinx/Migration/Manager.php
@@ -602,7 +602,7 @@ public function rollback(string $environment, $target = null, bool $force = fals
      */
     public function seed(string $environment, ?string $seed = null): void
     {
-        $seeds = $this->getSeeds();
+        $seeds = $this->getSeeds($environment);
 
         if ($seed === null) {
             // run all seeders
@@ -918,10 +918,11 @@ protected function orderSeedsByDependencies(array $seeds): array
     /**
      * Gets an array of database seeders.
      *
+     * @param string $environment Environment
      * @throws \InvalidArgumentException
      * @return \Phinx\Seed\SeedInterface[]
      */
-    public function getSeeds(): array
+    public function getSeeds(string $environment): array
     {
         if ($this->seeds === null) {
             $phpFiles = $this->getSeedFiles();
@@ -958,6 +959,7 @@ public function getSeeds(): array
                     } else {
                         $seed = new $class();
                     }
+                    $seed->setEnvironment($environment);
                     $input = $this->getInput();
                     if ($input !== null) {
                         $seed->setInput($input);

diff --git a/src/Phinx/Seed/AbstractSeed.php b/src/Phinx/Seed/AbstractSeed.php
@@ -24,6 +24,11 @@
  */
 abstract class AbstractSeed implements SeedInterface
 {
+    /**
+     * @var string
+     */
+    protected $environment;
+
     /**
      * @var \Phinx\Db\Adapter\AdapterInterface
      */
@@ -61,6 +66,24 @@ public function getDependencies(): array
         return [];
     }
 
+    /**
+     * @inheritDoc
+     */
+    public function setEnvironment(string $environment)
+    {
+        $this->environment = $environment;
+
+        return $this;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function getEnvironment()
+    {
+        return $this->environment;
+    }
+
     /**
      * @inheritDoc
      */

diff --git a/src/Phinx/Seed/SeedInterface.php b/src/Phinx/Seed/SeedInterface.php
@@ -42,6 +42,20 @@ public function run(): void;
      */
     public function getDependencies(): array;
 
+    /**
+     * Sets the environment.
+     *
+     * @return \Phinx\Seed\SeedInterface
+     */
+    public function setEnvironment(string $environment);
+
+    /**
+     * Gets the environment.
+     *
+     * @return string
+     */
+    public function getEnvironment();
+
     /**
      * Sets the database adapter.
      *

diff --git a/tests/Phinx/Migration/ManagerTest.php b/tests/Phinx/Migration/ManagerTest.php
@@ -5450,7 +5450,7 @@ public function testExecuteANonExistentSeedWorksAsExpectedWithMixedNamespace()
 
     public function testOrderSeeds()
     {
-        $seeds = array_values($this->manager->getSeeds());
+        $seeds = array_values($this->manager->getSeeds('mockenv'));
         $this->assertInstanceOf('UserSeeder', $seeds[0]);
         $this->assertInstanceOf('GSeeder', $seeds[1]);
         $this->assertInstanceOf('PostSeeder', $seeds[2]);
@@ -5472,7 +5472,7 @@ public function testSeedWillNotBeExecuted()
     public function testGettingInputObject()
     {
         $migrations = $this->manager->getMigrations('mockenv');
-        $seeds = $this->manager->getSeeds();
+        $seeds = $this->manager->getSeeds('mockenv');
         $inputObject = $this->manager->getInput();
         $this->assertInstanceOf('\Symfony\Component\Console\Input\InputInterface', $inputObject);
 
@@ -5487,7 +5487,7 @@ public function testGettingInputObject()
     public function testGettingOutputObject()
     {
         $migrations = $this->manager->getMigrations('mockenv');
-        $seeds = $this->manager->getSeeds();
+        $seeds = $this->manager->getSeeds('mockenv');
         $outputObject = $this->manager->getOutput();
         $this->assertInstanceOf('\Symfony\Component\Console\Output\OutputInterface', $outputObject);