TenantAtlas/tests/Feature/ExecuteRestoreRunJobTest.php

<?php

use App\Jobs\ExecuteRestoreRunJob;
use App\Models\BackupItem;
use App\Models\BackupSet;
use App\Models\Policy;
use App\Models\RestoreRun;
use App\Models\Tenant;
use App\Services\Intune\AuditLogger;
use App\Services\Intune\RestoreService;
use App\Services\OperationRunService;
use App\Support\RestoreRunStatus;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Mockery\MockInterface;

uses(RefreshDatabase::class);

test('execute restore run job moves queued to running and calls the executor', function () {
    $tenant = Tenant::factory()->create([
        'tenant_id' => 'tenant-1',
        'name' => 'Tenant One',
        'metadata' => [],
    ]);

    $backupSet = BackupSet::create([
        'tenant_id' => $tenant->id,
        'name' => 'Backup',
        'status' => 'completed',
        'item_count' => 0,
    ]);

    $restoreRun = RestoreRun::create([
        'tenant_id' => $tenant->id,
        'backup_set_id' => $backupSet->id,
        'requested_by' => 'actor@example.com',
        'is_dry_run' => false,
        'status' => RestoreRunStatus::Queued->value,
        'requested_items' => null,
        'preview' => [],
        'results' => null,
        'metadata' => [],
    ]);

    $restoreService = $this->mock(RestoreService::class, function (MockInterface $mock) use ($tenant, $backupSet) {
        $mock->shouldReceive('executeForRun')
            ->once()
            ->withArgs(function (RestoreRun $run, Tenant $runTenant, BackupSet $runBackupSet, ?string $email, ?string $name) use ($tenant, $backupSet): bool {
                return $run->status === RestoreRunStatus::Running->value
                    && $runTenant->is($tenant)
                    && $runBackupSet->is($backupSet)
                    && $email === 'actor@example.com'
                    && $name === 'Actor';
            })
            ->andReturnUsing(function (RestoreRun $run): RestoreRun {
                $run->update([
                    'status' => RestoreRunStatus::Completed->value,
                    'completed_at' => now(),
                ]);

                return $run->refresh();
            });
    });

    $operationRun = app(OperationRunService::class)->ensureRun(
        tenant: $tenant,
        type: 'restore.execute',
        inputs: [
            'restore_run_id' => $restoreRun->id,
            'backup_set_id' => $backupSet->id,
            'is_dry_run' => false,
        ],
        initiator: null,
    );

    $job = new ExecuteRestoreRunJob($restoreRun->id, 'actor@example.com', 'Actor', $operationRun);
    $job->handle($restoreService, app(AuditLogger::class));

    $restoreRun->refresh();

    expect($restoreRun->started_at)->not->toBeNull();
    expect($restoreRun->status)->toBe(RestoreRunStatus::Completed->value);
});

test('execute restore run job persists per-item outcomes keyed by backup_item_id', function () {
    $tenant = Tenant::factory()->create([
        'tenant_id' => 'tenant-results',
        'name' => 'Tenant Results',
        'metadata' => [],
    ]);

    $backupSet = BackupSet::create([
        'tenant_id' => $tenant->id,
        'name' => 'Backup',
        'status' => 'completed',
        'item_count' => 1,
    ]);

    $policy = Policy::create([
        'tenant_id' => $tenant->id,
        'external_id' => 'policy-results',
        'policy_type' => 'unknownPreviewOnlyType',
        'display_name' => 'Preview-only policy',
        'platform' => 'windows',
    ]);

    $backupItem = BackupItem::create([
        'tenant_id' => $tenant->id,
        'backup_set_id' => $backupSet->id,
        'policy_id' => $policy->id,
        'policy_identifier' => $policy->external_id,
        'policy_type' => $policy->policy_type,
        'platform' => $policy->platform,
        'payload' => ['id' => $policy->external_id],
        'metadata' => [
            'displayName' => 'Backup Policy',
        ],
    ]);

    $restoreRun = RestoreRun::create([
        'tenant_id' => $tenant->id,
        'backup_set_id' => $backupSet->id,
        'requested_by' => 'actor@example.com',
        'is_dry_run' => false,
        'status' => RestoreRunStatus::Queued->value,
        'requested_items' => [$backupItem->id],
        'preview' => [],
        'results' => null,
        'metadata' => [],
    ]);

    $restoreService = $this->mock(RestoreService::class, function (MockInterface $mock) use ($backupItem): void {
        $mock->shouldReceive('executeForRun')
            ->once()
            ->andReturnUsing(function (...$args) use ($backupItem): RestoreRun {
                /** @var RestoreRun $run */
                $run = $args[0];

                $run->update([
                    'status' => RestoreRunStatus::Completed->value,
                    'completed_at' => now(),
                    'results' => [
                        'items' => [
                            (string) $backupItem->id => [
                                'backup_item_id' => $backupItem->id,
                                'status' => 'skipped',
                            ],
                        ],
                    ],
                ]);

                return $run->refresh();
            });
    });

    $operationRun = app(OperationRunService::class)->ensureRun(
        tenant: $tenant,
        type: 'restore.execute',
        inputs: [
            'restore_run_id' => $restoreRun->id,
            'backup_set_id' => $backupSet->id,
            'is_dry_run' => false,
        ],
        initiator: null,
    );

    $job = new ExecuteRestoreRunJob($restoreRun->id, 'actor@example.com', 'Actor', $operationRun);
    $job->handle($restoreService, app(AuditLogger::class));

    $restoreRun->refresh();

    expect($restoreRun->completed_at)->not->toBeNull();
    expect($restoreRun->results)->toBeArray();
    expect($restoreRun->results['items'][(string) $backupItem->id]['backup_item_id'] ?? null)->toBe($backupItem->id);
    expect($restoreRun->results['items'][(string) $backupItem->id]['status'] ?? null)->toBe('skipped');
});