fix test

SanderHagen · SanderHagen · commit 998c0ebe1e9b · 2022-04-01T16:07:28.000+02:00
diff --git a/src/Symfony/Component/Messenger/Tests/WorkerTest.php b/src/Symfony/Component/Messenger/Tests/WorkerTest.php
@@ -86,19 +86,20 @@ public function testHandlingErrorCausesReject()
         $this->assertSame(0, $receiver->getAcknowledgeCount());
     }
 
-    public function testWorkerDoesNotSendNullMessagesToTheBus()
+    public function testWorkerResetsConnectionIfReceiverIsResettable()
     {
-        $resettableReceiver = $this->createMock(ResetInterface::class);
-        $resettableReceiver->expects($this->once())->method('reset');
+        $resettableReceiver = new ResettableDummyReceiver([]);
 
         $bus = $this->createMock(MessageBusInterface::class);
         $dispatcher = new EventDispatcher();
 
         $worker = new Worker([$resettableReceiver], $bus, $dispatcher);
+        $worker->stop();
         $worker->run();
+        $this->assertTrue($resettableReceiver->hasBeenReset());
     }
 
-    public function testWorkerResetsConnectionIfReceiverIsResettable()
+    public function testWorkerDoesNotSendNullMessagesToTheBus()
     {
         $receiver = new DummyReceiver([
             null,
@@ -296,3 +297,18 @@ public function getRejectCount(): int
         return $this->rejectCount;
     }
 }
+
+class ResettableDummyReceiver extends DummyReceiver implements ResetInterface
+{
+    private $hasBeenReset = false;
+
+    public function reset()
+    {
+        $this->hasBeenReset = true;
+    }
+
+    public function hasBeenReset(): bool
+    {
+        return $this->hasBeenReset;
+    }
+}
