Update test for HTTP 413

reactphp · Oct 19, 2017 · d6b95a9 · d6b95a9
1 parent 96d43b6
commit d6b95a9
Showing 1 changed file with 4 additions and 16 deletions.
diff --git a/tests/Middleware/RequestBodyBufferMiddlewareTest.php b/tests/Middleware/RequestBodyBufferMiddlewareTest.php
@@ -10,7 +10,6 @@
 use React\EventLoop\Factory;
 use React\Tests\Http\TestCase;
 use RingCentral\Psr7\BufferStream;
-use Clue\React\Block;
 
 final class RequestBodyBufferMiddlewareTest extends TestCase
 {
@@ -70,33 +69,22 @@ public function testExcessiveSizeReturnsError413()
         $loop = Factory::create();
 
         $stream = new ThroughStream();
+        $stream->end('aa');
         $serverRequest = new ServerRequest(
             'GET',
             'https://example.com/',
             array(),
-            new HttpBodyStream($stream)
+            new HttpBodyStream($stream, 2)
         );
 
         $buffer = new RequestBodyBufferMiddleware(1);
-        $promise = $buffer(
+        $response = $buffer(
             $serverRequest,
             function (ServerRequestInterface $request) {
                 return $request;
             }
         );
 
-        $stream->write('aa');
-
-        $exposedResponse = null;
-        $promise->then(
-            function($response) use (&$exposedResponse) {
-                $exposedResponse = $response;
-            }, 
-            $this->expectCallableNever()
-        );
-
-        $this->assertSame(413, $exposedResponse->getStatusCode());
-
-        Block\await($promise, $loop);
+        $this->assertSame(413, $response->getStatusCode());
     }
 }