REDROOM

Navigation

Breadcrumbs

/ / home / ecopackply / www / demo / system / Test

Create

Create File | Create Folder | Upload Files | Run Command

Edit File Size: 7.55 KB

/home/ecopackply/www/demo/system/Test/ControllerTestTrait.php

Text Base64

<?php

/**
 * This file is part of CodeIgniter 4 framework.
 *
 * (c) CodeIgniter Foundation <admin@codeigniter.com>
 *
 * For the full copyright and license information, please view
 * the LICENSE file that was distributed with this source code.
 */

namespace CodeIgniter\Test;

use CodeIgniter\Controller;
use CodeIgniter\Exceptions\InvalidArgumentException;
use CodeIgniter\HTTP\Exceptions\HTTPException;
use CodeIgniter\HTTP\IncomingRequest;
use CodeIgniter\HTTP\ResponseInterface;
use CodeIgniter\HTTP\URI;
use Config\App;
use Config\Services;
use Psr\Log\LoggerInterface;
use Throwable;

/**
 * Controller Test Trait
 *
 * Provides features that make testing controllers simple and fluent.
 *
 * Example:
 *
 *  $this->withRequest($request)
 *       ->withResponse($response)
 *       ->withUri($uri)
 *       ->withBody($body)
 *       ->controller('App\Controllers\Home')
 *       ->execute('methodName');
 */
trait ControllerTestTrait
{
    /**
     * Controller configuration.
     *
     * @var App
     */
    protected $appConfig;

/**
     * Request.
     *
     * @var IncomingRequest
     */
    protected $request;

/**
     * Response.
     *
     * @var ResponseInterface
     */
    protected $response;

/**
     * Message logger.
     *
     * @var LoggerInterface
     */
    protected $logger;

/**
     * Initialized controller.
     *
     * @var Controller
     */
    protected $controller;

/**
     * URI of this request.
     *
     * @var string
     */
    protected $uri = 'http://example.com';

/**
     * Request body.
     *
     * @var string|null
     */
    protected $body;

/**
     * Initializes required components.
     */
    protected function setUpControllerTestTrait(): void
    {
        // The URL helper is always loaded by the system so ensure it is available.
        helper('url');

if (! $this->appConfig instanceof App) {
            $this->appConfig = config(App::class);
        }

if (! $this->uri instanceof URI) {
            $factory   = Services::siteurifactory($this->appConfig, service('superglobals'), false);
            $this->uri = $factory->createFromGlobals();
        }

if (! $this->request instanceof IncomingRequest) {
            // Do some acrobatics, so we can use the Request service with our own URI
            $tempUri = service('uri');
            Services::injectMock('uri', $this->uri);

$this->withRequest(service('incomingrequest', $this->appConfig, false));

// Restore the URI service
            Services::injectMock('uri', $tempUri);
        }

if (! $this->response instanceof ResponseInterface) {
            $this->response = service('response', $this->appConfig, false);
        }

if (! $this->logger instanceof LoggerInterface) {
            $this->logger = service('logger');
        }
    }

/**
     * Loads the specified controller, and generates any needed dependencies.
     *
     * @return $this
     */
    public function controller(string $name)
    {
        if (! class_exists($name)) {
            throw new InvalidArgumentException('Invalid Controller: ' . $name);
        }

$this->controller = new $name();
        $this->controller->initController($this->request, $this->response, $this->logger);

return $this;
    }

/**
     * Runs the specified method on the controller and returns the results.
     *
     * @param array $params
     *
     * @return TestResponse
     *
     * @throws InvalidArgumentException
     */
    public function execute(string $method, ...$params)
    {
        if (! method_exists($this->controller, $method) || ! is_callable([$this->controller, $method])) {
            throw new InvalidArgumentException('Method does not exist or is not callable in controller: ' . $method);
        }

$response = null;
        $this->request->setBody($this->body);

try {
            ob_start();
            // The controller method param types may not be string.
            // So cannot set `declare(strict_types=1)` in this file.
            $response = $this->controller->{$method}(...$params);
        } catch (Throwable $e) {
            $code = $e->getCode();

// If code is not a valid HTTP status then assume there is an error
            if ($code < 100 || $code >= 600) {
                throw $e;
            }
        } finally {
            $output = ob_get_clean();
        }

// If the controller returned a view then add it to the output
        if (is_string($response)) {
            $output = is_string($output) ? $output . $response : $response;
        }

// If the controller did not return a response then start one
        if (! $response instanceof ResponseInterface) {
            $response = $this->response;
        }

// Check for output to set or prepend
        // @see \CodeIgniter\CodeIgniter::gatherOutput()
        if (is_string($output)) {
            if (is_string($response->getBody())) {
                $response->setBody($output . $response->getBody());
            } else {
                $response->setBody($output);
            }
        }

// Check for an overriding code from exceptions
        if (isset($code)) {
            $response->setStatusCode($code);
        }
        // Otherwise ensure there is a status code
        else {
            // getStatusCode() throws for empty codes
            try {
                $response->getStatusCode();
            } catch (HTTPException) {
                // If no code has been set then assume success
                $response->setStatusCode(200);
            }
        }

// Create the result and add the Request for reference
        return (new TestResponse($response))->setRequest($this->request);
    }

/**
     * Set controller's config, with method chaining.
     *
     * @param App $appConfig
     *
     * @return $this
     */
    public function withConfig($appConfig)
    {
        $this->appConfig = $appConfig;

return $this;
    }

/**
     * Set controller's request, with method chaining.
     *
     * @param IncomingRequest $request
     *
     * @return $this
     */
    public function withRequest($request)
    {
        $this->request = $request;

// Make sure it's available for other classes
        Services::injectMock('request', $request);

return $this;
    }

/**
     * Set controller's response, with method chaining.
     *
     * @param ResponseInterface $response
     *
     * @return $this
     */
    public function withResponse($response)
    {
        $this->response = $response;

return $this;
    }

/**
     * Set controller's logger, with method chaining.
     *
     * @param LoggerInterface $logger
     *
     * @return $this
     */
    public function withLogger($logger)
    {
        $this->logger = $logger;

return $this;
    }

/**
     * Set the controller's URI, with method chaining.
     *
     * @return $this
     */
    public function withUri(string $uri)
    {
        $factory   = service('siteurifactory');
        $this->uri = $factory->createFromString($uri);
        Services::injectMock('uri', $this->uri);

// Update the Request instance, because Request has the SiteURI instance.
        $this->request = service('incomingrequest', null, false);
        Services::injectMock('request', $this->request);

return $this;
    }

/**
     * Set the method's body, with method chaining.
     *
     * @param string|null $body
     *
     * @return $this
     */
    public function withBody($body)
    {
        $this->body = $body;

return $this;
    }
}

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Directory Contents

