funkwhale/api/funkwhale_api/radios/factories.py

import factory

from funkwhale_api.factories import registry
from funkwhale_api.users.factories import UserFactory


@registry.register
class RadioFactory(factory.django.DjangoModelFactory):
    name = factory.Faker("name")
    description = factory.Faker("paragraphs")
    user = factory.SubFactory(UserFactory)
    config = []

    class Meta:
        model = "radios.Radio"


@registry.register
class RadioSessionFactory(factory.django.DjangoModelFactory):
    user = factory.SubFactory(UserFactory)

    class Meta:
        model = "radios.RadioSession"


@registry.register(name="radios.CustomRadioSession")
class CustomRadioSessionFactory(factory.django.DjangoModelFactory):
    user = factory.SubFactory(UserFactory)
    radio_type = "custom"
    custom_radio = factory.SubFactory(
        RadioFactory, user=factory.SelfAttribute("..user")
    )

    class Meta:
        model = "radios.RadioSession"