1 files changed, 30 insertions, 0 deletions
diff --git a/tests/Controller/OutputTest.php b/tests/Controller/OutputTest.php
index 7ca0aa4..d77ccd8 100644
--- a/tests/Controller/OutputTest.php
+++ b/tests/Controller/OutputTest.php
@@ -13,6 +13,36 @@ class OutputTest extends \PHPUnit_Framework_TestCase
 {
     use TestHelper;
 
+    
+    public function getSetResponseHeaderProvider()
+    {
+        return [
+            [true, 'withHeader'],
+            [false, 'withAddedHeader']
+        ];
+    }
+    
+    /**
+     * @dataProvider getSetResponseHeaderProvider
+     * 
+     * @param boolean $replace
+     * @param string  $method
+     */
+    public function testSetResponseHeader($replace, $method)
+    {
+        $response = $this->createMock(ResponseInterface::class);
+        $finalResponse = $this->createMock(ResponseInterface::class);
+        
+        $response->expects($this->once())->method($method)->with('Foo', 'bar')->willReturn($finalResponse);
+        
+        $controller = $this->getController(['getResponse', 'setResponse']);
+        
+        $controller->expects($this->once())->method('getResponse')->willReturn($response);
+        $controller->expects($this->once())->method('setResponse')->with($finalResponse);
+        
+        $controller->setResponseHeader('Foo', 'bar', $replace);
+    }
+    
     /**
      * Test function respondWith
      *