tenancy/tests/MaintenanceModeTest.php

<?php

declare(strict_types=1);

use Illuminate\Support\Facades\Artisan;
use Stancl\Tenancy\Database\Concerns\MaintenanceMode;
use Symfony\Component\HttpKernel\Exception\HttpException;
use Illuminate\Support\Facades\Route;
use Stancl\Tenancy\Middleware\CheckTenantForMaintenanceMode;
use Stancl\Tenancy\Middleware\InitializeTenancyByDomain;
use Stancl\Tenancy\Tests\Etc\Tenant;

test('tenant can be in maintenance mode', function () {
    Route::get('/foo', function () {
        return 'bar';
    })->middleware([InitializeTenancyByDomain::class, CheckTenantForMaintenanceMode::class]);

    $tenant = MaintenanceTenant::create();
    $tenant->domains()->create([
        'domain' => 'acme.localhost',
    ]);

    pest()->get('http://acme.localhost/foo')
        ->assertSuccessful();

    tenancy()->end(); // Flush stored tenant instance

    $tenant->putDownForMaintenance();

    pest()->expectException(HttpException::class);
    pest()->get('http://acme.localhost/foo');

    tenancy()->end();

    $tenant->bringUpFromMaintenance();

    tenancy()->end();

    pest()->get('http://acme.localhost/foo')
        ->assertSuccessful()
        ->assertSeeText('bar');
});

test('tenant can be in maintenance mode from command', function() {
    Route::get('/foo', function () {
        return 'bar';
    })->middleware([InitializeTenancyByDomain::class, CheckTenantForMaintenanceMode::class]);

    $tenant = MaintenanceTenant::create();
    $tenant->domains()->create([
        'domain' => 'acme.localhost',
    ]);

    pest()->get('http://acme.localhost/foo')
        ->assertSuccessful();

    tenancy()->end(); // Flush stored tenant instance

    Artisan::call('tenancy:down');

    pest()->expectException(HttpException::class);
    pest()->get('http://acme.localhost/foo');

    tenancy()->end();

    Artisan::call('tenancy:up');

    tenancy()->end();

    pest()->get('http://acme.localhost/foo')
        ->assertSuccessful()
        ->assertSeeText('bar');
});

class MaintenanceTenant extends Tenant
{
    use MaintenanceMode;
}