funkwhale/api/funkwhale_api/radios/factories.py

36 wiersze
981 B
Python

import factory
from funkwhale_api.factories import NoUpdateOnCreate, registry
from funkwhale_api.users.factories import UserFactory
@registry.register
class RadioFactory(NoUpdateOnCreate, 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(NoUpdateOnCreate, factory.django.DjangoModelFactory):
user = factory.SubFactory(UserFactory)
class Meta:
model = "radios.RadioSession"
@registry.register(name="radios.CustomRadioSession")
class CustomRadioSessionFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
user = factory.SubFactory(UserFactory)
radio_type = "custom"
custom_radio = factory.SubFactory(
RadioFactory, user=factory.SelfAttribute("..user")
)
class Meta:
model = "radios.RadioSession"