TenantAtlas/apps/platform/database/factories/BaselineProfileFactory.php

<?php

namespace Database\Factories;

use App\Models\BaselineProfile;
use App\Models\User;
use App\Models\Workspace;
use App\Support\Baselines\BaselineCaptureMode;
use App\Support\Baselines\BaselineProfileStatus;
use Illuminate\Database\Eloquent\Factories\Factory;

/**
 * @extends Factory<BaselineProfile>
 */
class BaselineProfileFactory extends Factory
{
    protected $model = BaselineProfile::class;

    /**
     * @return array<string, mixed>
     */
    public function definition(): array
    {
        return [
            'workspace_id' => Workspace::factory(),
            'name' => fake()->unique()->words(3, true),
            'description' => fake()->optional()->sentence(),
            'version_label' => fake()->optional()->numerify('v#.#'),
            'status' => BaselineProfileStatus::Draft->value,
            'capture_mode' => BaselineCaptureMode::Opportunistic->value,
            'scope_jsonb' => ['policy_types' => [], 'foundation_types' => []],
            'active_snapshot_id' => null,
            'created_by_user_id' => null,
        ];
    }

    public function active(): static
    {
        return $this->state(fn (): array => [
            'status' => BaselineProfileStatus::Active->value,
        ]);
    }

    public function archived(): static
    {
        return $this->state(fn (): array => [
            'status' => BaselineProfileStatus::Archived->value,
        ]);
    }

    public function withScope(array $policyTypes): static
    {
        return $this->state(fn (): array => [
            'scope_jsonb' => ['policy_types' => $policyTypes, 'foundation_types' => []],
        ]);
    }

    public function createdBy(User $user): static
    {
        return $this->state(fn (): array => [
            'created_by_user_id' => $user->getKey(),
        ]);
    }
}