Fixed #54: Now use pytest everywhere \o/

api/funkwhale_api/music/factories.py

@@ -0,0 +1,102 @@
+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'