Add JsonResponseHandler test case, add previously forgotten import

Author: Mateusz Lopacinski

src/Middleware/WhoopsErrorResponseGenerator.php

@@ -10,6 +10,7 @@
 use InvalidArgumentException;
 use Psr\Http\Message\ResponseInterface;
 use Psr\Http\Message\ServerRequestInterface;
+use Whoops\Handler\JsonResponseHandler;
 use Whoops\Handler\PrettyPageHandler;
 use Whoops\Run;
 use Whoops\RunInterface;

test/Middleware/WhoopsErrorResponseGeneratorTest.php

@@ -13,6 +13,7 @@
 use Psr\Http\Message\ServerRequestInterface;
 use Psr\Http\Message\StreamInterface;
 use RuntimeException;
+use Whoops\Handler\JsonResponseHandler;
 use Whoops\Handler\PrettyPageHandler;
 use Whoops\Run;
 use Whoops\RunInterface;
@@ -111,4 +112,37 @@ public function testAddsRequestMetadataToWhoopsPrettyPageHandler()
             $generator($error, $this->request->reveal(), $this->response->reveal())
         );
     }
+
+    public function testJsonContentTypeResponseWithJsonResponseHandler()
+    {
+        $error = new RuntimeException();
+
+        $handler = $this->prophesize(JsonResponseHandler::class);
+        $handler->contentType()->willReturn('application/json');
+
+        $this->whoops->getHandlers()->willReturn([$handler->reveal()]);
+        $this->whoops->handleException($error)->willReturn('error');
+
+        $this->request->getAttribute('originalUri', false)->willReturn('https://example.com/foo');
+        $this->request->getAttribute('originalRequest', false)->will([$this->request, 'reveal']);
+        $this->request->getMethod()->willReturn('POST');
+        $this->request->getServerParams()->willReturn(['SCRIPT_NAME' => __FILE__]);
+        $this->request->getHeaders()->willReturn([]);
+        $this->request->getCookieParams()->willReturn([]);
+        $this->request->getAttributes()->willReturn([]);
+        $this->request->getQueryParams()->willReturn([]);
+        $this->request->getParsedBody()->willReturn([]);
+
+        $this->response->withHeader('Content-Type', 'application/json')->will([$this->response, 'reveal']);
+        $this->response->getBody()->will([$this->stream, 'reveal']);
+
+        $this->stream->write('error')->shouldBeCalled();
+
+        $generator = new WhoopsErrorResponseGenerator($this->whoops->reveal());
+
+        $this->assertSame(
+            $this->response->reveal(),
+            $generator($error, $this->request->reveal(), $this->response->reveal())
+        );
+    }
 }