From 43c32b7b584ed3ded618cc2cc0b633ce497741f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Denis=20Duli=C3=A7i?= <denisdulici@gmail.com>
Date: Sat, 15 May 2021 10:00:14 +0300
Subject: [PATCH] fixed tests

---
 app/Jobs/Auth/CreateUser.php |  6 +++---
 app/Jobs/Auth/UpdateUser.php | 16 ++++++++++------
 2 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/app/Jobs/Auth/CreateUser.php b/app/Jobs/Auth/CreateUser.php
index 07c727a93..e01d4f71b 100644
--- a/app/Jobs/Auth/CreateUser.php
+++ b/app/Jobs/Auth/CreateUser.php
@@ -52,7 +52,9 @@ class CreateUser extends Job
             }
 
             if ($this->request->has('companies')) {
-                if (request()->isNotInstall()) {
+                if (app()->runningInConsole() || request()->isInstall()) {
+                    $this->user->companies()->attach($this->request->get('companies'));
+                } else {
                     $user = user();
 
                     $companies = $user->withoutEvents(function () use ($user) {
@@ -62,8 +64,6 @@ class CreateUser extends Job
                     if ($companies->isNotEmpty()) {
                         $this->user->companies()->attach($companies->toArray());
                     }
-                } else {
-                    $this->user->companies()->attach($this->request->get('companies'));
                 }
             }
 
diff --git a/app/Jobs/Auth/UpdateUser.php b/app/Jobs/Auth/UpdateUser.php
index 4b1123ff3..61b2d68ea 100644
--- a/app/Jobs/Auth/UpdateUser.php
+++ b/app/Jobs/Auth/UpdateUser.php
@@ -53,14 +53,18 @@ class UpdateUser extends Job
             }
 
             if ($this->request->has('companies')) {
-                $user = user();
+                if (app()->runningInConsole() || request()->isInstall()) {
+                    $this->user->companies()->attach($this->request->get('companies'));
+                } else {
+                    $user = user();
 
-                $companies = $user->withoutEvents(function () use ($user) {
-                    return $user->companies()->whereIn('id', $this->request->get('companies'))->pluck('id');
-                });
+                    $companies = $user->withoutEvents(function () use ($user) {
+                        return $user->companies()->whereIn('id', $this->request->get('companies'))->pluck('id');
+                    });
 
-                if ($companies->isNotEmpty()) {
-                    $this->user->companies()->sync($companies->toArray());
+                    if ($companies->isNotEmpty()) {
+                        $this->user->companies()->sync($companies->toArray());
+                    }
                 }
             }