minor #52085 [Scheduler] postRun event test is missing (alli83)

fabpot · fabpot · commit 3fb0e11df2e1 · 2023-10-17T07:43:44.000+02:00
This PR was merged into the 6.4 branch. Discussion ---------- [Scheduler] postRun event test is missing | Q | A | ------------- | --- | Branch? | 6.4 | Bug fix? | no | New feature? | no | Deprecations? | | Tickets | | License | MIT The PostRun event test is missing in `DispatchSchedulerEventListenerTest` and there is an error to be rectified Commits ------- c263424 [Scheduler] postRun event test is missing
diff --git a/src/Symfony/Component/Scheduler/Tests/EventListener/DispatchSchedulerEventListenerTest.php b/src/Symfony/Component/Scheduler/Tests/EventListener/DispatchSchedulerEventListenerTest.php
@@ -15,7 +15,9 @@
 use Psr\Container\ContainerInterface;
 use Symfony\Component\EventDispatcher\EventDispatcher;
 use Symfony\Component\Messenger\Envelope;
+use Symfony\Component\Messenger\Event\WorkerMessageHandledEvent;
 use Symfony\Component\Messenger\Event\WorkerMessageReceivedEvent;
+use Symfony\Component\Scheduler\Event\PostRunEvent;
 use Symfony\Component\Scheduler\Event\PreRunEvent;
 use Symfony\Component\Scheduler\EventListener\DispatchSchedulerEventListener;
 use Symfony\Component\Scheduler\Generator\MessageContext;
@@ -33,20 +35,22 @@ public function testDispatchSchedulerEvents()
 
         $schedulerProvider = new SomeScheduleProvider([$defaultRecurringMessage]);
         $scheduleProviderLocator = $this->createMock(ContainerInterface::class);
-        $scheduleProviderLocator->expects($this->once())->method('has')->willReturn(true);
-        $scheduleProviderLocator->expects($this->once())->method('get')->willReturn($schedulerProvider);
+        $scheduleProviderLocator->expects($this->any())->method('has')->willReturn(true);
+        $scheduleProviderLocator->expects($this->any())->method('get')->willReturn($schedulerProvider);
 
         $context = new MessageContext('default', 'default', $trigger, $this->createMock(\DateTimeImmutable::class));
         $envelope = (new Envelope(new \stdClass()))->with(new ScheduledStamp($context));
 
         /** @var ContainerInterface $scheduleProviderLocator */
         $listener = new DispatchSchedulerEventListener($scheduleProviderLocator, $eventDispatcher = new EventDispatcher());
         $workerReceivedEvent = new WorkerMessageReceivedEvent($envelope, 'default');
+        $workerHandledEvent = new WorkerMessageHandledEvent($envelope, 'default');
         $secondListener = new TestEventListener();
 
         $eventDispatcher->addListener(PreRunEvent::class, [$secondListener, 'preRun']);
-        $eventDispatcher->addListener(PreRunEvent::class, [$secondListener, 'postRun']);
+        $eventDispatcher->addListener(PostRunEvent::class, [$secondListener, 'postRun']);
         $listener->onMessageReceived($workerReceivedEvent);
+        $listener->onMessageHandled($workerHandledEvent);
 
         $this->assertTrue($secondListener->preInvoked);
         $this->assertTrue($secondListener->postInvoked);
