From d2df9a9866f3a3f4412b2400eaa4b55b390abb22 Mon Sep 17 00:00:00 2001
From: Devon Mather <devon.mather@beta.fashion>
Date: Fri, 20 Mar 2020 19:47:56 +0100
Subject: [PATCH] Add compatability for commands in Laravel 7

---
 src/Traits/TenantAwareCommand.php       |  2 ++
 tests/Etc/AddUserCommand.php            | 35 +++++++++++++++++++++++++
 tests/Traits/TenantAwareCommandTest.php | 31 ++++++++++++++++++++++
 3 files changed, 68 insertions(+)
 create mode 100644 tests/Etc/AddUserCommand.php
 create mode 100644 tests/Traits/TenantAwareCommandTest.php

diff --git a/src/Traits/TenantAwareCommand.php b/src/Traits/TenantAwareCommand.php
index 6cdb22e8..2d18a265 100644
--- a/src/Traits/TenantAwareCommand.php
+++ b/src/Traits/TenantAwareCommand.php
@@ -26,6 +26,8 @@ trait TenantAwareCommand
                 $this->laravel->call([$this, 'handle']);
             });
         }
+
+        return 1;
     }
 
     /**
diff --git a/tests/Etc/AddUserCommand.php b/tests/Etc/AddUserCommand.php
new file mode 100644
index 00000000..7df7ee6d
--- /dev/null
+++ b/tests/Etc/AddUserCommand.php
@@ -0,0 +1,35 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Stancl\Tenancy\Tests\Etc;
+
+use Illuminate\Console\Command;
+use Stancl\Tenancy\Traits\TenantAwareCommand;
+
+class AddUserCommand extends Command
+{
+    use TenantAwareCommand;
+
+    /**
+     * The name and signature of the console command.
+     *
+     * @var string
+     */
+    protected $signature = 'user:add';
+
+    /**
+     * Execute the console command.
+     *
+     * @return mixed
+     */
+    public function handle()
+    {
+        User::create();
+    }
+}
+
+class User extends \Illuminate\Database\Eloquent\Model
+{
+    protected $guarded = [];
+}
diff --git a/tests/Traits/TenantAwareCommandTest.php b/tests/Traits/TenantAwareCommandTest.php
new file mode 100644
index 00000000..9752682d
--- /dev/null
+++ b/tests/Traits/TenantAwareCommandTest.php
@@ -0,0 +1,31 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Stancl\Tenancy\Tests\Traits;
+
+use Stancl\Tenancy\Tenant;
+use Stancl\Tenancy\Tests\TestCase;
+
+class TenantAwareCommandTest extends TestCase
+{
+    public $autoCreateTenant = false;
+    public $autoInitTenancy = false;
+
+    /** @test */
+    public function commands_run_globally_are_tenant_aware()
+    {
+        $tenant1 = Tenant::new()->save();
+        $tenant2 = Tenant::new()->save();
+
+        $this->artisan('user:add')->assertExitCode(1);
+
+        tenancy()->initializeTenancy($tenant1);
+        $this->assertNotEmpty(User::all());
+        tenancy()->endTenancy()
+
+        tenancy()->initializeTenancy($tenant2);
+        $this->assertNotEmpty(User::all());
+        tenancy()->endTenancy()
+    }
+}