osu-web / database / factories / BeatmapDiscussionFactory.php

<?php

// Copyright (c) ppy Pty Ltd <[e-mail]>. Licensed under the GNU Affero General Public License v3.0.
// See the LICENCE file in the repository root for full licence text.

namespace Database\Factories;

use App\Models\BeatmapDiscussion;

class BeatmapDiscussionFactory extends Factory
{
    // TODO: decouple message_type
    const DEFAULTS = [
        'general' => [
            'timestamp' => null,
            'message_type' => 'problem',
        ],
        'review' => [
            'timestamp' => null,
            'message_type' => 'review',
        ],
        'timeline' => [
            'timestamp' => 0,
            'message_type' => 'problem',
        ],
    ];

    protected $model = BeatmapDiscussion::class;

    public function definition(): array
    {
        return array_merge(array_rand_val(static::DEFAULTS), [
            'resolved' => false,
        ]);
    }

    public function general()
    {
        return $this->state(static::DEFAULTS['general']);
    }

    public function mapperNote()
    {
        return $this->messageType('mapper_note');
    }

    public function messageType(string $type)
    {
        return $this->state(['message_type' => $type]);
    }

    public function problem()
    {
        return $this->messageType('problem');
    }

    public function review()
    {
        return $this->state(static::DEFAULTS['review']);
    }

    public function timeline()
    {
        return $this->state(static::DEFAULTS['timeline']);
    }
}