tests: Add unit tests

Author: matapatos

README.md

@@ -1,3 +1,3 @@
 # WP Exceptions
 
-Stop returning WP_Error's. Throw exceptions instead.
+*WP_Error* was a cool WordPress feature in 2007 but today, we should throw exceptions instead.

composer.json

@@ -2,7 +2,7 @@
   "name": "attributes-php/wp-exceptions",
   "type": "library",
   "license": "MIT",
-  "description": "Rely on exceptions instead of the WP_Error. WordPress exceptions made easy",
+  "description": "Stop returning WP_Error. Throw exceptions instead",
   "authors": [
     {
       "name": "André Gil"

rector.php

@@ -10,6 +10,7 @@
 
 namespace Attributes\Wp\Exceptions;
 
+use Exception;
 use Throwable;
 
 use function wp_die;
@@ -34,6 +35,8 @@ class ExceptionHandler
      */
     protected array $onExceptionHandlers = [];
 
+    protected $invoker = null;
+
     /**
      * Registers the exception handler
      */
@@ -87,7 +90,13 @@ public function __invoke(Throwable $ex): void
             throw $ex;
         }
 
-        call_user_func($handler, $ex);
+        if (! $this->invoker) {
+            call_user_func($handler, $ex);
+
+            return;
+        }
+
+        call_user_func($this->invoker, $handler, $ex);
     }
 
     /**
@@ -121,12 +130,18 @@ protected function handleHttpException(HttpException $ex): void
             require_once ABSPATH.WPINC.'/class-wp-error.php';
         }
 
+        if (! headers_sent() && (! function_exists('did_action') || ! did_action('admin_head'))) {
+            foreach ($ex->getHeaders() as $name => $value) {
+                header("$name: $value");
+            }
+        }
+
         wp_die($ex->toWpError());
     }
 
     /**
-     * Retrieves an instance of this class. If it doesn't exist will store it on a global variable,
-     * to allow compatibility with tools like Mozart which change the namespace.
+     * Retrieves an instance of this class. If it doesn't exist creates a new one and stores it in a global
+     * variable, to be compatible with tools like Mozart or PHP-Scoper which change the namespace.
      */
     public static function getInstance(): ExceptionHandler
     {
@@ -140,4 +155,9 @@ public static function getInstance(): ExceptionHandler
 
         return $attributes_wp_exceptions_exception_handler;
     }
+
+    public function setInvoker(callable $invoker): void
+    {
+        $this->invoker = $invoker;
+    }
 }

src/ExceptionHandler.php

@@ -35,47 +35,55 @@ class HttpException extends Exception
 
     protected ?WP_Error $wpError = null;
 
+    /**
+     * @param  int  $status  HTTP status code to be sent
+     * @param  string  $message  Short description
+     * @param  array<string,mixed>  $data  Additional data to be sent
+     * @param  array<string,mixed>  $headers  Headers to be sent
+     */
     public function __construct(
         int $status = WP_Http::INTERNAL_SERVER_ERROR,
         string $message = 'Something went wrong',
         array $data = [],
         array $headers = [],
         ?Throwable $previous = null,
-        ?WP_Error $wpError = null,
     ) {
+        $this->status = $status;
         $this->data = $data;
-        $this->data['status'] = $status;
+        $this->data['status'] = $this->status;
 
         $this->headers = $headers;
-        $this->status = $status;
-        $this->wpError = $wpError;
         parent::__construct($message, $this->status, $previous);
     }
 
     /**
      * Creates a new instance using a WP_Error class
+     *
+     * @param  WP_Error  $error  The WP_Error to base the exception from. Nested WP_Error's are also supported.
+     * @param  int  $defaultStatus  If the provided error code is a string and no 'data.status' is set, this value is used.
      */
     public static function fromWpError(WP_Error $error, int $defaultStatus = 500): HttpException
     {
         $data = $error->get_error_data();
         $data = $data ?: [];
         $data = is_array($data) ? $data : ['data' => $data];
 
-        $status = $error->get_error_code();
-        if (! is_int($status)) {
-            $status = isset($data['status']) && is_int($data['status']) ? $data['status'] : $defaultStatus;
-        }
+        $status = isset($data['status']) && is_int($data['status']) ? $data['status'] : $error->get_error_code();
+        $status = is_int($status) ? $status : $defaultStatus;
 
-        return new HttpException(
+        $httpException = new HttpException(
             $status,
             $error->get_error_message(),
             $data,
-            [],
-            null,
-            $error,
         );
+        $httpException->wpError = $error;
+
+        return $httpException;
     }
 
+    /**
+     * Converts this exception into a WP_Error object
+     */
     public function toWpError(): WP_Error
     {
         if ($this->wpError) {