Enforce response type

mako-framework · May 30, 2024 · a09401c · a09401c
1 parent 86e820e
commit a09401c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/mako/http/Response.php b/src/mako/http/Response.php
@@ -13,6 +13,7 @@
 use mako\http\response\senders\ResponseSenderInterface;
 use mako\http\response\Status;
 use mako\security\Signer;
+use Stringable;
 
 use function hash;
 use function header;
@@ -98,7 +99,7 @@ public function getRequest(): Request
 	/**
 	 * Sets the response body.
 	 */
-	public function setBody(mixed $body): Response
+	public function setBody(ResponseBuilderInterface|ResponseSenderInterface|string|Stringable $body): Response
 	{
 		$this->body = $body;
 

diff --git a/tests/unit/http/routing/DispatcherTest.php b/tests/unit/http/routing/DispatcherTest.php
@@ -18,17 +18,23 @@
 use mako\syringe\Container;
 use mako\tests\TestCase;
 use Mockery;
+use Stringable;
 
 // --------------------------------------------------------------------------
 // START CLASSES
 // --------------------------------------------------------------------------
 
-class InjectMe
+class InjectMe implements Stringable
 {
 	public function helloWorld()
 	{
 		return 'Hello, world!';
 	}
+
+	public function __toString()
+	{
+		return $this->helloWorld();
+	}
 }
 
 class SimpleController extends Controller