yetangitu/funkwhale
1
0
Fork 0
mirror of https://code.eliotberriot.com/funkwhale/funkwhale.git synced 2025-10-06 02:49:56 +02:00
Code Issues Releases Wiki Activity

More factories

Browse source
This commit is contained in:
Eliot Berriot 2018-04-05 23:27:19 +02:00
parent feab0f98ba
commit 87daa81762
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
2 changed files with 65 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
api/funkwhale_api/music/factories.py
View file

@ -2,6 +2,10 @@ import factory
import os
from funkwhale_api.factories import registry, ManyToManyFromList
from funkwhale_api.federation.factories import (
AudioMetadataFactory,
ActorFactory,
)
from funkwhale_api.users.factories import UserFactory
SAMPLES_PATH = os.path.join(
@ -61,6 +65,13 @@ class ImportBatchFactory(factory.django.DjangoModelFactory):
class Meta:
model = 'music.ImportBatch'
class Params:
federation = factory.Trait(
submitted_by=None,
federation_actor=factory.SubFactory(ActorFactory),
source='federation',
)
@registry.register
class ImportJobFactory(factory.django.DjangoModelFactory):
@ -71,6 +82,13 @@ class ImportJobFactory(factory.django.DjangoModelFactory):
class Meta:
model = 'music.ImportJob'
class Params:
federation = factory.Trait(
batch=factory.SubFactory(ImportBatchFactory, federation=True),
federation_source=factory.Faker('url'),
metadata=factory.SubFactory(AudioMetadataFactory),
)
@registry.register(name='music.FileImportJob')
class FileImportJobFactory(ImportJobFactory):