tenancy/tests/TenantClassTest.php

<?php

declare(strict_types=1);

namespace Stancl\Tenancy\Tests;

use Mockery;
use Stancl\Tenancy\Contracts\StorageDriver;
use Stancl\Tenancy\Tenant;
use Tenancy;

class TenantClassTest extends TestCase
{
    public $autoInitTenancy = false;
    public $autoCreateTenant = false;

    /** @test */
    public function data_cache_works_properly()
    {
        // $spy = Mockery::spy(config('tenancy.storage_driver'))->makePartial();
        // $this->instance(StorageDriver::class, $spy);

        $tenant = Tenant::create(['foo.localhost'], ['foo' => 'bar']);
        $this->assertSame('bar', $tenant->data['foo']);

        $tenant->put('abc', 'xyz');
        $this->assertSame('xyz', $tenant->data['abc']);

        $tenant->put(['aaa' => 'bbb', 'ccc' => 'ddd']);
        $this->assertSame('bbb', $tenant->data['aaa']);
        $this->assertSame('ddd', $tenant->data['ccc']);

        // $spy->shouldNotHaveReceived('get');

        $this->assertSame(null, $tenant->dfuighdfuigfhdui);
        // $spy->shouldHaveReceived('get')->once();

        Mockery::close();
    }

    /** @test */
    public function tenant_can_have_multiple_domains()
    {
        $tenant = Tenant::create(['foo.localhost', 'bar.localhost']);
        $this->assertSame(['foo.localhost', 'bar.localhost'], $tenant->domains);
        $this->assertSame($tenant->id, Tenancy::findByDomain('foo.localhost')->id);
        $this->assertSame($tenant->id, Tenancy::findByDomain('bar.localhost')->id);
    }

    /** @test */
    public function updating_a_tenant_works()
    {
        $id = 'abc' . $this->randomString();
        $tenant = Tenant::create(['foo.localhost'], ['id' => $id]);
        $tenant->foo = 'bar';
        $tenant->save();
        $this->assertEquals(['id' => $id, 'foo' => 'bar', '_tenancy_db_name' => $tenant->database()->getName()], $tenant->data);
        $this->assertEquals(['id' => $id, 'foo' => 'bar', '_tenancy_db_name' => $tenant->database()->getName()], tenancy()->find($id)->data);

        $tenant->addDomains('abc.localhost');
        $tenant->save();
        $this->assertEqualsCanonicalizing(['foo.localhost', 'abc.localhost'], $tenant->domains);
        $this->assertEqualsCanonicalizing(['foo.localhost', 'abc.localhost'], tenancy()->find($id)->domains);

        $tenant->removeDomains(['foo.localhost']);
        $tenant->save();
        $this->assertEqualsCanonicalizing(['abc.localhost'], $tenant->domains);
        $this->assertEqualsCanonicalizing(['abc.localhost'], tenancy()->find($id)->domains);

        $tenant->withDomains(['completely.localhost', 'different.localhost', 'domains.localhost']);
        $tenant->save();
        $this->assertEqualsCanonicalizing(['completely.localhost', 'different.localhost', 'domains.localhost'], $tenant->domains);
        $this->assertEqualsCanonicalizing(['completely.localhost', 'different.localhost', 'domains.localhost'], tenancy()->find($id)->domains);
    }

    /** @test */
    public function with_methods_work()
    {
        $id = 'foo' . $this->randomString();
        $tenant = Tenant::new()->withDomains(['foo.localhost'])->with('id', $id);
        $this->assertSame($id, $tenant->id);

        $id2 = 'bar' . $this->randomString();
        $tenant2 = Tenant::new()->withDomains(['bar.localhost'])->withId($id2)->withFooBar('xyz');
        $this->assertSame($id2, $tenant2->data['id']);
        $this->assertSame('xyz', $tenant2->foo_bar);
        $this->assertArrayHasKey('foo_bar', $tenant2->data);
    }

