fix: Apply body to controller under test

codeigniter4 · Nov 12, 2022 · 3f5e063 · 3f5e063
1 parent a9859e8
commit 3f5e063
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/system/Test/ControllerTestTrait.php b/system/Test/ControllerTestTrait.php
@@ -108,7 +108,7 @@ protected function setUpControllerTestTrait(): void
             $tempUri = Services::uri();
             Services::injectMock('uri', $this->uri);
 
-            $this->withRequest(Services::request($this->appConfig, false)->setBody($this->body));
+            $this->withRequest(Services::request($this->appConfig, false));
 
             // Restore the URI service
             Services::injectMock('uri', $tempUri);
@@ -156,6 +156,7 @@ public function execute(string $method, ...$params)
         }
 
         $response = null;
+        $this->request->setBody($this->body);
 
         try {
             ob_start();

diff --git a/tests/system/Test/ControllerTestTraitTest.php b/tests/system/Test/ControllerTestTraitTest.php
@@ -13,10 +13,12 @@
 
 use App\Controllers\Home;
 use App\Controllers\NeverHeardOfIt;
+use CodeIgniter\Controller;
 use CodeIgniter\Log\Logger;
 use CodeIgniter\Test\Mock\MockLogger as LoggerConfig;
 use Config\App;
 use Config\Services;
+use Exception;
 use Tests\Support\Controllers\Popcorn;
 
 /**
@@ -241,4 +243,20 @@ public function testRedirectRoute()
             ->execute('toindex');
         $this->assertTrue($result->isRedirect());
     }
+
+    public function testUsesRequestBody()
+    {
+        $this->controller = new class () extends Controller {
+            public function throwsBody(): void
+            {
+                throw new Exception($this->request->getBody());
+            }
+        };
+        $this->controller->initController($this->request, $this->response, $this->logger);
+
+        $this->expectException(Exception::class);
+        $this->expectExceptionMessage('banana');
+
+        $this->withBody('banana')->execute('throwsBody');
+    }
 }