refactor(core): improvements on AbstractFiberHandler

Guikingone · May 4, 2022 · 406201a · 406201a
1 parent 335bacd
commit 406201a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/Fiber/AbstractFiberHandler.php b/src/Fiber/AbstractFiberHandler.php
@@ -5,9 +5,16 @@
 namespace SchedulerBundle\Fiber;
 
 use Closure;
+use DateTimeZone;
 use Fiber;
 use Psr\Log\LoggerInterface;
 use Psr\Log\NullLogger;
+use SchedulerBundle\Pool\Configuration\SchedulerConfiguration;
+use SchedulerBundle\Task\LazyTask;
+use SchedulerBundle\Task\LazyTaskList;
+use SchedulerBundle\Task\TaskInterface;
+use SchedulerBundle\Task\TaskListInterface;
+use SchedulerBundle\Transport\Configuration\ConfigurationInterface;
 use Throwable;
 use function sprintf;
 
@@ -23,7 +30,7 @@ public function __construct(?LoggerInterface $logger = null)
         $this->logger = $logger ?? new NullLogger();
     }
 
-    protected function handleOperationViaFiber(Closure $func): mixed
+    protected function handleOperationViaFiber(Closure $func): TaskListInterface|LazyTaskList|TaskInterface|LazyTask|SchedulerConfiguration|ConfigurationInterface|DateTimeZone|string|float|int|bool|array|null
     {
         $fiber = new Fiber(callback: function (Closure $operation): void {
             $value = $operation();

diff --git a/tests/Task/Builder/ChainedBuilderTest.php b/tests/Task/Builder/ChainedBuilderTest.php
@@ -110,6 +110,7 @@ public function testBuilderCanBuild(array $configuration): void
 
         self::assertInstanceOf(ChainedTask::class, $task);
         self::assertCount(3, $task->getTasks());
+        self::assertIsArray($configuration['tasks']);
         self::assertInstanceOf(ShellTask::class, $task->getTask($configuration['tasks'][0]['name']));
         self::assertSame('foo', $task->getTask($configuration['tasks'][0]['name'])->getName());
         self::assertInstanceOf(NullTask::class, $task->getTask($configuration['tasks'][1]['name']));