fix integration tests

AlptisAssurances · AlptisAssurances · commit f836b55f0c9d · 2018-04-20T21:21:26.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
@@ -116,9 +116,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();
+            }
         });
     }
 
