fix integration tests

AlptisAssurances · AlptisAssurances · commit a4f1205b1f98 · 2018-04-20T21:22:58.000+02:00
diff --git a/src/Symfony/Component/Messenger/Tests/Adapter/AmqpExt/AmqpExtIntegrationTest.php b/src/Symfony/Component/Messenger/Tests/Adapter/AmqpExt/AmqpExtIntegrationTest.php
@@ -16,7 +16,6 @@
 use Symfony\Component\Messenger\Adapter\AmqpExt\AmqpSender;
 use Symfony\Component\Messenger\Adapter\AmqpExt\Connection;
 use Symfony\Component\Messenger\Tests\Fixtures\DummyMessage;
-use Symfony\Component\Messenger\Transport\Enhancers\MaximumCountReceiver;
 use Symfony\Component\Messenger\Transport\Serialization\Serializer;
 use Symfony\Component\Process\PhpProcess;
 use Symfony\Component\Process\Process;
@@ -116,9 +115,15 @@ public function testItSupportsTimeoutAndTicksNullMessagesToTheHandler()
         $connection->queue()->purge();
 
         $sender = new AmqpSender($serializer, $connection);
-        $receiver = new MaximumCountReceiver(new AmqpReceiver($serializer, $connection), 2);
-        $receiver->receive(function ($message) {
+        $receiver = new AmqpReceiver($serializer, $connection);
+
+        $receivedMessages = 0;
+        $receiver->receive(function ($message) use ($receiver, &$receivedMessages) {
             $this->assertNull($message);
+
+            if (2 == ++$receivedMessages) {
+                $receiver->stop();
+            }
         });
     }
 
