From 4f7fa09a78336af3b1c20eca032e6d63cc689568 Mon Sep 17 00:00:00 2001
From: Eliot Berriot <contact@eliotberriot.com>
Date: Sun, 18 Mar 2018 21:31:59 +0100
Subject: [PATCH] Playlisttrack factory
---
api/funkwhale_api/playlists/factories.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/api/funkwhale_api/playlists/factories.py b/api/funkwhale_api/playlists/factories.py
index 19e4770c..cddea600 100644
--- a/api/funkwhale_api/playlists/factories.py
+++ b/api/funkwhale_api/playlists/factories.py
@@ -1,6 +1,7 @@
import factory
from funkwhale_api.factories import registry
+from funkwhale_api.music.factories import TrackFactory
from funkwhale_api.users.factories import UserFactory
@@ -11,3 +12,12 @@ class PlaylistFactory(factory.django.DjangoModelFactory):
class Meta:
model = 'playlists.Playlist'
+
+
+@registry.register
+class PlaylistTrackFactory(factory.django.DjangoModelFactory):
+ playlist = factory.SubFactory(PlaylistFactory)
+ track = factory.SubFactory(TrackFactory)
+
+ class Meta:
+ model = 'playlists.PlaylistTrack'
--
GitLab