From b2c3392a832c95444b328dde06dab3c1546023f1 Mon Sep 17 00:00:00 2001
From: Franz Liedke <franz@develophp.org>
Date: Fri, 7 Feb 2020 23:29:14 +0100
Subject: [PATCH] Integration tests: Allow registering extenders

---
 framework/core/tests/integration/TestCase.php | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/framework/core/tests/integration/TestCase.php b/framework/core/tests/integration/TestCase.php
index cdb3fdc9a..687023dbe 100644
--- a/framework/core/tests/integration/TestCase.php
+++ b/framework/core/tests/integration/TestCase.php
@@ -10,6 +10,7 @@
 namespace Flarum\Tests\integration;
 
 use Dflydev\FigCookies\SetCookie;
+use Flarum\Extend\ExtenderInterface;
 use Flarum\Foundation\InstalledSite;
 use Illuminate\Database\ConnectionInterface;
 use Laminas\Diactoros\CallbackStream;
@@ -41,13 +42,24 @@ abstract class TestCase extends \PHPUnit\Framework\TestCase
                 include __DIR__.'/tmp/config.php'
             );
 
+            $site->extendWith($this->extenders);
+
             $this->app = $site->bootApp();
-            $this->server = $this->app->getRequestHandler();
         }
 
         return $this->app;
     }
 
+    /**
+     * @var ExtenderInterface[]
+     */
+    protected $extenders = [];
+
+    protected function extend(ExtenderInterface $extender)
+    {
+        $this->extenders[] = $extender;
+    }
+
     /**
      * @var RequestHandlerInterface
      */