throw an exception when route handler does not return response

1 files changed, 8 insertions, 1 deletions

diff --git a/src/BasicStrategy.php b/src/BasicStrategy.php
index 62bc48e..5043576 100644
--- a/src/BasicStrategy.php
+++ b/src/BasicStrategy.php
@@ -5,6 +5,7 @@ declare(strict_types=1);
 namespace Lightscale\Router;
 
 use Lightscale\Router\Contracts\Strategy;
+use Lightscale\Router\Exceptions\InvalidResponseException;
 use Lightscale\Router\Exceptions\NotFoundException;
 use Psr\Http\Message\RequestInterface;
 use Psr\Http\Message\ResponseInterface;
@@ -13,7 +14,13 @@ class BasicStrategy implements Strategy
 {
     public function runRoute(RouteCall $call): ResponseInterface
     {
-        return ($call->route->getHandler())($call);
+        $result = ($call->route->getHandler())($call);
+
+        if ($result instanceof ResponseInterface) {
+            return $result;
+        }
+
+        throw new InvalidResponseException();
     }
 
     public function notFound(RequestInterface $request): ResponseInterface