Merge pull request #173 from jakubkulhan/0.6.x-strengthen-channel-consume-test-assertations

[0.6.x] Strengthen channel consume test assertations

Author: WyriHaximus

test/ChannelTest.php

@@ -11,7 +11,9 @@
 use Bunny\Message;
 use Bunny\Test\Library\ClientHelper;
 use PHPUnit\Framework\TestCase;
+use React\Promise\Deferred;
 use WyriHaximus\React\PHPUnit\RunTestsInFibersTrait;
+use function React\Async\await;
 use function str_repeat;
 
 class ChannelTest extends TestCase
@@ -92,34 +94,46 @@ public function testPublish(): void
 
     public function testConsume(): void
     {
+        /**
+         * @var Deferred<string> $deferred
+         */
+        $deferred = new Deferred();
         $c = $this->helper->createClient();
 
         $ch = $c->connect()->channel();
         self::assertTrue($c->isConnected());
         $ch->queueDeclare('test_queue', false, false, false, true);
         self::assertTrue($c->isConnected());
-        $ch->consume(static function (Message $msg, Channel $ch, Client $c): void {
-            self::assertEquals('hi', $msg->content);
+        $ch->consume(static function (Message $msg, Channel $ch, Client $c) use ($deferred): void {
+            $deferred->resolve($msg->content);
         });
         self::assertTrue($c->isConnected());
         $ch->publish('hi', [], '', 'test_queue');
+        self::assertEquals('hi', await($deferred->promise()));
+
         self::assertTrue($c->isConnected());
         $c->disconnect();
         self::assertFalse($c->isConnected());
     }
 
     public function testHeaders(): void
     {
+        /**
+         * @var Deferred<bool> $deferred
+         */
+        $deferred = new Deferred();
         $c = $this->helper->createClient();
 
         $ch = $c->connect()->channel();
         $ch->queueDeclare('test_queue', false, false, false, true);
-        $ch->consume(static function (Message $msg, Channel $ch, Client $c): void {
+        $ch->consume(static function (Message $msg, Channel $ch, Client $c) use ($deferred): void {
             self::assertTrue($msg->hasHeader('content-type'));
             self::assertEquals('text/html', $msg->getHeader('content-type'));
             self::assertEquals('<b>hi html</b>', $msg->content);
+            $deferred->resolve(true);
         });
         $ch->publish('<b>hi html</b>', ['content-type' => 'text/html'], '', 'test_queue');
+        self::assertTrue(await($deferred->promise()));
 
         self::assertTrue($c->isConnected());
         $c->disconnect();
@@ -128,16 +142,22 @@ public function testHeaders(): void
 
     public function testBigMessage(): void
     {
+        /**
+         * @var Deferred<bool> $deferred
+         */
+        $deferred = new Deferred();
         $body = str_repeat('a', 10 << 20 /* 10 MiB */);
 
         $c = $this->helper->createClient();
 
         $ch = $c->connect()->channel();
         $ch->queueDeclare('test_queue', false, false, false, true);
-        $ch->consume(static function (Message $msg, Channel $ch, Client $c) use ($body): void {
+        $ch->consume(static function (Message $msg, Channel $ch, Client $c) use ($body, $deferred): void {
             self::assertEquals($body, $msg->content);
+            $deferred->resolve(true);
         });
         $ch->publish($body, [], '', 'test_queue');
+        self::assertTrue(await($deferred->promise()));
 
         self::assertTrue($c->isConnected());
         $c->disconnect();