Split exception handling methods

TavoNiievez · TavoNiievez · commit 051173c62c52 · 2021-09-07T17:25:11.000-05:00
diff --git a/src/Codeception/Module/Laravel.php b/src/Codeception/Module/Laravel.php
@@ -16,6 +16,7 @@
 use Codeception\Module\Laravel\InteractsWithContainer;
 use Codeception\Module\Laravel\InteractsWithEloquent;
 use Codeception\Module\Laravel\InteractsWithEvents;
+use Codeception\Module\Laravel\InteractsWithExceptionHandling;
 use Codeception\Subscriber\ErrorHandler;
 use Codeception\TestInterface;
 use Codeception\Util\ReflectionHelper;
@@ -132,6 +133,7 @@ class Laravel extends Framework implements ActiveRecord, PartedModule
     use InteractsWithContainer;
     use InteractsWithEloquent;
     use InteractsWithEvents;
+    use InteractsWithExceptionHandling;
 
     /**
      * @var Application
@@ -293,32 +295,6 @@ protected function revertErrorHandler(): void
         set_error_handler([$handler, 'errorHandler']);
     }
 
-    /**
-     * Enable Laravel exception handling.
-     *
-     * ```php
-     * <?php
-     * $I->enableExceptionHandling();
-     * ```
-     */
-    public function enableExceptionHandling()
-    {
-        $this->client->enableExceptionHandling();
-    }
-
-    /**
-     * Disable Laravel exception handling.
-     *
-     * ```php
-     * <?php
-     * $I->disableExceptionHandling();
-     * ```
-     */
-    public function disableExceptionHandling()
-    {
-        $this->client->disableExceptionHandling();
-    }
-
     /**
      * Disable middleware for the next requests.
      *
diff --git a/src/Codeception/Module/Laravel/InteractsWithExceptionHandling.php b/src/Codeception/Module/Laravel/InteractsWithExceptionHandling.php
@@ -0,0 +1,34 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Codeception\Module\Laravel;
+
+trait InteractsWithExceptionHandling
+{
+    /**
+     * Disable Laravel exception handling.
+     *
+     * ```php
+     * <?php
+     * $I->disableExceptionHandling();
+     * ```
+     */
+    public function disableExceptionHandling()
+    {
+        $this->client->disableExceptionHandling();
+    }
+
+    /**
+     * Enable Laravel exception handling.
+     *
+     * ```php
+     * <?php
+     * $I->enableExceptionHandling();
+     * ```
+     */
+    public function enableExceptionHandling()
+    {
+        $this->client->enableExceptionHandling();
+    }
+}
