kopia lustrzana https://dev.funkwhale.audio/funkwhale/funkwhale
103 wiersze
2.5 KiB
Python
103 wiersze
2.5 KiB
Python
|
import factory
|
||
|
import os
|
||
|
|
||
|
from funkwhale_api.factories import registry, ManyToManyFromList
|
||
|
from funkwhale_api.users.factories import UserFactory
|
||
|
|
||
|
SAMPLES_PATH = os.path.join(
|
||
|
os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))),
|
||
|
'tests', 'music'
|
||
|
)
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class ArtistFactory(factory.django.DjangoModelFactory):
|
||
|
name = factory.Faker('name')
|
||
|
mbid = factory.Faker('uuid4')
|
||
|
|
||
|
class Meta:
|
||
|
model = 'music.Artist'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class AlbumFactory(factory.django.DjangoModelFactory):
|
||
|
title = factory.Faker('sentence', nb_words=3)
|
||
|
mbid = factory.Faker('uuid4')
|
||
|
release_date = factory.Faker('date')
|
||
|
cover = factory.django.ImageField()
|
||
|
artist = factory.SubFactory(ArtistFactory)
|
||
|
release_group_id = factory.Faker('uuid4')
|
||
|
|
||
|
class Meta:
|
||
|
model = 'music.Album'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class TrackFactory(factory.django.DjangoModelFactory):
|
||
|
title = factory.Faker('sentence', nb_words=3)
|
||
|
mbid = factory.Faker('uuid4')
|
||
|
album = factory.SubFactory(AlbumFactory)
|
||
|
artist = factory.SelfAttribute('album.artist')
|
||
|
position = 1
|
||
|
tags = ManyToManyFromList('tags')
|
||
|
class Meta:
|
||
|
model = 'music.Track'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class TrackFileFactory(factory.django.DjangoModelFactory):
|
||
|
track = factory.SubFactory(TrackFactory)
|
||
|
audio_file = factory.django.FileField(
|
||
|
from_path=os.path.join(SAMPLES_PATH, 'test.ogg'))
|
||
|
|
||
|
class Meta:
|
||
|
model = 'music.TrackFile'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class ImportBatchFactory(factory.django.DjangoModelFactory):
|
||
|
submitted_by = factory.SubFactory(UserFactory)
|
||
|
|
||
|
class Meta:
|
||
|
model = 'music.ImportBatch'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class ImportJobFactory(factory.django.DjangoModelFactory):
|
||
|
batch = factory.SubFactory(ImportBatchFactory)
|
||
|
source = factory.Faker('url')
|
||
|
mbid = factory.Faker('uuid4')
|
||
|
|
||
|
class Meta:
|
||
|
model = 'music.ImportJob'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class WorkFactory(factory.django.DjangoModelFactory):
|
||
|
mbid = factory.Faker('uuid4')
|
||
|
language = 'eng'
|
||
|
nature = 'song'
|
||
|
title = factory.Faker('sentence', nb_words=3)
|
||
|
|
||
|
class Meta:
|
||
|
model = 'music.Work'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class LyricsFactory(factory.django.DjangoModelFactory):
|
||
|
work = factory.SubFactory(WorkFactory)
|
||
|
url = factory.Faker('url')
|
||
|
content = factory.Faker('paragraphs', nb=4)
|
||
|
|
||
|
class Meta:
|
||
|
model = 'music.Lyrics'
|
||
|
|
||
|
|
||
|
@registry.register
|
||
|
class TagFactory(factory.django.DjangoModelFactory):
|
||
|
name = factory.SelfAttribute('slug')
|
||
|
slug = factory.Faker('slug')
|
||
|
|
||
|
class Meta:
|
||
|
model = 'taggit.Tag'
|