Tests and fixes for RouteAction

1 files changed, 2 insertions, 11 deletions

diff --git a/tests/ControllerTest.php b/tests/ControllerTest.php
index 04323a1..b980d48 100644
--- a/tests/ControllerTest.php
+++ b/tests/ControllerTest.php
@@ -31,7 +31,8 @@ class ControllerTest extends \PHPUnit_Framework_TestCase
                 $test->assertSame($request, $this->getRequest());
                 $test->assertSame($response, $this->getResponse());
                 
-                return $finalResponse;
+                $this->setResponse($finalResponse);
+                return null;
             }, $controller, Controller::class));
 
         $result = $controller($request, $response);
@@ -39,16 +40,6 @@ class ControllerTest extends \PHPUnit_Framework_TestCase
         $this->assertEquals($finalResponse, $result);
     }
     
-    public function testSetResponse()
-    {
-        $response = $this->createMock(ResponseInterface::class);
-        
-        $controller = $this->getController();
-        $controller->setResponse($response);
-        
-        $this->assertSame($response, $controller->getResponse());
-    }
-    
     /**
      * @expectedException LogicException
      */