tenancy/tests/AutomaticModeTest.php

<?php

declare(strict_types=1);

namespace Stancl\Tenancy\Tests;

use Illuminate\Support\Facades\Event;
use Stancl\Tenancy\Contracts\TenancyBootstrapper;
use Stancl\Tenancy\Events\TenancyEnded;
use Stancl\Tenancy\Events\TenancyInitialized;
use Stancl\Tenancy\Listeners\BootstrapTenancy;
use Stancl\Tenancy\Listeners\RevertToCentralContext;
use Stancl\Tenancy\Tests\Etc\Tenant;

class AutomaticModeTest extends TestCase
{
    public function setUp(): void
    {
        parent::setUp();

        Event::listen(TenancyInitialized::class, BootstrapTenancy::class);
        Event::listen(TenancyEnded::class, RevertToCentralContext::class);
    }

    /** @test */
    public function context_is_switched_when_tenancy_is_initialized()
    {
        config(['tenancy.bootstrappers' => [
            MyBootstrapper::class,
        ]]);

        $tenant = Tenant::create([
            'id' => 'acme',
        ]);

        tenancy()->initialize($tenant);

        $this->assertSame('acme', app('tenancy_initialized_for_tenant'));
    }

    /** @test */
    public function context_is_reverted_when_tenancy_is_ended()
    {
        $this->context_is_switched_when_tenancy_is_initialized();

        tenancy()->end();

        $this->assertSame(true, app('tenancy_ended'));
    }

    /** @test */
    public function context_is_switched_when_tenancy_is_reinitialized()
    {
        config(['tenancy.bootstrappers' => [
            MyBootstrapper::class,
        ]]);

        $tenant = Tenant::create([
            'id' => 'acme',
        ]);

        tenancy()->initialize($tenant);

        $this->assertSame('acme', app('tenancy_initialized_for_tenant'));

        $tenant2 = Tenant::create([
            'id' => 'foobar',
        ]);

        tenancy()->initialize($tenant2);

        $this->assertSame('foobar', app('tenancy_initialized_for_tenant'));
    }

    /** @test */
    public function central_helper_runs_callbacks_in_the_central_state()
    {
        tenancy()->initialize($tenant = Tenant::create());

        tenancy()->central(function () {
            $this->assertSame(null, tenant());
        });

        $this->assertSame($tenant, tenant());
    }

    /** @test */
    public function central_helper_returns_the_value_from_the_callback()
    {
        tenancy()->initialize(Tenant::create());

        $this->assertSame('foo', tenancy()->central(function () {
            return 'foo';
        }));
    }

    /** @test */
    public function central_helper_reverts_back_to_tenant_context()
    {
        tenancy()->initialize($tenant = Tenant::create());

        tenancy()->central(function () {
            //
        });

        $this->assertSame($tenant, tenant());
    }

    /** @test */
    public function central_helper_doesnt_change_tenancy_state_when_called_in_central_context()
    {
        $this->assertFalse(tenancy()->initialized);
        $this->assertNull(tenant());

        tenancy()->central(function () {
            //
        });

        $this->assertFalse(tenancy()->initialized);
        $this->assertNull(tenant());
    }
}

class MyBootstrapper implements TenancyBootstrapper
{
    public function bootstrap(\Stancl\Tenancy\Contracts\Tenant $tenant)
    {
        app()->instance('tenancy_initialized_for_tenant', $tenant->getTenantKey());
    }

    public function revert()
    {
        app()->instance('tenancy_ended', true);
    }
}