Fix custom columns, add tests (#122)

* Fix custom columns & add test

* Apply fixes from StyleCI

src/Tenant.php

@@ -90,7 +90,7 @@ class Tenant extends Model
 
     public function put(string $key, $value)
     {
-        if (\array_key_exists($key, $this->customColumns())) {
+        if (\in_array($key, $this->customColumns())) {
             $this->update([$key => $value]);
         } else {
             $obj = \json_decode($this->{$this->dataColumn()});

tests/Etc/2019_08_08_000001_add_custom_column.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+
+use Illuminate\Support\Facades\Schema;
+use Illuminate\Database\Schema\Blueprint;
+use Illuminate\Database\Migrations\Migration;
+
+class AddCustomColumn extends Migration
+{
+    /**
+     * Run the migrations.
+     *
+     * @return void
+     */
+    public function up()
+    {
+        Schema::table('tenants', function (Blueprint $table) {
+            $table->string('foo')->nullable();
+        });
+    }
+
+    public function down()
+    {
+    }
+}

tests/TenantStorageTest.php

@@ -170,4 +170,32 @@ class TenantStorageTest extends TestCase
         $this->assertSame('bar', tenancy()->get('foo'));
         $this->assertSame(['bar'], tenancy()->get(['foo']));
     }
+
+    /** @test */
+    public function custom_columns_work_with_db_storage_driver()
+    {
+        if (config('tenancy.storage_driver') != 'Stancl\Tenancy\StorageDrivers\DatabaseStorageDriver') {
+            $this->markTestSkipped();
+        }
+
+        tenancy()->end();
+
+        $this->loadMigrationsFrom([
+            '--path' => __DIR__ . '/Etc',
+            '--database' => 'central',
+        ]);
+        config(['database.default' => 'sqlite']); // fix issue caused by loadMigrationsFrom
+
+        config(['tenancy.storage.db.custom_columns' => [
+            'foo',
+        ]]);
+
+        tenancy()->create('foo.localhost');
+        tenancy()->init('foo.localhost');
+
+        tenancy()->put(['foo' => 'bar', 'abc' => 'xyz']);
+        $this->assertSame(['bar', 'xyz'], tenancy()->get(['foo', 'abc']));
+
+        $this->assertSame('bar', \DB::connection('central')->table('tenants')->where('uuid', tenant('uuid'))->first()->foo);
+    }
 }