Allow a timeout to be passed into the await* methods

Author: WyriHaximus

src/TestCase.php

@@ -15,6 +15,8 @@
 
 abstract class TestCase extends PHPUnitTestCase
 {
+    const DEFAULT_AWAIT_TIMEOUT = 60;
+
     /**
      * @var string
      */
@@ -146,40 +148,40 @@ protected function getFilesInDirectory(string $path): array
      * @param  LoopInterface|null $loop
      * @return mixed
      */
-    protected function await(PromiseInterface $promise, LoopInterface $loop = null)
+    protected function await(PromiseInterface $promise, LoopInterface $loop = null, float $timeout = self::DEFAULT_AWAIT_TIMEOUT)
     {
         if (!($loop instanceof LoopInterface)) {
             $loop = Factory::create();
         }
 
-        return await($promise, $loop);
+        return await($promise, $loop, $timeout);
     }
 
     /**
      * @param  array              $promises
      * @param  LoopInterface|null $loop
      * @return array
      */
-    protected function awaitAll(array $promises, LoopInterface $loop = null)
+    protected function awaitAll(array $promises, LoopInterface $loop = null, float $timeout = self::DEFAULT_AWAIT_TIMEOUT)
     {
         if (!($loop instanceof LoopInterface)) {
             $loop = Factory::create();
         }
 
-        return awaitAll($promises, $loop);
+        return awaitAll($promises, $loop, $timeout);
     }
 
     /**
      * @param  array              $promises
      * @param  LoopInterface|null $loop
      * @return mixed
      */
-    protected function awaitAny(array $promises, LoopInterface $loop = null)
+    protected function awaitAny(array $promises, LoopInterface $loop = null, float $timeout = self::DEFAULT_AWAIT_TIMEOUT)
     {
         if (!($loop instanceof LoopInterface)) {
             $loop = Factory::create();
         }
 
-        return awaitAny($promises, $loop);
+        return awaitAny($promises, $loop, $timeout);
     }
 }

tests/TestCaseTest.php

@@ -6,7 +6,10 @@
 use React\EventLoop\Factory;
 use React\EventLoop\LoopInterface;
 use React\EventLoop\StreamSelectLoop;
+use React\Promise\Deferred;
 use function React\Promise\resolve;
+use function React\Promise\Timer\timeout;
+use React\Promise\Timer\TimeoutException;
 
 final class TestCaseTest extends TestCase
 {
@@ -94,4 +97,14 @@ public function testTrueFalse(bool $bool)
     {
         static::assertInternalType('bool', $bool);
     }
+
+    /**
+     * @dataProvider provideEventLoop
+     */
+    public function testAwaitTimeout(LoopInterface $loop = null)
+    {
+        self::expectException(TimeoutException::class);
+
+        $this->await((new Deferred())->promise(), $loop, 0.1);
+    }
 }