Add 100% code coverage

snapshotpl · snapshotpl · commit 863ea995a15a · 2015-08-31T20:07:44.000+02:00
diff --git a/test/PhpDebugBarMiddlewareTest.php b/test/PhpDebugBarMiddlewareTest.php
@@ -4,6 +4,8 @@
 
 use DebugBar\JavascriptRenderer;
 use PhpMiddleware\PhpDebugBar\PhpDebugBarMiddleware;
+use Zend\Diactoros\Response;
+use Zend\Diactoros\ServerRequest;
 
 /**
  * PhpDebugBarMiddlewareTest
@@ -23,8 +25,8 @@ protected function setUp()
 
     public function testNotAttachIfNotAccept()
     {
-        $request = new \Zend\Diactoros\ServerRequest([], [], null, null, 'php://input', ['Accept: application/json']);
-        $response = new \Zend\Diactoros\Response();
+        $request = new ServerRequest();
+        $response = new Response();
         $calledOut = false;
         $outFunction = function ($request, $response) use (&$calledOut) {
             $calledOut = true;
@@ -36,4 +38,46 @@ public function testNotAttachIfNotAccept()
         $this->assertTrue($calledOut, 'Out is not called');
         $this->assertSame($response, $result);
     }
+
+    public function testAttachToNoneHtmlResponse()
+    {
+        $request = new ServerRequest([], [], null, null, 'php://input', ['Accept' => 'text/html']);
+        $response = new Response();
+        $response->getBody()->write('ResponseBody');
+        $calledOut = false;
+        $outFunction = function ($request, $response) use (&$calledOut) {
+            $calledOut = true;
+            return $response;
+        };
+
+        $this->debugbarRenderer->expects($this->once())->method('renderHead')->willReturn('RenderHead');
+        $this->debugbarRenderer->expects($this->once())->method('render')->willReturn('RenderBody');
+
+        $result = call_user_func($this->middleware, $request, $response, $outFunction);
+
+        $this->assertTrue($calledOut, 'Out is not called');
+        $this->assertNotSame($response, $result);
+        $this->assertSame("<html><head>RenderHead</head><body><h1>DebugBar</h1><p>Response:</p><pre>HTTP/1.1 200 OK\r\n\r\nResponseBody</pre>RenderBody</body></html>", (string) $result->getBody());
+    }
+
+    public function testAttachToHtmlResponse()
+    {
+        $request = new ServerRequest([], [], null, null, 'php://input', ['Accept' => 'text/html']);
+        $response = new Response('php://memory', 200, ['Content-Type' => 'text/html']);
+        $response->getBody()->write('ResponseBody');
+        $calledOut = false;
+        $outFunction = function ($request, $response) use (&$calledOut) {
+            $calledOut = true;
+            return $response;
+        };
+
+        $this->debugbarRenderer->expects($this->once())->method('renderHead')->willReturn('RenderHead');
+        $this->debugbarRenderer->expects($this->once())->method('render')->willReturn('RenderBody');
+
+        $result = call_user_func($this->middleware, $request, $response, $outFunction);
+
+        $this->assertTrue($calledOut, 'Out is not called');
+        $this->assertSame($response, $result);
+        $this->assertSame("ResponseBodyRenderHeadRenderBody", (string) $result->getBody());
+    }
 }
