Merge pull request #132 from leafsphp/v3.x-dev

V3.x dev
leafsphp · Aug 11, 2022 · 7cc488d · 7cc488d
2 parents ce7ce8b + 67713a4
commit 7cc488d
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/src/App.php b/src/App.php
@@ -254,6 +254,30 @@ public function config($name, $value = null)
         $this->setupErrorHandler();
     }
 
+    /**
+     * Swap out Leaf request instance
+     *
+     * @param mixed $class The new request class to attach
+     */
+    public function setRequestClass($class)
+    {
+        $this->container->singleton('request', function () {
+            return new \Leaf\Http\Request();
+        });
+    }
+
+    /**
+     * Swap out Leaf response instance
+     *
+     * @param mixed $class The new response class to attach
+     */
+    public function setResponseClass($class)
+    {
+        $this->container->singleton('response', function () use ($class) {
+            return new $class();
+        });
+    }
+
     /********************************************************************************
      * Logging
      *******************************************************************************/

diff --git a/tests/app.test.php b/tests/app.test.php
@@ -119,3 +119,18 @@ public function call()
 
     expect($app->config('inTest2'))->toBe('false');
 });
+
+test('swap out leaf response', function () {
+    class TestResponse extends \Leaf\Http\Response
+    {
+        public function customMethod()
+        {
+            return 'This is some test response';
+        }
+    }
+
+    $leafInstance1 = new \Leaf\App();
+    $leafInstance1->setResponseClass(TestResponse::class);
+
+    expect($leafInstance1->response->customMethod())->toBe('This is some test response');
+});