jgrossi · August 14, 2021 18:14 · Sep 20, 2017 · Sep 20, 2017
diff --git a/AttachJwtToken.php b/AttachJwtToken.php
@@ -7,12 +7,28 @@
 
 trait AttachJwtToken
 {
+    /**
+     * @var User
+     */
+    protected $loginUser;
+
+    /**
+     * @param User $user
+     * @return $this
+     */
+    public function loginAs(User $user)
+    {
+        $this->loginUser = $user;
+
+        return $this;
+    }
+
     /**
      * @return string
      */
     protected function getJwtToken()
     {
-        $user = factory(User::class)->create([
+        $user = $this->loginUser ?: factory(User::class)->create([
             'company_id' => null,
         ]);
 

diff --git a/AttachJwtToken.php b/AttachJwtToken.php
@@ -0,0 +1,61 @@
+<?php
+
+namespace Tests\Concerns;
+
+use App\Models\User;
+use Tymon\JWTAuth\Facades\JWTAuth;
+
+trait AttachJwtToken
+{
+    /**
+     * @return string
+     */
+    protected function getJwtToken()
+    {
+        $user = factory(User::class)->create([
+            'company_id' => null,
+        ]);
+
+        return JWTAuth::fromUser($user);
+    }
+
+    /**
+     * @param string $method
+     * @param string $uri
+     * @param array $parameters
+     * @param array $cookies
+     * @param array $files
+     * @param array $server
+     * @param string $content
+     * @return \Illuminate\Http\Response
+     */
+    public function call($method, $uri, $parameters = [], $cookies = [], $files = [], $server = [], $content = null)
+    {
+        if ($this->requestNeedsToken($method, $uri)) {
+            $server = $this->attachToken($server);
+        }
+
+        return parent::call($method, $uri, $parameters, $cookies, $files, $server, $content);
+    }
+
+    /**
+     * @param string $method
+     * @param string $uri
+     * @return bool
+     */
+    protected function requestNeedsToken($method, $uri)
+    {
+        return !('/auth/login' === $uri && 'POST' === $method);
+    }
+
+    /**
+     * @param array $server
+     * @return string
+     */
+    protected function attachToken(array $server)
+    {
+        return array_merge($server, $this->transformHeadersToServerVars([
+            'Authorization' => 'Bearer ' . $this->getJwtToken(),
+        ]));
+    }
+}