36 lines
981 B
Python
36 lines
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"
|