    /** @test */
    public function an_exception_is_thrown_when_an_unknown_method_is_called()
    {
        $tenant = Tenant::new();
        $this->expectException(\BadMethodCallException::class);
        $tenant->sdjigndfgnjdfgj();
    }

    /** @test */
    public function tenant_data_can_be_set_during_creation()
    {
        Tenant::new()->withData(['foo' => 'bar'])->save();

        $data = tenancy()->all()->first()->data;
        unset($data['id']);
        unset($data['_tenancy_db_name']);

        $this->assertSame(['foo' => 'bar'], $data);
    }

    /** @test */
    public function run_method_works()
    {
        $this->assertSame(null, tenancy()->getTenant());

        $users_table_empty = function () {
            return count(\DB::table('users')->get()) === 0;
        };

        $tenant = Tenant::new()->save();
        \Artisan::call('tenants:migrate', [
            '--tenants' => [$tenant->id],
        ]);
        tenancy()->initialize($tenant);
        $this->assertTrue($users_table_empty());
        tenancy()->end();

        $foo = $tenant->run(function () {
            \DB::table('users')->insert([
                'name' => 'foo',
                'email' => 'foo@bar.xy',
                'password' => bcrypt('secret'),
            ]);

            return 'foo';
        });

        // test return value
        $this->assertSame('foo', $foo);

        // test that tenancy was ended
        $this->assertSame(false, tenancy()->initialized);
        $this->assertSame(null, tenancy()->getTenant());

        // test closure
        tenancy()->initialize($tenant);
        $this->assertFalse($users_table_empty());

        // test returning to original tenant
        $tenant2 = Tenant::new()->save();
        \Artisan::call('tenants:migrate', [
            '--tenants' => [$tenant2->id],
        ]);

        tenancy()->initialize($tenant2);
        $this->assertSame($tenant2, tenancy()->getTenant());
        $this->assertTrue($users_table_empty());

        $tenant->run(function () {
            \DB::table('users')->insert([
                'name' => 'bar',
                'email' => 'bar@bar.xy',
                'password' => bcrypt('secret'),
            ]);
        });

        $this->assertSame($tenant2, tenancy()->getTenant());

        $this->assertSame(2, $tenant->run(function () {
            return \DB::table('users')->count();
        }));

        // test that the tenant variable can be accessed
        $this->assertSame($tenant->id, $tenant->run(function ($tenant) {
            return $tenant->id;
        }));
    }

    /** @test */
    public function extra_config_is_merged_into_the_connection_config_array()
    {
        $tenant = Tenant::new()->withData([
            '_tenancy_db_link' => 'foo',
            '_tenancy_db_name' => 'dbname',
            '_tenancy_db_username' => 'usernamefoo',
            '_tenancy_db_password' => 'passwordfoo',
            '_tenancy_db_connection' => 'mysql',
        ]);

        config(['database.connections.mysql' => [
            'driver' => 'mysql',
            'url' => null,
            'host' => 'mysql',
            'port' => '3306',
            'database' => 'main',
            'username' => 'root',
            'password' => 'password',
            'unix_socket' => '',
            'charset' => 'utf8mb4',
            'collation' => 'utf8mb4_unicode_ci',
            'prefix' => '',
            'prefix_indexes' => true,
            'strict' => true,
            'engine' => null,
            'options' => [],
        ]]);

        $this->assertEquals([
            'database' => 'dbname',
            'username' => 'usernamefoo',
            'password' => 'passwordfoo',
            'link' => 'foo',

            'driver' => 'mysql',
            'url' => null,
            'host' => 'mysql',
            'port' => '3306',
            'unix_socket' => '',
            'charset' => 'utf8mb4',
            'collation' => 'utf8mb4_unicode_ci',
            'prefix' => '',
            'prefix_indexes' => true,
            'strict' => true,
            'engine' => null,
            'options' => [],
        ], $tenant->database()->connection());
    }
}