Controller check response status methods

2 files changed, 150 insertions, 2 deletions

diff --git a/src/Controller.php b/src/Controller.php
index 8dfa176..ddec2c0 100644
--- a/src/Controller.php
+++ b/src/Controller.php
@@ -63,5 +63,73 @@ abstract class Controller
 
         return $this->run();
     }
+
+    /**
+     * Check if response is 2xx succesful, or empty
+     * 
+     * @return boolean
+     */
+    public function isSuccessful()
+    {
+        $code = $this->getResponseStatusCode();
+
+        return !$code || ($code >= 200 && $code < 300);
+    }
+    
+    /**
+     * Check if response is a 3xx redirect
+     * 
+     * @return boolean
+     */
+    public function isRedirection()
+    {
+        $code = $this->getResponseStatusCode();
+
+        return $code >= 300 && $code < 400;
+    }
+    
+    /**
+     * Check if response is a 4xx client error
+     * 
+     * @return boolean
+     */
+    public function isClientError()
+    {
+        $code = $this->getResponseStatusCode();
+
+        return $code >= 400 && $code < 500;
+    }
+    
+    /**
+     * Check if response is a 5xx redirect
+     * 
+     * @return boolean
+     */
+    public function isServerError()
+    {
+        return $this->getResponseStatusCode() >= 500;
+    }   
+
+    /**
+      * Check if response is 4xx or 5xx error
+      *
+      * @return boolean
+      */
+     public function isError()
+     {
+         return $this->isClientError() || $this->isServerError();
+     } 
+
+     /**
+      * Get status code of response
+      *
+      * @return int
+      */
+     protected function getResponseStatusCode()
+     {
+        $response = $this->getResponse();
+
+        return $response ? $response->getStatusCode() : 0;
+     }
 }
 
diff --git a/tests/ControllerTest.php b/tests/ControllerTest.php
index 9a4caef..8cdbf7d 100644
--- a/tests/ControllerTest.php
+++ b/tests/ControllerTest.php
@@ -12,8 +12,7 @@ class ControllerTest extends PHPUnit_Framework_TestCase
     public function testInvoke()
     {
         $controller = $this->getMockBuilder(Controller::class)->disableOriginalConstructor()->getMockForAbstractClass();
-        $request = $this->createMock(ServerRequestInterface::class);
-        $response = $this->createMock(ResponseInterface::class);
+        list($request, $response) = $this->getRequests();
 
         $controller->expects($this->once())->method('run')->will($this->returnValue($response));
 
@@ -23,4 +22,85 @@ class ControllerTest extends PHPUnit_Framework_TestCase
         $this->assertEquals($response, $controller->getResponse(), "Can not get 'ResponseInterface' instance from controller");
         $this->assertEquals($request, $controller->getRequest(), "Can not get 'ServerRequestInterface' instance from controller");
     }
+
+    /**
+     * Test response status functions if response object is not set
+     */
+    public function testResponseStatusEmptyResponse()
+    {
+        $controller = $this->getMockBuilder(Controller::class)->disableOriginalConstructor()->getMockForAbstractClass();   
+        $data = $this->getStatusCodesMap(null);
+
+        foreach ($data as $func => $value) {
+            $this->assertEquals($value, $controller->$func(), "Method '$func' returns incorrect value");
+        }
+    }
+
+    /**
+     * Test functions that check response status code
+     *
+     * @dataProvider responseStatusProvider
+     * @param int $statusCode
+     */
+    public function testResponseStatus($code)
+    {
+        $controller = $this->getMockBuilder(Controller::class)->disableOriginalConstructor()->getMockForAbstractClass();
+        list($request, $response) = $this->getRequests();
+        $response->method('getStatusCode')->will($this->returnValue($code));
+
+        $controller($request, $response);                
+
+        $data = $this->getStatusCodesMap($code);
+
+        foreach ($data as $func => $value) {
+            $this->assertEquals($value, $controller->$func(), "Method '$func' returns incorrect value");
+        }
+
+        $this->assertEquals($data['isClientError'] || $data['isServerError'], $controller->isError(), "Method 'isError' returns incorrect value");
+    }
+
+    /**
+     * Provide data for testing status methods
+     *
+     * @return array
+     */
+    public function responseStatusProvider()
+    {
+        return [
+            [null], [199],
+            [200], [201], [299],
+            [300], [304], [399],
+            [400], [403], [499],
+            [500], [503]
+        ];
+    }
+
+    /**
+     * Get map of status codes to states
+     *
+     * @param int $code
+     * @return []
+     */
+    public function getStatusCodesMap($code)
+    {
+        return [
+            'isSuccessful' => !$code || ($code >= 200 && $code < 300),
+            'isRedirection' => $code >= 300 && $code < 400,
+            'isClientError' => $code >= 400 && $code < 500,
+            'isServerError' => $code >= 500            
+        ];
+    }
+
+    /**
+     * Get request and response instances
+     *
+     * @return array
+     */
+    public function getRequests()
+    {
+        return [
+            $this->createMock(ServerRequestInterface::class),
+            $this->createMock(ResponseInterface::class)            
+        ];
+    }
 }