Fix issue #626

Author: Moln

src/Container/WhoopsFactory.php

@@ -53,7 +53,6 @@ public function __invoke(ContainerInterface $container) : Whoops
         $config = $config['whoops'] ?? [];
 
         $whoops = new Whoops();
-        $whoops->writeToOutput(false);
         $whoops->allowQuit(false);
         $whoops->pushHandler($container->get('Zend\Expressive\WhoopsPageHandler'));
         $this->registerJsonHandler($whoops, $config);

src/Middleware/WhoopsErrorResponseGenerator.php

@@ -79,9 +79,12 @@ public function __invoke(
 
         $response = $response->withStatus(Utils::getStatusCode($e, $response));
 
+        $sendOutput = $this->whoops->writeToOutput();
+        $this->whoops->writeToOutput(false);
         $response
             ->getBody()
             ->write($this->whoops->handleException($e));
+        $this->whoops->writeToOutput($sendOutput);
 
         return $response;
     }

test/Middleware/WhoopsErrorResponseGeneratorTest.php

@@ -55,9 +55,13 @@ public function setUp()
     public function testWritesResultsOfWhoopsExceptionsHandlingToResponse()
     {
         $error = new RuntimeException();
+        $sendOutput = true;
 
         $this->whoops->getHandlers()->willReturn([]);
         $this->whoops->handleException($error)->willReturn('WHOOPS');
+        $this->whoops->writeToOutput()->willReturn($sendOutput);
+        $this->whoops->writeToOutput(false)->shouldBeCalled();
+        $this->whoops->writeToOutput($sendOutput)->shouldBeCalled();
 
         // Could do more assertions here, but these will be sufficent for
         // ensuring that the method for injecting metadata is never called.
@@ -81,6 +85,7 @@ public function testWritesResultsOfWhoopsExceptionsHandlingToResponse()
     public function testAddsRequestMetadataToWhoopsPrettyPageHandler()
     {
         $error = new RuntimeException('STATUS_INTERNAL_SERVER_ERROR', StatusCode::STATUS_INTERNAL_SERVER_ERROR);
+        $sendOutput = true;
 
         $handler = $this->prophesize(PrettyPageHandler::class);
         $handler
@@ -98,6 +103,9 @@ public function testAddsRequestMetadataToWhoopsPrettyPageHandler()
 
         $this->whoops->getHandlers()->willReturn([$handler->reveal()]);
         $this->whoops->handleException($error)->willReturn('WHOOPS');
+        $this->whoops->writeToOutput()->willReturn($sendOutput);
+        $this->whoops->writeToOutput(false)->shouldBeCalled();
+        $this->whoops->writeToOutput($sendOutput)->shouldBeCalled();
 
         $this->request->getAttribute('originalUri', false)->willReturn('https://example.com/foo');
         $this->request->getAttribute('originalRequest', false)->will([$this->request, 'reveal']);
@@ -126,6 +134,7 @@ public function testAddsRequestMetadataToWhoopsPrettyPageHandler()
     public function testJsonContentTypeResponseWithJsonResponseHandler()
     {
         $error = new RuntimeException('STATUS_NOT_IMPLEMENTED', StatusCode::STATUS_NOT_IMPLEMENTED);
+        $sendOutput = true;
 
         $handler = $this->prophesize(JsonResponseHandler::class);
 
@@ -135,6 +144,9 @@ public function testJsonContentTypeResponseWithJsonResponseHandler()
 
         $this->whoops->getHandlers()->willReturn([$handler->reveal()]);
         $this->whoops->handleException($error)->willReturn('error');
+        $this->whoops->writeToOutput()->willReturn($sendOutput);
+        $this->whoops->writeToOutput(false)->shouldBeCalled();
+        $this->whoops->writeToOutput($sendOutput)->shouldBeCalled();
 
         $this->request->getAttribute('originalUri', false)->willReturn('https://example.com/foo');
         $this->request->getAttribute('originalRequest', false)->will([$this->request, 'reveal']);