Fix mocks and logger

andrewmy · andrewmy · commit c0ce9b6f3381 · 2021-01-24T21:33:39.000+02:00
diff --git a/pkg/enqueue-bundle/Tests/Unit/Consumption/Extension/DoctrinePingConnectionExtensionTest.php b/pkg/enqueue-bundle/Tests/Unit/Consumption/Extension/DoctrinePingConnectionExtensionTest.php
@@ -135,10 +135,6 @@ public function testShouldSkipIfConnectionWasNotOpened()
         ;
 
         $context = $this->createContext();
-        $context->getLogger()
-            ->expects($this->never())
-            ->method('debug')
-        ;
 
         $registry = $this->createRegistryMock();
         $registry
@@ -149,6 +145,10 @@ public function testShouldSkipIfConnectionWasNotOpened()
 
         $extension = new DoctrinePingConnectionExtension($registry);
         $extension->onMessageReceived($context);
+
+        /** @var TestLogger $logger */
+        $logger = $context->getLogger();
+        $this->assertFalse($logger->hasDebugRecords());
     }
 
     protected function createContext(): MessageReceived
diff --git a/pkg/enqueue/Tests/Client/ProducerSendCommandTest.php b/pkg/enqueue/Tests/Client/ProducerSendCommandTest.php
@@ -479,7 +479,7 @@ public function testShouldCallPostSendExtensionMethod()
 
         $extension
             ->expects(self::once())
-            ->method('onDriverPreSend')
+            ->method('onPostSend')
             ->willReturnCallback(function (PostSend $context) use ($message, $producer, $driver) {
                 $this->assertSame($message, $context->getMessage());
                 $this->assertSame($producer, $context->getProducer());
diff --git a/pkg/enqueue/Tests/Client/ProducerSendEventTest.php b/pkg/enqueue/Tests/Client/ProducerSendEventTest.php
@@ -499,7 +499,7 @@ public function testShouldCallPostSendExtensionMethodWhenSendToApplicationRouter
 
         $extension
             ->expects(self::once())
-            ->method('onDriverPreSend')
+            ->method('onPostSend')
             ->willReturnCallback(function (PostSend $context) use ($message, $producer, $driver) {
                 $this->assertSame($message, $context->getMessage());
                 $this->assertSame($producer, $context->getProducer());