Dirs: 4 × Files: 15

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
Constraints DIR	-	drwxr-xr-x	2026-01-24 08:10:54	Edit Download recursive recursive
Filters DIR	-	drwxr-xr-x	2026-01-24 08:10:54	Edit Download recursive recursive
Interfaces DIR	-	drwxr-xr-x	2026-01-24 08:11:01	Edit Download recursive recursive
Mock DIR	-	drwxr-xr-x	2026-01-24 08:11:37	Edit Download recursive recursive
bootstrap.php	3.48 KB	lrw-r--r--	2026-01-24 08:03:33	Edit Download recursive recursive
CIUnitTestCase.php	13.61 KB	lrw-r--r--	2026-01-24 08:03:33	Edit Download recursive recursive
ConfigFromArrayTrait.php	1.09 KB	lrw-r--r--	2026-01-24 08:03:35	Edit Download recursive recursive
ControllerTestTrait.php	7.55 KB	lrw-r--r--	2026-01-24 08:03:35	Edit Download recursive recursive
DatabaseTestTrait.php	9.70 KB	lrw-r--r--	2026-01-24 08:03:37	Edit Download recursive recursive
DOMParser.php	7.84 KB	lrw-r--r--	2026-01-24 08:03:37	Edit Download recursive recursive
Fabricator.php	17.08 KB	lrw-r--r--	2026-01-24 08:03:39	Edit Download recursive recursive
FeatureTestTrait.php	11.82 KB	lrw-r--r--	2026-01-24 08:03:39	Edit Download recursive recursive
FilterTestTrait.php	9.19 KB	lrw-r--r--	2026-01-24 08:03:41	Edit Download recursive recursive
IniTestTrait.php	729 B	lrw-r--r--	2026-01-24 08:03:41	Edit Download recursive recursive
PhpStreamWrapper.php	1.57 KB	lrw-r--r--	2026-01-24 08:03:43	Edit Download recursive recursive
ReflectionHelper.php	2.47 KB	lrw-r--r--	2026-01-24 08:03:43	Edit Download recursive recursive
StreamFilterTrait.php	979 B	lrw-r--r--	2026-01-24 08:03:46	Edit Download recursive recursive
TestLogger.php	2.40 KB	lrw-r--r--	2026-01-24 08:03:46	Edit Download recursive recursive
TestResponse.php	13.51 KB	lrw-r--r--	2026-01-24 08:03:48	Edit Download recursive recursive