allow definition of init task after adding workers

aternosorg · Jul 10, 2024 · 9a17344 · 9a17344
1 parent f89230a
commit 9a17344
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/Taskmaster.php b/src/Taskmaster.php
@@ -602,6 +602,10 @@ public function setDefaultInitTask(TaskInterface|string $task): static
             $this->initTaskFactory = new CloneTaskFactory($task);
         }
 
+        foreach ($this->workers as $worker) {
+            $worker->setInitTaskFactoryIfNecessary($this->initTaskFactory);
+        }
+
         return $this;
     }
 

diff --git a/test/Integration/AsyncWorkerTestCase.php b/test/Integration/AsyncWorkerTestCase.php
@@ -100,4 +100,20 @@ public function testInitTask(): void
         }
         $this->assertEquals(3, $counter);
     }
+
+    public function testInitTaskAfterWorkerCreation(): void
+    {
+        $this->taskmaster = new Taskmaster();
+        $this->taskmaster->addWorkers($this->createWorker(), 3);
+        $this->taskmaster->setDefaultInitTask(InitTask::class);
+
+        $this->addTasks(new InitValidateTask(), 3);
+        $counter = 0;
+        foreach ($this->taskmaster->waitAndHandleTasks() as $task) {
+            $this->assertNull($task->getError());
+            $this->assertTrue($task->getResult());
+            $counter++;
+        }
+        $this->assertEquals(3, $counter);
+    }
 }